Searched refs:MakeComputationPostOrder (Results 1 – 10 of 10) sorted by relevance
64 EXPECT_THAT(module->MakeComputationPostOrder(), in TEST_F()75 EXPECT_THAT(module->MakeComputationPostOrder(), in TEST_F()96 auto post_order = module->MakeComputationPostOrder(); in TEST_F()98 auto post_order_copied = cloned_module->MakeComputationPostOrder(); in TEST_F()132 auto post_order = module->MakeComputationPostOrder(); in TEST_F()134 auto post_order_copied = cloned_module->MakeComputationPostOrder(); in TEST_F()162 auto post_order = module->MakeComputationPostOrder(); in TEST_F()
79 for (auto* computation : module->MakeComputationPostOrder()) { in Run()88 std::list<HloComputation*> computations = module->MakeComputationPostOrder(); in Run()
353 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()360 EXPECT_EQ(module->MakeComputationPostOrder().size(), 1); in TEST_F()400 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()407 EXPECT_EQ(module->MakeComputationPostOrder().size(), 2); in TEST_F()
196 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToString()209 for (const HloComputation* computation : MakeComputationPostOrder()) { in ToProto()487 std::list<HloComputation*> HloModule::MakeComputationPostOrder() const { in MakeComputationPostOrder() function in xla::HloModule
31 for (xla::HloComputation* computation : module.MakeComputationPostOrder()) { in HloProfileIndexMap()
143 std::list<HloComputation*> MakeComputationPostOrder() const;
341 for (auto* comp : module->MakeComputationPostOrder()) { in Run()
621 for (auto* computation : module->MakeComputationPostOrder()) { in GatherComputationsByAllocationType()1277 for (const HloComputation* computation : module->MakeComputationPostOrder()) { in BuildColocatedBufferSets()
1535 module->MakeComputationPostOrder(); in Run()1536 for (auto* computation : module->MakeComputationPostOrder()) { in Run()
249 base_ptrs_.begin()->first->GetModule()->MakeComputationPostOrder()) { in ToString()