Searched refs:MakeComputationPostOrder (Results 1 – 14 of 14) sorted by relevance
69 EXPECT_THAT(module->MakeComputationPostOrder(), in TEST_F()80 EXPECT_THAT(module->MakeComputationPostOrder(), in TEST_F()101 auto post_order = module->MakeComputationPostOrder(); in TEST_F()103 auto post_order_copied = cloned_module->MakeComputationPostOrder(); in TEST_F()137 auto post_order = module->MakeComputationPostOrder(); in TEST_F()139 auto post_order_copied = cloned_module->MakeComputationPostOrder(); in TEST_F()167 auto post_order = module->MakeComputationPostOrder(); in TEST_F()
44 for (auto* computation : module->MakeComputationPostOrder()) { in Run()73 for (auto* computation : module->MakeComputationPostOrder()) { in Run()82 for (auto* computation : module->MakeComputationPostOrder()) { in Run()
355 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()362 EXPECT_EQ(module->MakeComputationPostOrder().size(), 1); in TEST_F()402 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()409 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()
589 module->MakeComputationPostOrder(); in ResolveInconsistencyOfAliasingBuffers()626 for (auto computation : module->MakeComputationPostOrder()) { in ResolveInconsistentFusions()672 for (auto computation : module->MakeComputationPostOrder()) { in ResolveConvertedConstants()724 auto computations_topological_order = module->MakeComputationPostOrder(); in Run()
208 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToString()229 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto()499 std::vector<HloComputation*> HloModule::MakeComputationPostOrder() const { in MakeComputationPostOrder() function in xla::HloModule
34 for (xla::HloComputation* computation : module.MakeComputationPostOrder()) { in HloProfileIndexMap()
182 std::vector<HloComputation*> MakeComputationPostOrder() const;
577 for (auto* computation : module->MakeComputationPostOrder()) { in ScheduleModule()623 for (HloComputation* computation : module->MakeComputationPostOrder()) { in Run()
403 for (auto* comp : module->MakeComputationPostOrder()) { in Run()
125 for (HloComputation* computation : module->MakeComputationPostOrder()) { in Build()
209 for (auto* computation : module->MakeComputationPostOrder()) { in GatherComputationsByAllocationType()1554 for (const HloComputation* computation : module->MakeComputationPostOrder()) { in BuildColocatedBufferSets()
1952 for (auto* computation : module->MakeComputationPostOrder()) { in Run()
85 for (auto computation : old_module_.MakeComputationPostOrder()) { in FinishVisit()
266 base_ptrs_.begin()->first->GetModule()->MakeComputationPostOrder()) { in ToString()