Searched refs:old_instr (Results 1 – 3 of 3) sorted by relevance
30 Status ReplaceUsesWhileKeepingLoopInvariance(HloInstruction* old_instr, in ReplaceUsesWhileKeepingLoopInvariance() argument37 users.reserve(old_instr->user_count()); in ReplaceUsesWhileKeepingLoopInvariance()38 absl::c_copy(old_instr->users(), std::back_inserter(users)); in ReplaceUsesWhileKeepingLoopInvariance()42 if (user->operand(i) == old_instr && in ReplaceUsesWhileKeepingLoopInvariance()
192 HloInstruction* new_instr, HloInstruction* old_instr,223 StatusOr<HloInstruction*> BatchToSpace(HloInstruction* old_instr);2287 HloInstruction* new_instr, HloInstruction* old_instr, in SelectValidPortion() argument2292 auto old_shape = old_instr->shape(); in SelectValidPortion()2369 HloInstruction* old_instr) { in BatchToSpace() argument2370 if (batch_to_space_map_.count(old_instr)) { in BatchToSpace()2371 CHECK_NE(batch_to_space_map_[old_instr], nullptr); in BatchToSpace()2372 return batch_to_space_map_[old_instr]; in BatchToSpace()2375 auto result = instr_to_dim_map_[old_instr]; in BatchToSpace()2379 const int64_t old_batch_size = old_instr->shape().dimensions(old_batch_dim); in BatchToSpace()[all …]
372 for (HloInstruction* old_instr : def_to_use_order) { in CreateFusedComputation()373 if (old_instr->opcode() == HloOpcode::kParameter) { in CreateFusedComputation()378 const auto& old_opnds = old_instr->operands(); in CreateFusedComputation()385 old_instr->CloneWithNewOperands(old_instr->shape(), new_opnds)); in CreateFusedComputation()386 clone_map.insert({old_instr, new_instr}); in CreateFusedComputation()388 metadata = &old_instr->metadata(); in CreateFusedComputation()