Home
last modified time | relevance | path

Searched refs:CopyCPUTensorToDevice (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc27 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
38 void VirtualDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::__anon7fd818060111::VirtualDeviceContext
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::GPUDeviceContext
Dgpu_device.cc679 device_context_->CopyCPUTensorToDevice( in MaybeCopyTensorToGPU()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc74 ctx->op_device_context()->CopyCPUTensorToDevice( in ComputeAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dgpu_device_context.h62 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
Dcopy_tensor.cc87 recv_dev_context->CopyCPUTensorToDevice( in CopyHostToDevice()
114 recv_dev_context->CopyCPUTensorToDevice(input, dst, output, std::move(done), in CopyHostToDevice()
Dring_reducer.cc144 op_dev_ctx->CopyCPUTensorToDevice( in ContinueAfterInputCopy()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.cc57 CopyCPUTensorToDevice(cpu_tensor, device, device_tensor, in CopyCPUTensorToDeviceSync()
Ddevice_base.h78 virtual void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.h63 void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
Dxla_device_context.cc113 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::XlaDeviceContext
Dxla_launch_util.cc364 ctx->op_device_context()->CopyCPUTensorToDevice( in SetOutputForConstant()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc69 op_dev_ctx->CopyCPUTensorToDevice( in Run()
Dstack.cc308 device_ctxt->CopyCPUTensorToDevice( in ComputeAsync()