Searched refs:pivot_operand (Results 1 – 1 of 1) sorted by relevance
687 HloInstruction* pivot_operand = nullptr; in CanPropagate() local714 if (pivot_operand == nullptr) { in CanPropagate()715 pivot_operand = old_producer; in CanPropagate()718 if (instr_to_dim_map_[pivot_operand] != in CanPropagate()725 auto pivot_new_instr = old_to_new_instrs_[pivot_operand]; in CanPropagate()743 j == instr_to_dim_map_[pivot_operand].second)) { in CanPropagate()1215 HloInstruction* pivot_operand = nullptr; in Propagate() local1223 old_to_new_instrs_[pivot_operand]->shape().dimensions() < in Propagate()1227 pivot_operand = consumer->mutable_operand(pivot_operand_number); in Propagate()1232 producer = pivot_operand; in Propagate()