Searched refs:replica_number (Results 1 – 4 of 4) sorted by relevance
190 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument192 replica_number, /*computation=*/0, options_.number_of_replicas(), in ReplicaNumberToDeviceOrdinal()197 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument199 if (replica_number >= buffers.size()) { in GlobalDataToShapedBuffer()202 replica_number, buffers.size()); in GlobalDataToShapedBuffer()204 return buffers[replica_number]; in GlobalDataToShapedBuffer()
60 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);65 const GlobalDataHandle& data, int replica_number);
167 const GlobalDataHandle& data, int replica_number);188 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);
424 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument425 return local_service_->GlobalDataToShapedBuffer(data, replica_number); in GlobalDataToShapedBuffer()444 StatusOr<int> LocalClient::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument445 return local_service_->ReplicaNumberToDeviceOrdinal(replica_number); in ReplicaNumberToDeviceOrdinal()