Home
last modified time | relevance | path

Searched defs:CopyDeviceTensorToCPU (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc35 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::GPUDeviceContext
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_context.cc34 void PluggableDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::PluggableDeviceContext
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc47 void VirtualDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::__anonb9fa19eb0111::VirtualDeviceContext
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc214 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function in tensorflow::XlaDeviceContext
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h98 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() function