| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule.h | 63 HloInstruction* new_instruction) { in replace_instruction() argument
72 *instruction_it = new_instruction; in replace_instruction()
73 *id_it = new_instruction->unique_id(); in replace_instruction()
159 HloInstruction* new_instruction) { in replace_instruction() argument
161 new_instruction); in replace_instruction()
|
| D | hlo_computation.cc | 895 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction() argument
897 AddInstruction(std::move(new_instruction))); in ReplaceWithNewInstruction()
902 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewEntryComputationParameter() argument
904 std::move(new_instruction))); in ReplaceWithNewEntryComputationParameter()
908 HloInstruction* new_instruction) { in ReplaceInstruction() argument
910 ShapeUtil::Compatible(old_instruction->shape(), new_instruction->shape())) in ReplaceInstruction()
912 << ShapeUtil::HumanString(new_instruction->shape()); in ReplaceInstruction()
915 << new_instruction->ToString(); in ReplaceInstruction()
922 bool overwrite_op_name = new_instruction->metadata().op_name().empty() && in ReplaceInstruction()
925 new_instruction->metadata().op_name().empty() && in ReplaceInstruction()
[all …]
|
| D | dfs_hlo_visitor_with_default.h | 297 std::unique_ptr<HloInstruction> new_instruction) { in ReplaceWithNewInstruction() argument
300 VLOG(3) << " new: " << new_instruction->ToString(); in ReplaceWithNewInstruction()
302 old_instruction, std::move(new_instruction))); in ReplaceWithNewInstruction()
311 HloInstruction* new_instruction) { in ReplaceInstruction() argument
314 VLOG(3) << " new: " << new_instruction->ToString(); in ReplaceInstruction()
316 old_instruction, new_instruction)); in ReplaceInstruction()
|
| 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 | memory_space_assignment.cc | 2762 HloInstruction* new_instruction, HloInstruction* tuple, in ReplaceTupleWith() argument
2768 << ", new_instruction = " << new_instruction->ToString() in ReplaceTupleWith()
2771 HloComputation* computation = new_instruction->parent(); in ReplaceTupleWith()
2791 ReplaceTupleWith(new_instruction, get_operand(), in ReplaceTupleWith()
2795 if (subshape != new_instruction->shape()) { in ReplaceTupleWith()
2797 << ", new shape = " << new_instruction->shape().ToString() in ReplaceTupleWith()
2799 new_instruction = computation->AddInstruction( in ReplaceTupleWith()
2800 HloInstruction::CreateBitcast(subshape, new_instruction)); in ReplaceTupleWith()
2802 tuple->operand(i) == new_instruction) { in ReplaceTupleWith()
2807 << new_instruction->ToShortString() in ReplaceTupleWith()
[all …]
|
| 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 | hlo_computation.h | 337 std::unique_ptr<HloInstruction> new_instruction);
344 std::unique_ptr<HloInstruction> new_instruction);
352 HloInstruction* new_instruction);
|
| D | while_loop_expensive_invariant_code_motion.cc | 91 HloInstruction* new_instruction = parent_of_while->AddInstruction( in CreateLoopInvariantCopy() local
95 info.hoisted_copy = new_instruction; in CreateLoopInvariantCopy()
|
| D | conditional_code_motion.cc | 289 HloInstruction* new_instruction = parent->AddInstruction( in CopyInOrOutOfConditional() local
291 VLOG(2) << "new instruction:" << new_instruction->ToString() << "\n"; in CopyInOrOutOfConditional()
300 b2.mutable_operands().push_back(new_instruction); in CopyInOrOutOfConditional()
|
| D | memory_space_assignment.h | 567 StatusOr<HloInstruction*> ReplaceTupleWith(HloInstruction* new_instruction,
867 HloInstruction* new_instruction, HloInstructionSequence* new_sequence,
|
| D | algebraic_simplifier.cc | 446 HloInstruction* new_instruction);
783 HloInstruction* old_instruction, HloInstruction* new_instruction) { in ReplaceInstructionIfSameShape() argument
784 if (!SameShape(old_instruction, new_instruction)) { in ReplaceInstructionIfSameShape()
787 TF_CHECK_OK(ReplaceInstruction(old_instruction, new_instruction)); in ReplaceInstructionIfSameShape()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 88 if (auto new_instruction = in FinishVisit() local
90 new_instruction->SetAndSanitizeName(old_instruction->name()); in FinishVisit()
|