Home
last modified time | relevance | path

Searched defs:CopyCPUTensorToDevice (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device_context.cc25 void SYCLDeviceContext::CopyCPUTensorToDevice(const Tensor *cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::SYCLDeviceContext
/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
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc106 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::XlaDeviceContext
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h79 virtual void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device, in CopyCPUTensorToDevice() function