| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
37 Device* device, Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| D | gpu_device_test.cc | 92 void InitCPUTensor(Tensor* cpu_tensor, int num_elements, float value) { in InitCPUTensor()
99 void CopyCPUToGPU(Tensor* cpu_tensor, Tensor* gpu_tensor, Device* device, in CopyCPUToGPU()
105 void CopyGPUToCPU(Tensor* gpu_tensor, Tensor* cpu_tensor, Device* device, in CopyGPUToCPU()
412 Tensor cpu_tensor(cpu_allocator(), DT_FLOAT, TensorShape({kNumElements})); in TEST_F() local
|
| D | gpu_util.cc | 256 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU()
299 void GPUUtil::CopyCPUTensorToGPU(const Tensor* cpu_tensor, in CopyCPUTensorToGPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_context.cc | 28 const Tensor* cpu_tensor, Device* device, Tensor* device_tensor, in CopyCPUTensorToDevice()
37 Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| D | pluggable_device_util.cc | 176 const Tensor* device_tensor, Tensor* cpu_tensor, StatusCallback done) { in CopyPluggableDeviceTensorToCPU()
222 const Tensor* cpu_tensor, const DeviceContext* device_context, in CopyCPUTensorToPluggableDevice()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 38 void VirtualDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
50 Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 40 Tensor* cpu_tensor) { in CopyDeviceTensorToCPUSync()
52 Status DeviceContext::CopyCPUTensorToDeviceSync(const Tensor* cpu_tensor, in CopyCPUTensorToDeviceSync()
|
| D | device_base.h | 100 Tensor* cpu_tensor, StatusCallback done) { in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 71 Tensor* cpu_tensor) { in PopulateTensorFromExtra()
81 Tensor* cpu_tensor) { in PopulateTensorFromResponse()
126 std::unique_ptr<Tensor> cpu_tensor; in RecvFromPeer() member
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 120 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
216 Device* device, Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 257 Tensor* cpu_tensor = in ComputeAsync() local
304 Tensor* cpu_tensor = &value.tensor; in ComputeAsync() local
|
| D | dynamic_partition_op_gpu.cu.cc | 276 Tensor cpu_tensor; in ComputeAsync() local
|
| D | collective_nccl_test.cc | 274 Tensor cpu_tensor(dtype, shape); in InitTensor() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | op_handler_selector_test.cc | 397 tensorflow::core::RefCountPtr<FakeTensorHandle> cpu_tensor( in TEST_F() local
426 tensorflow::core::RefCountPtr<FakeTensorHandle> cpu_tensor( in TEST_F() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 273 Tensor cpu_tensor(tensor.dtype(), tensor.shape()); in TensorDebugString() local
|
| D | copy_tensor.cc | 255 Tensor* cpu_tensor = in ViaDMA() local
|
| D | process_function_library_runtime_test.cc | 181 Tensor cpu_tensor(device_tensor.dtype(), device_tensor.shape()); in GPUToCPU() local
192 Tensor CPUToGPU(const Tensor& cpu_tensor) { in CPUToGPU()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 675 Tensor* cpu_tensor = in RecvBufAsync() local
|