| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 30 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
47 void VirtualDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::__anon10285b640111::VirtualDeviceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 35 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::GPUDeviceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_context.cc | 34 void PluggableDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::PluggableDeviceContext
|
| D | pluggable_device_context.h | 56 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 45 ctx->op_device_context()->CopyDeviceTensorToCPU( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 71 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
|
| D | copy_tensor.cc | 373 send_dev_context->CopyDeviceTensorToCPU(&from, edge_name, src, to, in CopyDeviceToHost()
400 send_dev_context->CopyDeviceTensorToCPU(input, edge_name, src, output, in CopyDeviceToHost()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 43 CopyDeviceTensorToCPU(device_tensor, tensor_name, device, cpu_tensor, in CopyDeviceTensorToCPUSync()
|
| D | device_base.h | 99 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 68 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
|
| D | xla_device_context.cc | 209 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::XlaDeviceContext
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 259 device_ctxt->CopyDeviceTensorToCPU( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 684 hook->prod_ctx->CopyDeviceTensorToCPU( in RecvBufAsync()
|