Searched refs:number_of_replicas (Results 1 – 6 of 6) 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 | 107 ServiceOptions& ServiceOptions::set_number_of_replicas(int number_of_replicas) { in set_number_of_replicas() argument
108 number_of_replicas_ = number_of_replicas; in set_number_of_replicas()
112 int ServiceOptions::number_of_replicas() const { return number_of_replicas_; } in number_of_replicas() function in xla::ServiceOptions
163 CHECK_GT(options_.number_of_replicas(), 0); in Service()
166 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas()) in Service()
235 CHECK_EQ(options_.number_of_replicas(), stream_executors.size()); in ResolveAndValidateArguments()
237 replicated_arguments.resize(options_.number_of_replicas()); in ResolveAndValidateArguments()
246 CHECK_EQ(options_.number_of_replicas(), replicated_buffers.size()); in ResolveAndValidateArguments()
247 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in ResolveAndValidateArguments()
260 int default_num_replicas = 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 | 192 replica_number, /*computation=*/0, options_.number_of_replicas(), in ReplicaNumberToDeviceOrdinal()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_feed.py | 855 number_of_replicas = len(sharded_inputs)
861 for replica_index in range(number_of_replicas):
|