Searched refs:CopyDeviceTensorToCPU (Results 1 - 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util_platform_specific.cc | 32 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, function in class:tensorflow::GPUDeviceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| H A D | sycl_device_context.h | 38 void CopyDeviceTensorToCPU(const Tensor *device_tensor, StringPiece edge_name,
|
| H A D | sycl_device_context.cc | 102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, function in class:tensorflow::SYCLDeviceContext
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device_context.h | 56 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
77 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
|
| H A D | xla_device_context.cc | 92 void XlaTransferManager::CopyDeviceTensorToCPU(const Tensor* device_tensor, function in class:tensorflow::XlaTransferManager
98 VLOG(2) << "CopyDeviceTensorToCPU "
124 VLOG(2) << "CopyDeviceTensorToCPU empty tensor";
137 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, function in class:tensorflow::XlaDeviceContext
141 manager_.CopyDeviceTensorToCPU(device_tensor, tensor_name, device, cpu_tensor,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | gpu_device_context.h | 59 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
|
| H A D | copy_tensor.cc | 137 send_dev_context->CopyDeviceTensorToCPU(&from, edge_name, src, to,
161 send_dev_context->CopyDeviceTensorToCPU(input, edge_name, src, output,
|
| /external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_gateway.cc | 97 device_ctxt->CopyDeviceTensorToCPU(
|
| /external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.h | 87 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, function in class:tensorflow::DeviceContext
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | stack_ops.cc | 297 device_ctxt->CopyDeviceTensorToCPU(
|
Completed in 481 milliseconds