Searched refs:module_configs (Results 1 – 5 of 5) sorted by relevance
80 absl::Span<const HloModuleConfig> module_configs) { in CreateFromProto() argument84 TF_RET_CHECK(proto.hlo_modules_size() == module_configs.size()); in CreateFromProto()91 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto()
174 std::vector<std::unique_ptr<HloModuleConfig>> module_configs; in CompileExecutables() local175 module_configs.push_back(std::move(module_config)); in CompileExecutables()182 /*module_protos=*/{&proto}, std::move(module_configs), in CompileExecutables()
80 absl::Span<const HloModuleConfig> module_configs);
287 std::vector<std::unique_ptr<HloModuleConfig>> module_configs, in BuildExecutables() argument305 CHECK_EQ(module_protos.size(), module_configs.size()); in BuildExecutables()310 const HloModuleConfig& config = *module_configs[i]; in BuildExecutables()333 const auto& debug_opts = module_configs[i]->debug_options(); in BuildExecutables()573 std::vector<std::unique_ptr<HloModuleConfig>> module_configs; in ExecuteGraphParallel() local624 module_configs.push_back(std::move(module_config)); in ExecuteGraphParallel()638 BuildExecutables(module_protos, std::move(module_configs), in ExecuteGraphParallel()
245 std::vector<std::unique_ptr<HloModuleConfig>> module_configs,