Searched refs:CopyTensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 34 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf) in RegistrationInfo()
40 CopyTensor::CopyFunction copy_function;
120 void CopyDeviceToDevice(CopyTensor::CopyFunction copy_function, in CopyDeviceToDevice()
200 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA()
297 Status CopyTensor::Register(DeviceType sender_device_type, in Register()
|
| D | copy_tensor.h | 29 class CopyTensor {
|
| D | threadpool_device_test.cc | 49 TEST(ThreadPoolDeviceTest, CopyTensor) { in TEST() argument
|
| D | collective_rma_local.cc | 153 CopyTensor::ViaDMA("", // edge name (non-existent) in MemCpyAsync()
|
| D | rendezvous_mgr.cc | 109 CopyTensor::ViaDMA( in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 143 CopyTensor::ViaDMA("", // edge name (non-existent) in RecvFromPeer()
|
| D | base_rendezvous_mgr.cc | 286 CopyTensor::ViaDMA( in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 250 static CopyTensor::Registration register_gpu_gpu_copy(
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 447 static auto* const register_tpu_tpu_copy = new CopyTensor::Registration( in RegisterTpuDeviceToDeviceCopy()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 1022 tensorflow::CopyTensor::ViaDMA("copy", src_device_context, dst_device_context, in CopyToDevice()
|