Searched refs:set_replica_count (Results 1 – 4 of 4) sorted by relevance
44 config.set_replica_count(2); in XLA_TEST_F()
80 void set_replica_count(int64 replica_count) { in set_replica_count() function
42 proto->set_replica_count(replica_count()); in Serialize()
307 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()