Searched refs:dim_map (Results 1 – 2 of 2) sorted by relevance
42 dim_map = {d: i for i, d in enumerate(data_format_src)}43 transpose_dims = [dim_map[d] for d in data_format_dst]63 dim_map = {d: i for i, d in enumerate(data_format_src)}64 permuted_dims = [dims[dim_map[d]] for d in data_format_dst]
1306 absl::InlinedVector<int64, 8> dim_map; in AlignLayouts() local1307 dim_map.reserve(simple_output_shape->rank()); in AlignLayouts()1310 dim_map.push_back(i); in AlignLayouts()1314 d = dim_map[d]; in AlignLayouts()