| /external/tensorflow/tensorflow/core/tpu/ |
| 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::__anon7fd818060111::VirtualDeviceContext
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 113 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function
116 underlying_device_->CopyTensorInSameDevice(input_tensor, output_tensor, in CopyTensorInSameDevice()
|
| D | threadpool_device.h | 41 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor,
|
| D | threadpool_device_test.cc | 60 device.CopyTensorInSameDevice(&input, &output, device_context, in TEST()
|
| D | single_threaded_cpu_device.cc | 72 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice() function in tensorflow::__anon6eac52db0111::SingleThreadedCpuDevice
|
| D | gpu_device_context.h | 70 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device,
|
| D | threadpool_device.cc | 102 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice() function in tensorflow::ThreadPoolDevice
|
| /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
|
| D | op_kernel.cc | 940 device()->CopyTensorInSameDevice(&tensor, new_tensor.get(), in maybe_set_output_by_allocate_and_copy()
|
| /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
429 device->CopyTensorInSameDevice(&input_tensor, &output_tensor, device_context, in TEST_F()
|
| D | gpu_device.h | 76 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor,
|
| D | gpu_device.cc | 752 void BaseGPUDevice::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::BaseGPUDevice
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 69 void CopyTensorInSameDevice(const Tensor* input_tensor, Device* device,
|
| D | xla_device_context.cc | 106 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice() function in tensorflow::XlaDeviceContext
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | resource_variable_ops.cc | 109 ctx->op_device_context()->CopyTensorInSameDevice( in CopyVariable()
|