Searched refs:set_replica_count (Results 1 – 12 of 12) sorted by relevance
94 config.set_replica_count(kNumReplicas); in TestTwoReplicasOneOperand()249 config.set_replica_count(2); in XLA_TEST_F()290 config.set_replica_count(2); in XLA_TEST_F()315 config.set_replica_count(devices.size()); in XLA_TEST_F()357 config.set_replica_count(devices.size()); in XLA_TEST_F()416 config.set_replica_count(2); in XLA_TEST_F()467 config.set_replica_count(kNumReplicas); in XLA_TEST_F()503 config.set_replica_count(4); in XLA_TEST_F()541 config.set_replica_count(kNumReplicas); in XLA_TEST_F()570 config.set_replica_count(kNumReplicas); in XLA_TEST_F()[all …]
141 config.set_replica_count(replica_count);
89 config->set_replica_count(execution_options->num_replicas()); in CreateModuleConfig()91 config->set_replica_count(default_num_replicas); in CreateModuleConfig()105 config->set_replica_count(default_num_replicas); in CreateModuleConfig()
906 config.set_replica_count(*replica_count); in MakeAllReduceComputation()908 config.set_replica_count(ReplicaCount(replica_groups)); in MakeAllReduceComputation()972 config.set_replica_count(ReplicaCount(replica_groups)); in MakeAllToAllComputation()1022 config.set_replica_count(2); in TEST_F()1039 config.set_replica_count(3); in TEST_F()
126 void set_replica_count(int64 replica_count) { in set_replica_count() function
75 proto->set_replica_count(replica_count()); in Serialize()
442 module_config.set_replica_count(execution_options->num_replicas()); in CreateModuleConfigFromShape()
42 config->set_replica_count(ovr_config.num_replicas); in OverrideConfig()
96 dst->set_replica_count(std::max(src.replica_count(), dst->replica_count())); in CombineRunEnvironment()
298 re.set_replica_count(run_environment.replica_count()); in ComputeRunEnvironment()
99 config->set_replica_count(replica_count); in CreateModuleConfig()
459 config.set_replica_count(c_config.replica_count); in FromC()