Searched defs:CopyTensorInSameDevice (Results 1 – 10 of 10) 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 | 309 TEST_F(GPUDeviceTest, CopyTensorInSameDevice) { in TEST_F() argument
|
| D | gpu_device.cc | 695 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::__anon9ec7ae7f0111::SingleThreadedCpuDevice
|
| D | threadpool_device.cc | 106 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::ThreadPoolDevice
|
| D | renamed_device.h | 123 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 88 virtual void CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function
271 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::__anonc00318dd0111::DummyDeviceContext
|
| D | op_kernel_test.cc | 469 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anonf4c990e70111::ScopedAllocatorDevice
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 104 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext
|