Home
last modified time | relevance | path

Searched refs:MakeInstructionPostOrder (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation.cc67 fusion->fused_instructions_computation()->MakeInstructionPostOrder(); in DetermineFusionComputationPrecision()
123 fusion->fused_instructions_computation()->MakeInstructionPostOrder()) { in RevertIfFusionInternalBF16Changes()
191 auto body_insts = body->MakeInstructionPostOrder(); in DetermineWhileComputationsPrecision()
198 auto condition_insts = condition->MakeInstructionPostOrder(); in DetermineWhileComputationsPrecision()
505 auto insts = computation->MakeInstructionPostOrder(); in ResolveInconsistencyOfAliasingBuffersHelper()
627 auto insts = computation->MakeInstructionPostOrder(); in ResolveInconsistentFusions()
673 for (auto hlo : computation->MakeInstructionPostOrder()) { in ResolveConvertedConstants()
691 for (auto hlo : computation->MakeInstructionPostOrder()) { in SkipNoopConversions()
733 for (auto inst : computation->MakeInstructionPostOrder()) { in Run()
762 for (auto inst : computation->MakeInstructionPostOrder()) { in Run()
[all …]
Dhlo_computation_test.cc131 EXPECT_THAT(computation->MakeInstructionPostOrder(), ElementsAre(constant)); in TEST_F()
146 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
164 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
182 EXPECT_THAT(computation->MakeInstructionPostOrder(), in TEST_F()
204 auto post_order = computation->MakeInstructionPostOrder(); in TEST_F()
432 auto instructions = computation->MakeInstructionPostOrder(); in TEST_F()
699 EXPECT_THAT(module->entry_computation()->MakeInstructionPostOrder(), in TEST_F()
Dwhile_loop_invariant_code_motion.cc192 for (auto* instruction : while_body->MakeInstructionPostOrder()) { in TryHoistingInvariantInstructionsFromWhileBody()
203 for (auto* instruction : while_body->MakeInstructionPostOrder()) { in TryHoistingInvariantInstructionsFromWhileBody()
Dzero_sized_hlo_elimination.cc33 for (HloInstruction* instruction : comp->MakeInstructionPostOrder()) { in Run()
Dhlo_domain_isolator.cc34 computation->MakeInstructionPostOrder()) { in RunInternal()
Dhlo_constant_folding.cc77 for (auto instruction : computation->MakeInstructionPostOrder()) { in Run()
Ddefuser.cc66 fused_computation->MakeInstructionPostOrder()) { in Defuse()
Ddynamic_index_splitter.cc38 for (HloInstruction* dynamic_op : computation->MakeInstructionPostOrder()) { in Run()
Dhlo_domain_remover.cc77 computation->MakeInstructionPostOrder()) { in Run()
Dhlo_cse.cc143 for (auto instruction : computation->MakeInstructionPostOrder()) { in Run()
Dhlo_graph_dumper_test.cc71 ->MakeInstructionPostOrder()) { in TEST_F()
Dhlo_reachability.cc80 const auto& all = computation->MakeInstructionPostOrder(); in Build()
Dinstruction_fusion.cc330 post_order_ = computation->MakeInstructionPostOrder(); in ReversePostOrderFusionQueue()
464 ComputeGloballyUnfusible(computation_->MakeInstructionPostOrder()); in Run()
Dhlo_memory_scheduler.cc430 for (const HloInstruction* hlo : computation->MakeInstructionPostOrder()) { in DFSMemoryScheduler()
506 return HloInstructionSequence(computation->MakeInstructionPostOrder()); in PostOrderMemoryScheduler()
Dstable_sort_expander.cc132 comparator->MakeInstructionPostOrder(); in ExpandInstruction()
Dhlo_element_type_converter.cc125 for (auto* hlo : computation->MakeInstructionPostOrder()) { in Run()
Dhlo_domain_map.cc84 for (HloInstruction* instruction : computation->MakeInstructionPostOrder()) { in Populate()
Dhlo_computation.cc420 std::vector<HloInstruction*> HloComputation::MakeInstructionPostOrder() const { in MakeInstructionPostOrder() function in xla::HloComputation
469 return ToString(options, MakeInstructionPostOrder()); in ToString()
525 for (const HloInstruction* instruction : MakeInstructionPostOrder()) { in ToProto()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_assignment.cc135 for (const auto* hlo : computation.MakeInstructionPostOrder()) { in AssignStreams()
Dgpu_hlo_schedule.cc80 module->entry_computation()->MakeInstructionPostOrder()); in GpuHloOrdering()
Dfusion_merger.cc159 for (auto* instruction : computation_->MakeInstructionPostOrder()) { in Run()
Dmulti_output_fusion.cc133 for (HloInstruction* consumer : computation()->MakeInstructionPostOrder()) { in DoProducerConsumerMultiOutputFusion()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc86 for (auto old_instruction : computation->MakeInstructionPostOrder()) { in FinishVisit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc34 module->entry_computation()->MakeInstructionPostOrder()) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.cc229 fusion->fused_instructions_computation()->MakeInstructionPostOrder(); in IsFusedIrEmitterInefficient()

12