| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device_context.h | 34 void CopyCPUTensorToDevice(const Tensor *cpu_tensor, Device *device,
|
| D | sycl_device.cc | 70 device_context_->CopyCPUTensorToDevice( in MakeTensorFromProto()
|
| D | sycl_device_context.cc | 25 void SYCLDeviceContext::CopyCPUTensorToDevice(const Tensor *cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::SYCLDeviceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::GPUDeviceContext
|
| D | gpu_device.cc | 678 device_contexts_[0]->CopyCPUTensorToDevice(&from, this, copy, in MaybeCopyTensorToGPU()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 74 ctx->op_device_context()->CopyCPUTensorToDevice( in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 52 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
|
| D | copy_tensor.cc | 85 recv_dev_context->CopyCPUTensorToDevice(&from, dst, to, in CopyHostToDevice()
110 recv_dev_context->CopyCPUTensorToDevice(input, dst, output, in CopyHostToDevice()
|
| D | ring_reducer.cc | 126 op_dev_ctx->CopyCPUTensorToDevice(&group_size_val, col_ctx_->device, in ContinueAfterInputCopy()
|
| D | process_function_library_runtime_test.cc | 140 device_context->CopyCPUTensorToDevice(&cpu_tensor, gpu_device_, in CPUToGPU()
|
| D | ring_gatherer_test.cc | 430 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
|
| D | ring_reducer_test.cc | 454 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
|
| D | hierarchical_tree_broadcaster_test.cc | 618 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 63 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
|
| D | xla_device_context.cc | 106 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::XlaDeviceContext
|
| D | xla_device.cc | 461 device_context->CopyCPUTensorToDevice(&parsed, this, ©, in MakeTensorFromProto()
|
| D | xla_launch_util.cc | 322 ctx->op_device_context()->CopyCPUTensorToDevice( in PopulateOutputs()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 79 virtual void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device, in CopyCPUTensorToDevice() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 112 op_dev_ctx->CopyCPUTensorToDevice( in Run()
|
| D | stack.cc | 307 device_ctxt->CopyCPUTensorToDevice( in ComputeAsync()
|
| D | collective_nccl_reducer_test.cc | 239 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 493 device_context->CopyCPUTensorToDevice(copy, device, tensor, in TensorFromTransportOptions()
|