Home
last modified time | relevance | path

Searched refs:replica_number (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlocal_service.cc207 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument
209 replica_number, /*computation=*/0, options_.number_of_replicas(), in ReplicaNumberToDeviceOrdinal()
214 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument
216 if (replica_number >= buffers.size()) { in GlobalDataToShapedBuffer()
219 replica_number, buffers.size()); in GlobalDataToShapedBuffer()
221 return buffers[replica_number]; in GlobalDataToShapedBuffer()
Dlocal_service.h60 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);
65 const GlobalDataHandle& data, int replica_number);
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.h144 const GlobalDataHandle& data, int replica_number);
165 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);
Dlocal_client.cc391 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer() argument
392 return local_service_->GlobalDataToShapedBuffer(data, replica_number); in GlobalDataToShapedBuffer()
413 StatusOr<int> LocalClient::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal() argument
414 return local_service_->ReplicaNumberToDeviceOrdinal(replica_number); in ReplicaNumberToDeviceOrdinal()