Searched refs:module_protos (Results 1 – 2 of 2) sorted by relevance
347 const std::vector<const HloModuleProto*>& module_protos, in BuildExecutables() argument355 for (int64 i = 0; i < module_protos.size(); ++i) { in BuildExecutables()357 *hlo_snapshot->mutable_hlo()->mutable_hlo_module() = *module_protos[i]; in BuildExecutables()362 for (const HloModuleProto* proto : module_protos) { in BuildExecutables()366 CHECK_EQ(module_protos.size(), module_configs.size()); in BuildExecutables()368 absl::make_unique<HloModuleGroup>(module_protos[0]->name()); in BuildExecutables()369 for (int64 i = 0; i < module_protos.size(); ++i) { in BuildExecutables()370 const HloModuleProto* proto = module_protos[i]; in BuildExecutables()382 for (size_t i = 0; i < module_protos.size(); ++i) { in BuildExecutables()384 if (DumpingEnabledForHloModule(module_protos[i]->name(), debug_opts) && in BuildExecutables()[all …]
242 const std::vector<const HloModuleProto*>& module_protos,