Home
last modified time | relevance | path

Searched defs:CopyTensorInSameDevice (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util_platform_specific.cc42 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::GPUDeviceContext
Dgpu_device_test.cc519 TEST_F(GPUDeviceTest, CopyTensorInSameDevice) { in TEST_F() argument
Dgpu_device.cc891 void BaseGPUDevice::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::BaseGPUDevice
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_context.cc43 void PluggableDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::PluggableDeviceContext
Dpluggable_device.cc426 void PluggableDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::PluggableDevice
/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_cpu_device.cc72 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anona6fe44880111::SingleThreadedCpuDevice
Drenamed_device.h114 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function
Dthreadpool_device.cc135 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::ThreadPoolDevice
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h90 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
273 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
Drendezvous_test.cc403 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device, in CopyTensorInSameDevice() function in tensorflow::__anon1fc8bce60111::DummyDeviceContext
Dop_kernel_test.cc416 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anon0ef657d00111::ScopedAllocatorDevice
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::__anonb28748530111::VirtualDeviceContext
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc107 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_execute_compat.cc808 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::tfd::__anon4d6e666c0711::DeviceWithCustomAllocator