Searched refs:new_instruction (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | stable_sort_expander.cc | 172 auto new_instruction = in ExpandInstruction() local
174 replacements[inst] = new_instruction.get(); in ExpandInstruction()
176 cloned_root = new_instruction.get(); in ExpandInstruction()
178 comparator->AddInstruction(std::move(new_instruction)); in ExpandInstruction()
|
| D | hlo_clone_context.h | 45 HloInstruction* new_instruction) { in MapInstruction() argument
46 instructions_[old_instruction] = new_instruction; in MapInstruction()
|
| D | hlo_computation.cc | 757 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction() argument
759 AddInstruction(std::move(new_instruction))); in ReplaceWithNewInstruction()
763 HloInstruction* new_instruction) { in ReplaceInstruction() argument
765 ShapeUtil::Compatible(old_instruction->shape(), new_instruction->shape())) in ReplaceInstruction()
767 << ShapeUtil::HumanString(new_instruction->shape()); in ReplaceInstruction()
770 << new_instruction->ToString(); in ReplaceInstruction()
777 if (new_instruction->metadata().op_name().empty()) { in ReplaceInstruction()
778 new_instruction->set_metadata(old_instruction->metadata()); in ReplaceInstruction()
780 TF_RETURN_IF_ERROR(old_instruction->ReplaceAllUsesWith(new_instruction)); in ReplaceInstruction()
|
| D | while_loop_invariant_code_motion.cc | 74 HloInstruction* new_instruction = in CreateLoopInvariantCopy() local
78 InsertOrDie(hoisted_instructions, old_instruction, new_instruction); in CreateLoopInvariantCopy()
|
| D | batchnorm_expander.cc | 141 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction() argument
143 old_instruction, std::move(new_instruction))); in ReplaceWithNewInstruction()
152 HloInstruction* new_instruction) { in ReplaceInstruction() argument
154 computation_->ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstruction()
|
| D | hlo_computation.h | 279 std::unique_ptr<HloInstruction> new_instruction);
285 HloInstruction* new_instruction);
|
| D | algebraic_simplifier.cc | 340 HloInstruction* new_instruction);
368 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction() argument
371 VLOG(3) << " new: " << new_instruction->ToString(); in ReplaceWithNewInstruction()
373 old_instruction, std::move(new_instruction))); in ReplaceWithNewInstruction()
382 HloInstruction* new_instruction) { in ReplaceInstruction() argument
385 VLOG(3) << " new: " << new_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/ |
| D | hlo_extractor.cc | 87 if (auto new_instruction = in FinishVisit() local
89 new_instruction->SetAndSanitizeName(old_instruction->name()); in FinishVisit()
|