Searched refs:CopyGPUTensorToCPU (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 37 GPUUtil::CopyGPUTensorToCPU(device, this, device_tensor, cpu_tensor, done); in CopyDeviceTensorToCPU()
|
| D | gpu_util.h | 47 static void CopyGPUTensorToCPU(Device* gpu_device,
|
| D | gpu_util.cc | 255 void GPUUtil::CopyGPUTensorToCPU(Device* gpu_device, in CopyGPUTensorToCPU() function in tensorflow::GPUUtil
399 CopyGPUTensorToCPU(gpu_device, device_context, &tensor, ©, in Checksum()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.cc | 967 GPUUtil::CopyGPUTensorToCPU(gpu_device, device_context, &dummy1, &dummy2, in StreamGPUOp()
1093 GPUUtil::CopyGPUTensorToCPU( in RecvHandler()
|