Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation.cc68 for (auto inst_it = insts.rbegin(); inst_it != insts.rend(); ++inst_it) { in DetermineFusionComputationPrecision() local
69 DetermineInstructionPrecision(*inst_it, /*skip_parameters=*/false); in DetermineFusionComputationPrecision()
192 for (auto inst_it = body_insts.rbegin(); inst_it != body_insts.rend(); in DetermineWhileComputationsPrecision() local
193 ++inst_it) { in DetermineWhileComputationsPrecision()
194 DetermineInstructionPrecision(*inst_it, /*skip_parameters=*/false); in DetermineWhileComputationsPrecision()
199 for (auto inst_it = condition_insts.rbegin(); in DetermineWhileComputationsPrecision() local
200 inst_it != condition_insts.rend(); ++inst_it) { in DetermineWhileComputationsPrecision()
201 DetermineInstructionPrecision(*inst_it, /*skip_parameters=*/false); in DetermineWhileComputationsPrecision()
508 for (auto inst_it = insts.rbegin(); inst_it != insts.rend(); ++inst_it) { in ResolveInconsistencyOfAliasingBuffersHelper() local
509 auto hlo = *inst_it; in ResolveInconsistencyOfAliasingBuffersHelper()
[all …]
Dhlo_cse.cc55 auto inst_it = computation->instructions().begin(); in CombineConstants() local
56 while (inst_it != computation->instructions().end()) { in CombineConstants()
57 HloInstruction* instruction = *inst_it; in CombineConstants()
61 ++inst_it; in CombineConstants()
Dhlo_computation.cc267 auto inst_it = instruction_iterators_.find(instruction); in RemoveInstruction() local
268 TF_RET_CHECK(inst_it != instruction_iterators_.end()); in RemoveInstruction()
269 (*inst_it->second)->set_parent(nullptr); in RemoveInstruction()
270 instructions_.erase(inst_it->second); in RemoveInstruction()
271 instruction_iterators_.erase(inst_it); in RemoveInstruction()