| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.h | 90 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
273 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
|
| H A D | rendezvous_test.cc | 403 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device, in CopyTensorInSameDevice() function in tensorflow::__anon8f1e23880111::DummyDeviceContext
|
| H A D | op_kernel_test.cc | 416 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anondc0872320111::ScopedAllocatorDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util_platform_specific.cc | 42 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::GPUDeviceContext
|
| H A D | gpu_device_test.cc | 519 TEST_F(GPUDeviceTest, CopyTensorInSameDevice) { in TEST_F() argument
|
| H A D | gpu_device.cc | 891 void BaseGPUDevice::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::BaseGPUDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_context.cc | 43 void PluggableDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::PluggableDeviceContext
|
| H A D | pluggable_device.cc | 426 void PluggableDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::PluggableDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | single_threaded_cpu_device.cc | 72 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anonde40af2a0111::SingleThreadedCpuDevice
|
| H A D | renamed_device.h | 114 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function
|
| H A D | threadpool_device.cc | 135 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::ThreadPoolDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | virtual_device.cc | 56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::__anon93b5c1350111::VirtualDeviceContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device_context.cc | 107 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_execute_compat.cc | 808 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::tfd::__anond671884e0711::DeviceWithCustomAllocator
|