| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device_context.cc | 25 void SYCLDeviceContext::CopyCPUTensorToDevice(const Tensor *cpu_tensor, in CopyCPUTensorToDevice()
105 Tensor *cpu_tensor, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 26 void GPUDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
35 Device* device, Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| D | gpu_util.cc | 257 const Tensor* gpu_tensor, Tensor* cpu_tensor, in CopyGPUTensorToCPU()
300 void GPUUtil::CopyCPUTensorToGPU(const Tensor* cpu_tensor, in CopyCPUTensorToGPU()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 106 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()
201 Device* device, Tensor* cpu_tensor, in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 65 Tensor* cpu_tensor) { in PopulateTensorFromExtra()
131 Tensor* cpu_tensor = new Tensor(cpu_dev->GetAllocator(cpu_attr), in RecvFromPeer() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 79 virtual void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device, in CopyCPUTensorToDevice()
97 Tensor* cpu_tensor, StatusCallback done) { in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 256 Tensor* cpu_tensor = in ComputeAsync() local
303 Tensor* cpu_tensor = &value.tensor; in ComputeAsync() local
|
| D | collective_nccl_reducer_test.cc | 234 Tensor cpu_tensor(dtype, shape); in InitTensor() local
|
| D | dynamic_partition_op_gpu.cu.cc | 280 Tensor cpu_tensor; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 256 Tensor cpu_tensor(tensor.dtype(), tensor.shape()); in TensorDebugString() local
|
| D | process_function_library_runtime_test.cc | 114 Tensor cpu_tensor(device_tensor.dtype(), device_tensor.shape()); in GPUToCPU() local
129 Tensor CPUToGPU(const Tensor& cpu_tensor) { in CPUToGPU()
|
| D | copy_tensor.cc | 295 Tensor* cpu_tensor = in ViaDMA() local
|
| D | ring_gatherer_test.cc | 425 Tensor cpu_tensor(dtype, shape); in InitTensor() local
|
| D | ring_reducer_test.cc | 449 Tensor cpu_tensor(dtype, shape); in InitTensor() local
|
| D | hierarchical_tree_broadcaster_test.cc | 613 Tensor cpu_tensor(dtype, shape); in InitTensor() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 593 Tensor* cpu_tensor = new Tensor(cpu_dev->GetAllocator(cpu_attr), in RecvBufAsync() local
|