Searched refs:permute_dims (Results 1 – 1 of 1) sorted by relevance
263 int64_t DimLookUp(absl::Span<const int64_t> permute_dims, int64_t id) { in DimLookUp() argument264 return permute_dims[id]; in DimLookUp()269 int64_t ReverseDimLookUp(absl::Span<const int64_t> permute_dims, int64_t id) { in ReverseDimLookUp() argument270 return std::distance(permute_dims.begin(), absl::c_find(permute_dims, id)); in ReverseDimLookUp()359 auto permute_dims = instr_to_dim_permute_map_[new_operand]; in GetSpatialDimsToSplit() local364 new_dims[0] = DimLookUp(permute_dims, old_dims[0]); in GetSpatialDimsToSplit()1004 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in CanPropagate() local1007 if (pivot_permute_dims[j] != permute_dims[j]) { in CanPropagate()1045 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in CanPropagate() local1049 if (pivot_permute_dims[j] != permute_dims[j]) { in CanPropagate()[all …]