Searched refs:old_to_new_instrs_ (Results 1 – 1 of 1) sorted by relevance
223 absl::flat_hash_map<HloInstruction*, HloInstruction*> old_to_new_instrs_; member in xla::__anon6418faad0111::ConvolutionVisitor639 if (old_to_new_instrs_.contains(instr->mutable_operand(0))) { in Run()641 } else if (old_to_new_instrs_.contains(instr->mutable_operand(1))) { in Run()650 instr, old_to_new_instrs_[instr])); in Run()658 if (old_to_new_instrs_.count(instr->mutable_operand(i))) { in Run()699 if (!old_to_new_instrs_.contains(old_producer) && in CanPropagate()711 CHECK(old_to_new_instrs_.contains(old_producer)); in CanPropagate()725 auto pivot_new_instr = old_to_new_instrs_[pivot_operand]; in CanPropagate()727 auto new_instr = old_to_new_instrs_[old_producer]; in CanPropagate()760 if (!old_to_new_instrs_.contains(consumer->mutable_operand(0))) { in CanPropagate()[all …]