Searched defs:CopyTensorInSameDevice (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 42 void GPUDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::GPUDeviceContext
|
| D | gpu_device_test.cc | 407 TEST_F(GPUDeviceTest, CopyTensorInSameDevice) { in TEST_F() argument
|
| D | gpu_device.cc | 752 void BaseGPUDevice::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::BaseGPUDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 72 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anon6eac52db0111::SingleThreadedCpuDevice
|
| D | threadpool_device.cc | 102 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::ThreadPoolDevice
|
| D | renamed_device.h | 113 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 89 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
261 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
|
| D | rendezvous_test.cc | 403 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device, in CopyTensorInSameDevice() function in tensorflow::__anon404285390111::DummyDeviceContext
|
| D | op_kernel_test.cc | 413 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anonc25ee1c30111::ScopedAllocatorDevice
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::__anon7fd818060111::VirtualDeviceContext
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 106 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext
|