diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/local_service.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/local_service.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/local_service.cc b/tensorflow/compiler/xla/service/local_service.cc index 131c2ee87b..748124ebc7 100644 --- a/tensorflow/compiler/xla/service/local_service.cc +++ b/tensorflow/compiler/xla/service/local_service.cc @@ -152,7 +152,7 @@ StatusOr<std::unique_ptr<Executable>> LocalService::CompileExecutable( // Construct computation layout from the argument layouts. auto module_config = MakeUnique<HloModuleConfig>(*program_shape); module_config->set_has_hybrid_result(has_hybrid_result); - module_config->set_replica_count(execute_backend_->Replicas().size()); + module_config->set_replica_count(execute_backend_->replica_count()); legacy_flags::ServiceFlags* flags = legacy_flags::GetServiceFlags(); if (flags->xla_hlo_profile) { module_config->enable_hlo_profiling(true); |