Searched refs:pivot_new_instr (Results 1 – 1 of 1) sorted by relevance
725 auto pivot_new_instr = old_to_new_instrs_[pivot_operand]; in CanPropagate() local726 auto pivot_permute_dims = instr_to_dim_permute_map_[pivot_new_instr]; in CanPropagate()739 if (pivot_new_instr->shape().dimensions(j) != in CanPropagate()1263 HloInstruction* pivot_new_instr = old_to_new_instrs_[producer]; in Propagate() local1271 pivot_new_instr->shape().dimensions(space_dim)) { in Propagate()1287 pivot_new_instr->shape().dimensions(space_dim) * batch_size / in Propagate()1307 MakeReshapeHlo(pivot_new_instr->shape().dimensions(), in Propagate()