Searched refs:replica_number (Results 1 – 4 of 4) sorted by relevance
183 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument185 replica_number, /*computation=*/0, options_.number_of_replicas(), in ReplicaNumberToDeviceOrdinal()190 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument192 if (replica_number >= buffers.size()) { in GlobalDataToShapedBuffer()195 replica_number, buffers.size()); in GlobalDataToShapedBuffer()197 return buffers[replica_number]; in GlobalDataToShapedBuffer()
59 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);64 const GlobalDataHandle& data, int replica_number);
287 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument288 return local_service_->GlobalDataToShapedBuffer(data, replica_number); in GlobalDataToShapedBuffer()309 StatusOr<int> LocalClient::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument310 return local_service_->ReplicaNumberToDeviceOrdinal(replica_number); in ReplicaNumberToDeviceOrdinal()
143 const GlobalDataHandle& data, int replica_number);164 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);