Searched refs:module_configs (Results 1 – 4 of 4) sorted by relevance
70 absl::Span<const HloModuleConfig> module_configs) { in CreateFromProto() argument74 TF_RET_CHECK(proto.hlo_modules_size() == module_configs.size()); in CreateFromProto()81 HloModule::CreateFromProto(module_proto, module_configs[i])); in CreateFromProto()
348 std::vector<std::unique_ptr<HloModuleConfig>> module_configs, in BuildExecutables() argument366 CHECK_EQ(module_protos.size(), module_configs.size()); in BuildExecutables()371 const HloModuleConfig& config = *module_configs[i]; in BuildExecutables()383 const auto& debug_opts = module_configs[i]->debug_options(); in BuildExecutables()621 std::vector<std::unique_ptr<HloModuleConfig>> module_configs; in ExecuteGraphParallel() local672 module_configs.push_back(std::move(module_config)); in ExecuteGraphParallel()686 BuildExecutables(module_protos, std::move(module_configs), in ExecuteGraphParallel()
71 absl::Span<const HloModuleConfig> module_configs);
243 std::vector<std::unique_ptr<HloModuleConfig>> module_configs,