Searched refs:replica_number (Results 1 – 4 of 4) sorted by relevance
207 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument209 replica_number, /*computation=*/0, options_.number_of_replicas(), in ReplicaNumberToDeviceOrdinal()214 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument216 if (replica_number >= buffers.size()) { in GlobalDataToShapedBuffer()219 replica_number, buffers.size()); in GlobalDataToShapedBuffer()221 return buffers[replica_number]; in GlobalDataToShapedBuffer()
60 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);65 const GlobalDataHandle& data, int replica_number);
144 const GlobalDataHandle& data, int replica_number);165 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);
391 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument392 return local_service_->GlobalDataToShapedBuffer(data, replica_number); in GlobalDataToShapedBuffer()413 StatusOr<int> LocalClient::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument414 return local_service_->ReplicaNumberToDeviceOrdinal(replica_number); in ReplicaNumberToDeviceOrdinal()