Searched refs:CopyTensor (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 35 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf, in RegistrationInfo()
43 CopyTensor::CopyFunction copy_function;
122 void CopyDeviceToDevice(CopyTensor::CopyFunction copy_function, in CopyDeviceToDevice()
202 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA()
304 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 | 112 CopyTensor::ViaDMA( in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_plugin_init.cc | 57 TF_RETURN_IF_ERROR(CopyTensor::Register( in InitDeviceAndGraphModule()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 195 CopyTensor::ViaDMA("", // edge name (non-existent) in RecvFromPeer()
|
| D | base_rendezvous_mgr.cc | 287 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 | 450 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()
|