HomeSort by relevance Sort by last modified time
    Searched refs:dim_map (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tests/
test_utils.py 41 dim_map = {d: i for i, d in enumerate(data_format_src)}
42 transpose_dims = [dim_map[d] for d in data_format_dst]
62 dim_map = {d: i for i, d in enumerate(data_format_src)}
63 permuted_dims = [dims[dim_map[d]] for d in data_format_dst]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.cc     [all...]

Completed in 650 milliseconds