Home
last modified time | relevance | path

Searched refs:old_instruction (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_clone_context.h44 void MapInstruction(const HloInstruction* old_instruction, in MapInstruction() argument
46 instructions_[old_instruction] = new_instruction; in MapInstruction()
56 HloInstruction* FindInstruction(const HloInstruction* old_instruction) const { in FindInstruction() argument
57 return FindOrDefault(instructions_, old_instruction, nullptr); in FindInstruction()
67 HloInstruction* GetInstruction(const HloInstruction* old_instruction) const { in GetInstruction() argument
68 return FindOrDie(instructions_, old_instruction); in GetInstruction()
Dwhile_loop_invariant_code_motion.cc59 HloInstruction* old_instruction = frame->instruction; in CreateLoopInvariantCopy() local
71 absl::c_transform(old_instruction->operands(), in CreateLoopInvariantCopy()
75 parent_of_while->AddInstruction(old_instruction->CloneWithNewOperands( in CreateLoopInvariantCopy()
76 old_instruction->shape(), new_operands)); in CreateLoopInvariantCopy()
78 InsertOrDie(hoisted_instructions, old_instruction, new_instruction); in CreateLoopInvariantCopy()
83 CHECK_EQ(unhoisted_invariant_instructions->erase(old_instruction), in CreateLoopInvariantCopy()
84 to_hoist != old_instruction && in CreateLoopInvariantCopy()
85 old_instruction->opcode() != HloOpcode::kConstant); in CreateLoopInvariantCopy()
Dhlo_computation.cc756 HloInstruction* old_instruction, in ReplaceWithNewInstruction() argument
758 return ReplaceInstruction(old_instruction, in ReplaceWithNewInstruction()
762 Status HloComputation::ReplaceInstruction(HloInstruction* old_instruction, in ReplaceInstruction() argument
765 ShapeUtil::Compatible(old_instruction->shape(), new_instruction->shape())) in ReplaceInstruction()
766 << ShapeUtil::HumanString(old_instruction->shape()) << " vs " in ReplaceInstruction()
769 VLOG(10) << "transformed " << old_instruction->ToString() << " to " in ReplaceInstruction()
778 new_instruction->set_metadata(old_instruction->metadata()); in ReplaceInstruction()
780 TF_RETURN_IF_ERROR(old_instruction->ReplaceAllUsesWith(new_instruction)); in ReplaceInstruction()
781 return RemoveInstructionAndUnusedOperands(old_instruction); in ReplaceInstruction()
Dbatchnorm_expander.cc140 HloInstruction* old_instruction, in ReplaceWithNewInstruction() argument
143 old_instruction, std::move(new_instruction))); in ReplaceWithNewInstruction()
151 Status ReplaceInstruction(HloInstruction* old_instruction, in ReplaceInstruction() argument
154 computation_->ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstruction()
Dhlo_computation.h278 HloInstruction* old_instruction,
284 Status ReplaceInstruction(HloInstruction* old_instruction,
Dalgebraic_simplifier.cc339 bool ReplaceInstructionIfSameShape(HloInstruction* old_instruction,
367 HloInstruction* old_instruction, in ReplaceWithNewInstruction() argument
370 VLOG(3) << " old: " << old_instruction->ToString(); in ReplaceWithNewInstruction()
373 old_instruction, std::move(new_instruction))); in ReplaceWithNewInstruction()
381 Status ReplaceInstruction(HloInstruction* old_instruction, in ReplaceInstruction() argument
384 VLOG(3) << " old: " << old_instruction->ToString(); in ReplaceInstruction()
387 computation_->ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstruction()
480 HloInstruction* old_instruction, HloInstruction* new_instruction) { in ReplaceInstructionIfSameShape() argument
481 if (!SameShape(old_instruction, new_instruction)) { in ReplaceInstructionIfSameShape()
484 TF_CHECK_OK(ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstructionIfSameShape()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc86 for (auto old_instruction : computation->MakeInstructionPostOrder()) { in FinishVisit() local
88 clone_context_.FindInstruction(old_instruction)) { in FinishVisit()
89 new_instruction->SetAndSanitizeName(old_instruction->name()); in FinishVisit()