| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 27 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
38 void VirtualDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::__anoncc35abb50111::VirtualDeviceContext
|
| /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 | 817 device_context_->CopyCPUTensorToDevice( in MaybeCopyTensorToGPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_context.cc | 27 void PluggableDeviceContext::CopyCPUTensorToDevice( in CopyCPUTensorToDevice() function in tensorflow::PluggableDeviceContext
|
| D | pluggable_device_context.h | 52 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
|
| D | pluggable_device.cc | 356 device_context_->CopyCPUTensorToDevice( in MaybeCopyTensorToPluggableDevice()
|
| /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 | 67 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
|
| D | copy_tensor.cc | 90 recv_dev_context->CopyCPUTensorToDevice(&from, dst, to, wrapped_done, in CopyHostToDevice()
117 recv_dev_context->CopyCPUTensorToDevice(input, dst, output, std::move(done), in CopyHostToDevice()
|
| D | ring_reducer.cc | 144 op_dev_ctx->CopyCPUTensorToDevice( in ContinueAfterInputCopy()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 57 CopyCPUTensorToDevice(cpu_tensor, device, device_tensor, in CopyCPUTensorToDeviceSync()
|
| D | device_base.h | 79 virtual void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 69 op_dev_ctx->CopyCPUTensorToDevice( in Run()
|
| D | stack.cc | 308 device_ctxt->CopyCPUTensorToDevice( in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 65 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
|
| D | xla_device_context.cc | 114 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::XlaDeviceContext
|
| D | xla_launch_util.cc | 399 ctx->op_device_context()->CopyCPUTensorToDevice( in SetOutputForConstant()
|