HomeSort by relevance Sort by last modified time
    Searched refs:device_tensor (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_util_platform_specific.cc 27 Tensor* device_tensor,
29 GPUUtil::CopyCPUTensorToGPU(cpu_tensor, this, device, device_tensor, done);
32 void GPUDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor,
36 GPUUtil::CopyGPUTensorToCPU(device, this, device_tensor, cpu_tensor, done);
  /external/tensorflow/tensorflow/compiler/jit/
xla_device_context.cc 59 Tensor* device_tensor,
66 device_tensor->tensor_data().data())
71 void* dst_ptr = DMAHelper::base(device_tensor);
92 void XlaTransferManager::CopyDeviceTensorToCPU(const Tensor* device_tensor,
97 if (device_tensor->NumElements() > 0) {
100 device_tensor->tensor_data().data())
103 << device_tensor->NumElements();
106 void* src_ptr = const_cast<void*>(DMAHelper::base(device_tensor));
132 Tensor* device_tensor,
134 manager_.CopyCPUTensorToDevice(cpu_tensor, device, device_tensor, done)
    [all...]
xla_device_context.h 55 Tensor* device_tensor, StatusCallback done) const;
56 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
75 Tensor* device_tensor,
77 void CopyDeviceTensorToCPU(const Tensor* device_tensor,
  /external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_device_context.h 35 Tensor *device_tensor,
38 void CopyDeviceTensorToCPU(const Tensor *device_tensor, StringPiece edge_name,
sycl_device_context.cc 27 Tensor *device_tensor,
32 void *dst_ptr = DMAHelper::base(device_tensor);
102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor,
107 const int64 total_bytes = device_tensor->TotalBytes();
109 const void *src_ptr = DMAHelper::base(device_tensor);
111 switch (device_tensor->dtype()) {
  /external/tensorflow/tensorflow/core/common_runtime/
function_testlib.cc 32 Tensor* device_tensor = nullptr; variable
34 &device_tensor));
35 device_tensor->scalar<string>()() =
gpu_device_context.h 56 Tensor* device_tensor,
59 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
  /external/tensorflow/tensorflow/core/framework/
device_base.h 76 // "device_tensor" which is on a GPU device "device". "device_tensor"
79 Tensor* device_tensor,
84 // "device_tensor" is a tensor on a non-CPU device. Copies
85 // device_tensor into "cpu_tensor". "cpu_tensor" must be allocated
86 // to be of the same size as "device_tensor".
87 virtual void CopyDeviceTensorToCPU(const Tensor* device_tensor,
  /external/tensorflow/tensorflow/core/kernels/
stack_ops.cc 416 Tensor* device_tensor = variable
419 cpu_tensor, device, device_tensor,
420 [device_tensor, ctx, done](const Status& s) {
423 ctx->set_output(0, *device_tensor);
426 delete device_tensor;

Completed in 341 milliseconds