HomeSort by relevance Sort by last modified time
    Searched defs:number_of_replicas (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/
client_library.cc 27 int number_of_replicas,
30 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
81 int replica_count = options.number_of_replicas();
  /external/tensorflow/tensorflow/compiler/xla/service/
service.cc 108 ServiceOptions& ServiceOptions::set_number_of_replicas(int number_of_replicas) {
109 number_of_replicas_ = number_of_replicas;
113 int ServiceOptions::number_of_replicas() const { return number_of_replicas_; } function in class:xla::ServiceOptions
153 CHECK_GT(options_.number_of_replicas(), 0);
156 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas())
306 config->set_replica_count(options_.number_of_replicas());
511 options_.number_of_replicas(), executables.size()));
647 options_.number_of_replicas(),
666 if (options_.number_of_replicas() == 1) {
672 repeated_arguments(options_.number_of_replicas(), arguments)
    [all...]

Completed in 108 milliseconds