| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device_context.h | 38 void CopyDeviceTensorToCPU(const Tensor *device_tensor, StringPiece edge_name,
|
| D | sycl_device_context.cc | 102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::SYCLDeviceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 33 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::GPUDeviceContext
|
| /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 | 56 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
|
| D | copy_tensor.cc | 150 send_dev_context->CopyDeviceTensorToCPU(&from, edge_name, src, to, in CopyDeviceToHost()
175 send_dev_context->CopyDeviceTensorToCPU(input, edge_name, src, output, in CopyDeviceToHost()
|
| D | hierarchical_tree_broadcaster_test.cc | 439 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
473 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
|
| D | ring_alg.cc | 258 gpu_device_info->default_context->CopyDeviceTensorToCPU( in TensorDebugString()
|
| D | process_function_library_runtime_test.cc | 115 device_context->CopyDeviceTensorToCPU(&device_tensor, "", gpu_device_, in GPUToCPU()
|
| D | ring_gatherer_test.cc | 310 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
|
| D | ring_reducer_test.cc | 330 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 66 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
|
| D | xla_compile_on_demand_op.cc | 146 ctx->op_device_context()->CopyDeviceTensorToCPU( in Compile()
|
| D | xla_device_context.cc | 199 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::XlaDeviceContext
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 95 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 136 send_dev_context->CopyDeviceTensorToCPU( in GrpcRecvTensorAsync()
|
| D | gdr_memory_manager.cc | 407 device_context->CopyDeviceTensorToCPU(&tensor, "" /* tensor_name */, device, in TransportOptionsFromTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 514 send_dev_context->CopyDeviceTensorToCPU( in GrpcRecvTensorAsync()
596 hook->prod_ctx->CopyDeviceTensorToCPU( in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 258 device_ctxt->CopyDeviceTensorToCPU( in ComputeAsync()
|
| D | collective_nccl_reducer_test.cc | 183 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
|
| D | functional_ops.cc | 421 device_ctx->CopyDeviceTensorToCPU( in StartBody()
|