Searched refs:instr_to_dim_permute_map_ (Results 1 – 1 of 1) sorted by relevance
232 instr_to_dim_permute_map_; member in xla::__anon6418faad0111::ConvolutionVisitor726 auto pivot_permute_dims = instr_to_dim_permute_map_[pivot_new_instr]; in CanPropagate()728 auto permute_dims = instr_to_dim_permute_map_[new_instr]; in CanPropagate()830 instr_to_dim_permute_map_[first_operand]; in CanPropagate()864 instr_to_dim_permute_map_[second_operand]; in CanPropagate()889 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate()891 instr_to_dim_permute_map_[second_operand]; in CanPropagate()952 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate()954 instr_to_dim_permute_map_[second_operand]; in CanPropagate()1001 auto permute_dims = instr_to_dim_permute_map_[new_producer]; in PropagateOnBroadcast()[all …]