Home
last modified time | relevance | path

Searched refs:set_replica_count (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmulti_device_all_reduce_test.cc44 config.set_replica_count(2); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h80 void set_replica_count(int64 replica_count) { in set_replica_count() function
Dcomputation_placer.cc42 proto->set_replica_count(replica_count()); in Serialize()
Dservice.cc307 config->set_replica_count(execution_options->num_replicas()); in CreateModuleConfig()
309 config->set_replica_count(options_.number_of_replicas()); in CreateModuleConfig()
314 config->set_replica_count(options_.number_of_replicas()); in CreateModuleConfig()