Searched refs:permute_dims_first_operand (Results 1 – 1 of 1) sorted by relevance
829 auto permute_dims_first_operand = in CanPropagate() local832 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()834 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()889 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate() local894 DimLookUp(permute_dims_first_operand, dim_map_val_op_0.first); in CanPropagate()896 DimLookUp(permute_dims_first_operand, dim_map_val_op_0.second); in CanPropagate()952 auto permute_dims_first_operand = instr_to_dim_permute_map_[first_operand]; in CanPropagate() local957 if (permute_dims_first_operand != permute_dims_second_operand) { in CanPropagate()967 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()969 DimLookUp(permute_dims_first_operand, old_space_dim); in CanPropagate()