Home
last modified time | relevance | path

Searched refs:old_instr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_constant_sinking.cc29 HloInstruction* old_instr, HloInstruction* new_instr, in ReplaceUsesWhileKeepingLoopInvariance() argument
34 users.reserve(old_instr->user_count()); in ReplaceUsesWhileKeepingLoopInvariance()
35 absl::c_copy(old_instr->users(), std::back_inserter(users)); in ReplaceUsesWhileKeepingLoopInvariance()
39 if (user->operand(i) == old_instr && in ReplaceUsesWhileKeepingLoopInvariance()
Dspace_to_batch_converter.cc151 HloInstruction* new_instr, HloInstruction* old_instr,
173 StatusOr<HloInstruction*> BatchToSpace(HloInstruction* old_instr);
1650 HloInstruction* new_instr, HloInstruction* old_instr, in SelectValidPortion() argument
1654 auto old_shape = old_instr->shape(); in SelectValidPortion()
1707 HloInstruction* old_instr) { in BatchToSpace() argument
1708 if (batch_to_space_map_.count(old_instr)) { in BatchToSpace()
1709 CHECK_NE(batch_to_space_map_[old_instr], nullptr); in BatchToSpace()
1710 return batch_to_space_map_[old_instr]; in BatchToSpace()
1713 auto result = instr_to_dim_map_[old_instr]; in BatchToSpace()
1717 const int64 old_batch_size = old_instr->shape().dimensions(old_batch_dim); in BatchToSpace()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion.cc361 for (auto old_instr : def_to_use_order) { in CreateFusedComputation() local
362 if (old_instr->opcode() == HloOpcode::kParameter) { in CreateFusedComputation()
367 for (auto old_opnd : old_instr->operands()) { in CreateFusedComputation()
372 old_instr->CloneWithNewOperands(old_instr->shape(), new_opnds)); in CreateFusedComputation()
373 clone_map.insert({old_instr, new_instr}); in CreateFusedComputation()