Searched refs:hlo_modules (Results 1 – 4 of 4) sorted by relevance
69 std::vector<std::unique_ptr<HloModule>> hlo_modules; in CompileAheadOfTime() local92 hlo_modules.push_back(std::move(hlo_module)); in CompileAheadOfTime()96 absl::make_unique<HloModuleGroup>(hlo_modules[0]->name(), in CompileAheadOfTime()97 absl::MakeSpan(hlo_modules)), in CompileAheadOfTime()
78 const HloModuleProto& module_proto = proto.hlo_modules(i); in CreateFromProto()
430 repeated HloModuleProto hlo_modules = 2; field
160 auto hlo_modules = module_group->ConsumeModules(); in Compile() local162 RunHloPasses(std::move(hlo_modules[0]), stream_exec[0][0], in Compile()