OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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