Searched refs:number_of_replicas (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 28 se::Platform* platform, int number_of_replicas, in LocalClientOptions() argument
32 number_of_replicas_(number_of_replicas), in LocalClientOptions()
44 int number_of_replicas) { in set_number_of_replicas() argument
45 number_of_replicas_ = number_of_replicas; in set_number_of_replicas()
49 int LocalClientOptions::number_of_replicas() const { in number_of_replicas() function in xla::LocalClientOptions
93 int replica_count = options.number_of_replicas(); in GetOrCreateLocalClient()
|
| D | client_library.h | 49 se::Platform* platform = nullptr, int number_of_replicas = 1,
59 LocalClientOptions& set_number_of_replicas(int number_of_replicas);
60 int number_of_replicas() const;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 106 ServiceOptions& ServiceOptions::set_number_of_replicas(int number_of_replicas) { in set_number_of_replicas() argument
107 number_of_replicas_ = number_of_replicas; in set_number_of_replicas()
111 int ServiceOptions::number_of_replicas() const { return number_of_replicas_; } in number_of_replicas() function in xla::ServiceOptions
162 CHECK_GT(options_.number_of_replicas(), 0); in Service()
165 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas()) in Service()
233 CHECK_EQ(options_.number_of_replicas(), stream_executors.size()); in ResolveAndValidateArguments()
235 replicated_arguments.resize(options_.number_of_replicas()); in ResolveAndValidateArguments()
244 CHECK_EQ(options_.number_of_replicas(), replicated_buffers.size()); in ResolveAndValidateArguments()
245 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in ResolveAndValidateArguments()
309 config->set_replica_count(options_.number_of_replicas()); in CreateModuleConfig()
[all …]
|
| D | service.h | 58 ServiceOptions& set_number_of_replicas(int number_of_replicas);
59 int number_of_replicas() const;
|
| D | local_service.cc | 185 replica_number, /*computation=*/0, options_.number_of_replicas(), in ReplicaNumberToDeviceOrdinal()
|