Home
last modified time | relevance | path

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

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