| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_test.cc | 84 EXPECT_THAT(module->MakeComputationPostOrder(), in TEST_F()
95 EXPECT_THAT(module->MakeComputationPostOrder(), in TEST_F()
120 auto post_order = module->MakeComputationPostOrder(); in TEST_F()
122 auto post_order_copied = cloned_module->MakeComputationPostOrder(); in TEST_F()
162 auto post_order = module->MakeComputationPostOrder(); in TEST_F()
164 auto post_order_copied = cloned_module->MakeComputationPostOrder(); in TEST_F()
292 auto post_order = module->MakeComputationPostOrder(); in TEST_F()
532 module->MakeComputationPostOrder(/*execution_threads=*/{}, allowList), in TEST_F()
542 module->MakeComputationPostOrder(/*execution_threads=*/{}, allowList); in TEST_F()
544 module->MakeComputationPostOrder(/*execution_threads=*/{}, allowList), in TEST_F()
[all …]
|
| D | hlo_dce.cc | 142 module->MakeComputationPostOrder(execution_threads)) { in RecursivelyRemoveDeadComputations()
153 module->MakeComputationPostOrder(execution_threads)) { in RecursivelyRemoveDeadComputations()
178 module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | hlo_dce_test.cc | 391 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()
398 EXPECT_EQ(module->MakeComputationPostOrder().size(), 1); in TEST_F()
438 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()
445 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()
498 EXPECT_EQ(module->MakeComputationPostOrder().size(), 4); in TEST_F()
507 EXPECT_EQ(module->MakeComputationPostOrder().size(), 1); in TEST_F()
|
| D | hlo_module.h | 263 std::vector<HloComputation*> MakeComputationPostOrder() const { in MakeComputationPostOrder() function
264 return MakeComputationPostOrder({}); in MakeComputationPostOrder()
269 std::vector<HloComputation*> MakeComputationPostOrder(
275 std::vector<HloComputation*> MakeComputationPostOrder(
|
| D | hlo_module.cc | 280 : MakeComputationPostOrder(); in ToCord()
311 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto()
714 std::vector<HloComputation*> HloModule::MakeComputationPostOrder( in MakeComputationPostOrder() function in xla::HloModule
718 auto post_order = this->MakeComputationPostOrder(execution_threads); in MakeComputationPostOrder()
731 std::vector<HloComputation*> HloModule::MakeComputationPostOrder( in MakeComputationPostOrder() function in xla::HloModule
837 MakeComputationPostOrder(execution_threads); in MakeComputationSorted()
847 MakeComputationPostOrder(execution_threads); in MakeNonfusionComputations()
|
| D | simplify_fp_conversions.cc | 62 module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | bfloat16_propagation.cc | 699 module->MakeComputationPostOrder(execution_threads); in ResolveInconsistencyOfAliasingBuffers()
738 for (auto computation : module->MakeComputationPostOrder(execution_threads)) { in ResolveInconsistentFusions()
786 for (auto computation : module->MakeComputationPostOrder(execution_threads)) { in ResolveConvertedConstants()
845 module->MakeComputationPostOrder(execution_threads); in Run()
|
| D | hlo_execution_profile.cc | 35 for (xla::HloComputation* computation : module.MakeComputationPostOrder()) { in HloProfileIndexMap()
|
| D | while_loop_invariant_code_motion.cc | 327 for (auto* comp : module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | hlo_memory_scheduler.cc | 539 module->MakeComputationPostOrder(execution_threads)) { in ComputationSchedulerToModuleScheduler()
778 module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | hlo_module_group_metadata.cc | 129 for (HloComputation* computation : module->MakeComputationPostOrder()) { in Build()
|
| D | bfloat16_normalization.cc | 530 for (auto* comp : module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | dynamic_padder.cc | 2165 module->MakeComputationPostOrder(execution_threads); in Run()
2296 computations = module->MakeComputationPostOrder(execution_threads); in Run()
|
| D | while_loop_concat_code_motion.cc | 1029 module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | conditional_code_motion.cc | 1418 for (auto* comp : module->MakeComputationPostOrder(execution_threads)) { in Run()
|
| D | hlo_dataflow_analysis.cc | 1210 auto comps = module_.MakeComputationPostOrder(); in Propagate()
|
| D | buffer_assignment.cc | 200 for (auto* computation : module->MakeComputationPostOrder()) { in GatherComputationsByAllocationType()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 87 for (auto computation : old_module_.MakeComputationPostOrder()) { in FinishVisit()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 215 base_ptrs_.begin()->first->GetModule()->MakeComputationPostOrder()) { in ToString()
|