Searched defs:peer_device (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 27 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer()
98 const string& peer_device, const string& peer_task, const string& key, in PostToPeer()
|
| D | collective_test_util.cc | 59 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer()
71 void FailTestRMA::PostToPeer(const string& peer_device, const string& peer_task, in PostToPeer()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | CUDAContext.cpp | 54 bool canDeviceAccessPeer(c10::DeviceIndex device, c10::DeviceIndex peer_device) { in canDeviceAccessPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 41 RecvBufCall(int64_t step_id, const string& peer_device, in RecvBufCall()
109 const string& peer_device, const string& peer_task, bool peer_is_local, in RecvFromPeer()
|
| /external/pytorch/torch/csrc/cuda/ |
| D | Module.cpp | 154 int64_t peer_device = THPUtils_unpackLong(arg2); in THCPModule_canDeviceAccessPeer_wrap() local
|