| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | virtual_device.cc | 33 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device,
56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::__anon93b5c1350111::VirtualDeviceContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | renamed_device.h | 114 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function
117 underlying_device_->CopyTensorInSameDevice(input_tensor, output_tensor, in CopyTensorInSameDevice()
|
| H A D | threadpool_device_test.cc | 60 device.CopyTensorInSameDevice(&input, &output, device_context, in TEST()
|
| H A D | threadpool_device.h | 42 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor,
|
| 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 | gpu_device_context.h | 75 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device,
|
| H A D | threadpool_device.cc | 135 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::ThreadPoolDevice
|
| /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
|
| H A D | op_kernel.cc | 951 device()->CopyTensorInSameDevice(&tensor, new_tensor.get(), in maybe_set_output_by_allocate_and_copy()
|
| /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
541 device->CopyTensorInSameDevice(&input_tensor, &output_tensor, device_context, in TEST_F()
|
| H A D | gpu_device.h | 100 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor,
|
| 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_context.h | 60 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device,
|
| H A D | pluggable_device.h | 72 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor,
|
| H A D | pluggable_device.cc | 426 void PluggableDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::PluggableDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device_context.h | 71 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device,
|
| 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
811 device_->CopyTensorInSameDevice(input_tensor, output_tensor, device_context, in CopyTensorInSameDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | resource_variable_ops.cc | 112 ctx->op_device_context()->CopyTensorInSameDevice( in CopyVariable()
|