| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 122 Tensor* device_tensor, in CopyCPUTensorToDevice() argument
135 << reinterpret_cast<const void*>(device_tensor->tensor_data().data()) in CopyCPUTensorToDevice()
138 << device_tensor->shape().DebugString(); in CopyCPUTensorToDevice()
140 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice()
146 shape_representation_fn_(device_tensor->shape(), device_tensor->dtype(), in CopyCPUTensorToDevice()
153 xla_tensor->AllocateShapedBuffer(device_tensor->dtype(), shape, client_, in CopyCPUTensorToDevice()
214 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
218 if (device_tensor->NumElements() == 0) { in CopyDeviceTensorToCPU()
224 << reinterpret_cast<const void*>(device_tensor->tensor_data().data()) in CopyDeviceTensorToCPU()
227 << " " << device_tensor->NumElements() << " " in CopyDeviceTensorToCPU()
[all …]
|
| D | xla_device_context.h | 65 Tensor* device_tensor, StatusCallback done,
67 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 28 Tensor* device_tensor, StatusCallback done,
30 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
40 Tensor* device_tensor, in CopyCPUTensorToDevice() argument
43 *device_tensor = *cpu_tensor; in CopyCPUTensorToDevice()
47 void VirtualDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
52 *cpu_tensor = *device_tensor; in CopyDeviceTensorToCPU()
|
| /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() argument
31 cpu_tensor, this, device, device_tensor, done, sync_dst_compute); in CopyCPUTensorToDevice()
34 void PluggableDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
40 device, this, device_tensor, cpu_tensor, done); in CopyDeviceTensorToCPU()
|
| D | pluggable_device_util.cc | 176 const Tensor* device_tensor, Tensor* cpu_tensor, StatusCallback done) { in CopyPluggableDeviceTensorToCPU() argument
180 Status s = PrepareCopy(device, device_context, *device_tensor, cpu_tensor, in CopyPluggableDeviceTensorToCPU()
198 const int64_t total_bytes = device_tensor->TotalBytes(); in CopyPluggableDeviceTensorToCPU()
200 void* src_ptr = GetBase(device_tensor); in CopyPluggableDeviceTensorToCPU()
208 TensorReference input_ref(*device_tensor); in CopyPluggableDeviceTensorToCPU()
223 Device* device, Tensor* device_tensor, StatusCallback done, in CopyCPUTensorToPluggableDevice() argument
228 Status s = PrepareCopy(device, device_context, *cpu_tensor, device_tensor, in CopyCPUTensorToPluggableDevice()
251 void* dst_ptr = GetBase(device_tensor); in CopyCPUTensorToPluggableDevice()
|
| D | pluggable_device_util.h | 38 const Tensor* device_tensor, Tensor* cpu_tensor, StatusCallback done);
51 Device* device, Tensor* device_tensor, StatusCallback done,
|
| D | pluggable_device_context.h | 53 Tensor* device_tensor, StatusCallback done,
56 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util_platform_specific.cc | 28 Tensor* device_tensor, in CopyCPUTensorToDevice() argument
31 GPUUtil::CopyCPUTensorToGPU(cpu_tensor, this, device, device_tensor, done, in CopyCPUTensorToDevice()
35 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
39 GPUUtil::CopyGPUTensorToCPU(device, this, device_tensor, cpu_tensor, done); in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 37 Status DeviceContext::CopyDeviceTensorToCPUSync(const Tensor* device_tensor, in CopyDeviceTensorToCPUSync() argument
43 CopyDeviceTensorToCPU(device_tensor, tensor_name, device, cpu_tensor, in CopyDeviceTensorToCPUSync()
54 Tensor* device_tensor) const { in CopyCPUTensorToDeviceSync()
57 CopyCPUTensorToDevice(cpu_tensor, device, device_tensor, in CopyCPUTensorToDeviceSync()
|
| D | device_base.h | 79 Tensor* device_tensor, StatusCallback done,
86 Tensor* device_tensor) const;
98 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
105 Status CopyDeviceTensorToCPUSync(const Tensor* device_tensor,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_testlib.cc | 33 Tensor* device_tensor = nullptr; in Compute() local
35 &device_tensor)); in Compute()
36 device_tensor->scalar<tstring>()() = in Compute()
|
| D | gpu_device_context.h | 63 Tensor* device_tensor, StatusCallback done,
66 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
|
| D | process_function_library_runtime_test.cc | 174 Tensor GPUToCPU(const Tensor& device_tensor) { in GPUToCPU() argument
181 Tensor cpu_tensor(device_tensor.dtype(), device_tensor.shape()); in GPUToCPU()
183 ->CopyDeviceTensorToCPUSync(&device_tensor, "", gpu_device_, in GPUToCPU()
199 Tensor device_tensor(gpu_device_->GetAllocator({}), cpu_tensor.dtype(), in CPUToGPU() local
203 &device_tensor) in CPUToGPU()
205 return device_tensor; in CPUToGPU()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 306 Tensor* device_tensor = in ComputeAsync() local
309 cpu_tensor, device, device_tensor, in ComputeAsync()
310 [device_tensor, ctx, done](const Status& s) { in ComputeAsync()
313 ctx->set_output(0, *device_tensor); in ComputeAsync()
316 delete device_tensor; in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 572 const auto device_tensor = &calib_ctx->device_tensors_.at(i); in ExecuteCalibration() local
573 CHECK_EQ(t.TotalBytes(), device_tensor->TotalBytes()); in ExecuteCalibration()
|