Searched defs:replica_count (Results 1 – 8 of 8) sorted by relevance
41 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment()47 int replica_count() const { return height(); } in replica_count() function
80 void set_replica_count(int64 replica_count) { in set_replica_count()83 int64 replica_count() const { return replica_count_; } in replica_count() function
92 int replica_count, in DeviceId()101 int replica_count, int computation_count) { in AssignDevices()
762 const int64 replica_count = options_.number_of_replicas(); in GetDeviceHandles() local1011 const int64 replica_count = options_.number_of_replicas(); in TransferToInfeed() local1040 const int64 replica_count = options_.number_of_replicas(); in TransferFromOutfeed() local
48 int64 replica_count; member345 int64 replica_count, int64 element_count, in NcclAllReduceThunk()
93 int replica_count = options.number_of_replicas(); in GetOrCreateLocalClient() local
343 int32 replica_count = 1; field
1828 def initialize_replica_count(replica_count): argument