Searched refs:dim_map (Results 1 – 2 of 2) sorted by relevance
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]
1278 absl::InlinedVector<int64, 8> dim_map; in AlignLayouts() local1279 dim_map.reserve(simple_output_shape->rank()); in AlignLayouts()1282 dim_map.push_back(i); in AlignLayouts()1286 d = dim_map[d]; in AlignLayouts()