Searched refs:ConsumeModules (Results 1 – 6 of 6) sorted by relevance
61 std::vector<std::unique_ptr<HloModule>> ConsumeModules();
59 module_group->ConsumeModules(); in Compile()
41 std::vector<std::unique_ptr<HloModule>> HloModuleGroup::ConsumeModules() { in ConsumeModules() function in xla::HloModuleGroup
64 std::vector<std::unique_ptr<HloModule>> modules = group.ConsumeModules(); in TEST_F()
160 auto hlo_modules = module_group->ConsumeModules(); in Compile()
733 module_group->ConsumeModules(); in CompileAheadOfTime()