Searched refs:instr_to_dim_permute_map_ (Results 1 – 1 of 1) sorted by relevance
306 instr_to_dim_permute_map_; member in xla::__anon5e1d62920111::ConvolutionVisitor359 auto permute_dims = instr_to_dim_permute_map_[new_operand]; in GetSpatialDimsToSplit()1002 auto pivot_permute_dims = instr_to_dim_permute_map_[pivot_new_instr]; in CanPropagate()1004 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in CanPropagate()1042 auto pivot_permute_dims = instr_to_dim_permute_map_[pivot_new_instr]; in CanPropagate()1045 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in CanPropagate()1208 instr_to_dim_permute_map_[first_operand]; in CanPropagate()1249 instr_to_dim_permute_map_[second_operand]; in CanPropagate()1281 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate()1283 instr_to_dim_permute_map_[second_operand]; in CanPropagate()[all …]