| /external/tensorflow/tensorflow/compiler/xla/client/ |
| client_library.cc | 28 se::Platform* platform, int number_of_replicas,
32 number_of_replicas_(number_of_replicas),
44 int number_of_replicas) {
45 number_of_replicas_ = number_of_replicas;
49 int LocalClientOptions::number_of_replicas() const { function in class:xla::LocalClientOptions
93 int replica_count = options.number_of_replicas();
|
| 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/ |
| service.cc | 106 ServiceOptions& ServiceOptions::set_number_of_replicas(int number_of_replicas) {
107 number_of_replicas_ = number_of_replicas;
111 int ServiceOptions::number_of_replicas() const { return number_of_replicas_; } function in class:xla::ServiceOptions
162 CHECK_GT(options_.number_of_replicas(), 0);
165 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas())
233 CHECK_EQ(options_.number_of_replicas(), stream_executors.size());
235 replicated_arguments.resize(options_.number_of_replicas());
244 CHECK_EQ(options_.number_of_replicas(), replicated_buffers.size());
245 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) {
309 config->set_replica_count(options_.number_of_replicas());
[all...] |
| service.h | 58 ServiceOptions& set_number_of_replicas(int number_of_replicas);
59 int number_of_replicas() const;
|
| local_service.cc | 185 replica_number, /*computation=*/0, options_.number_of_replicas(),
|
