Searched defs:CopyCPUTensorToDevice (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ | ||
| H A D | gpu_util_platform_specific.cc | 26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::GPUDeviceContext |
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ | ||
| H A D | pluggable_device_context.cc | 27 void PluggableDeviceContext::CopyCPUTensorToDevice( in CopyCPUTensorToDevice() function in tensorflow::PluggableDeviceContext |
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ | ||
| H A D | virtual_device.cc | 38 void VirtualDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::__anon93b5c1350111::VirtualDeviceContext |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ | ||
| H A D | xla_device_context.cc | 114 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice() function in tensorflow::XlaDeviceContext |