Home
last modified time | relevance | path

Searched defs:replica_number (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlocal_service.cc183 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal()
190 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc287 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer()
309 StatusOr<int> LocalClient::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal()