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.cc190 StatusOr<int> LocalService::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal()
197 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc424 const GlobalDataHandle& data, int replica_number) { in GlobalDataToShapedBuffer()
444 StatusOr<int> LocalClient::ReplicaNumberToDeviceOrdinal(int replica_number) { in ReplicaNumberToDeviceOrdinal()