| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 108 Tensor* device_tensor, in CopyCPUTensorToDevice() argument
119 << reinterpret_cast<const void*>(device_tensor->tensor_data().data()) in CopyCPUTensorToDevice()
122 << device_tensor->shape().DebugString(); in CopyCPUTensorToDevice()
125 XlaTensor* xla_tensor = XlaTensor::FromTensor(device_tensor); in CopyCPUTensorToDevice()
130 shape_representation_fn_(device_tensor->shape(), in CopyCPUTensorToDevice()
131 device_tensor->dtype())); in CopyCPUTensorToDevice()
138 xla_tensor->AllocateShapedBuffer(device_tensor->dtype(), shape, client_, in CopyCPUTensorToDevice()
199 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
203 if (device_tensor->NumElements() == 0) { in CopyDeviceTensorToCPU()
209 << reinterpret_cast<const void*>(device_tensor->tensor_data().data()) in CopyDeviceTensorToCPU()
[all …]
|
| D | xla_compile_on_demand_op.cc | 119 const Tensor& device_tensor = ctx->input(i); in Compile() local
120 if (const XlaTensor* xla_tensor = XlaTensor::FromTensor(&device_tensor)) { in Compile()
143 device_tensor.dtype(), device_tensor.shape(), &host_tensor, attrs)); in Compile()
147 &device_tensor, "ConstantArgument", in Compile()
156 << device_tensor.shape().DebugString() << " from " in Compile()
|
| D | xla_device_context.h | 64 Tensor* device_tensor,
66 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
30 GPUUtil::CopyCPUTensorToGPU(cpu_tensor, this, device, device_tensor, done); in CopyCPUTensorToDevice()
33 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
37 GPUUtil::CopyGPUTensorToCPU(device, this, device_tensor, cpu_tensor, done); in CopyDeviceTensorToCPU()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device_context.cc | 27 Tensor *device_tensor, in CopyCPUTensorToDevice() argument
32 void *dst_ptr = DMAHelper::base(device_tensor); in CopyCPUTensorToDevice()
102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, in CopyDeviceTensorToCPU() argument
107 const int64 total_bytes = device_tensor->TotalBytes(); in CopyDeviceTensorToCPU()
109 const void *src_ptr = DMAHelper::base(device_tensor); in CopyDeviceTensorToCPU()
111 switch (device_tensor->dtype()) { in CopyDeviceTensorToCPU()
|
| D | sycl_device_context.h | 35 Tensor *device_tensor,
38 void CopyDeviceTensorToCPU(const Tensor *device_tensor, StringPiece edge_name,
|
| /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<string>()() = in Compute()
|
| D | gpu_device_context.h | 53 Tensor* device_tensor,
56 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
|
| D | process_function_library_runtime_test.cc | 105 Tensor GPUToCPU(const Tensor& device_tensor) { in GPUToCPU() argument
114 Tensor cpu_tensor(device_tensor.dtype(), device_tensor.shape()); in GPUToCPU()
115 device_context->CopyDeviceTensorToCPU(&device_tensor, "", gpu_device_, in GPUToCPU()
138 Tensor device_tensor(gpu_device_->GetAllocator({}), cpu_tensor.dtype(), in CPUToGPU() local
141 &device_tensor, in CPUToGPU()
148 return device_tensor; in CPUToGPU()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 305 Tensor* device_tensor = in ComputeAsync() local
308 cpu_tensor, device, device_tensor, in ComputeAsync()
309 [device_tensor, ctx, done](const Status& s) { in ComputeAsync()
312 ctx->set_output(0, *device_tensor); in ComputeAsync()
315 delete device_tensor; in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 316 const auto device_tensor = in ExecuteCalibration() local
318 CHECK_EQ(t.TotalBytes(), device_tensor->TotalBytes()); in ExecuteCalibration()
669 Tensor* device_tensor; in AllocateCalibrationResources() local
671 t.dtype(), t.shape(), &cres->device_tensors_.at(i), &device_tensor)); in AllocateCalibrationResources()
672 CHECK_EQ(t.TotalBytes(), device_tensor->TotalBytes()); in AllocateCalibrationResources()
673 void* device_address = GetTensorAddress(device_tensor); in AllocateCalibrationResources()
680 std::pair<void*, size_t>(device_address, device_tensor->TotalBytes())); in AllocateCalibrationResources()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 80 Tensor* device_tensor, in CopyCPUTensorToDevice() argument
95 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU() argument
|