OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:DEVICE_MEMORY
(Results
1 - 11
of
11
) sorted by null
/external/tensorflow/tensorflow/core/framework/
memory_types_test.cc
71
MemoryTypeVector({
DEVICE_MEMORY
,
DEVICE_MEMORY
, HOST_MEMORY, HOST_MEMORY,
72
HOST_MEMORY,
DEVICE_MEMORY
,
DEVICE_MEMORY
,
73
DEVICE_MEMORY
, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY}),
76
EXPECT_EQ(MemoryTypeVector({
DEVICE_MEMORY
,
DEVICE_MEMORY
,
DEVICE_MEMORY
,
77
DEVICE_MEMORY
,
DEVICE_MEMORY
, DEVICE_MEMORY})
[
all
...]
memory_types.cc
41
// to
DEVICE_MEMORY
except those args in host_memory_args. Removes
65
:
DEVICE_MEMORY
;
108
// Now that we know the size, fill with the default '
DEVICE_MEMORY
'.
109
inp_mtypes->resize(GetTotal(inp_names),
DEVICE_MEMORY
);
110
out_mtypes->resize(GetTotal(out_names),
DEVICE_MEMORY
);
types.h
47
DEVICE_MEMORY
= 0,
op_kernel.cc
503
forward_input(input_index, type, shape,
DEVICE_MEMORY
, allocator_attr);
[
all
...]
/external/tensorflow/tensorflow/compiler/jit/
create_xla_launch_op.cc
86
MemoryTypeVector input_memory_types(fbody->arg_types.size(),
DEVICE_MEMORY
);
87
MemoryTypeVector output_memory_types(fbody->ret_types.size(),
DEVICE_MEMORY
);
/external/tensorflow/tensorflow/core/common_runtime/
memory_types.cc
51
// On non-GPU and non-SYCL devices, HOST_MEMORY and
DEVICE_MEMORY
are always
55
// For GPU and SYCL device, HOST_MEMORY and
DEVICE_MEMORY
is not
82
DEVICE_MEMORY
);
84
DEVICE_MEMORY
);
162
if (((sm == HOST_MEMORY) && (dm ==
DEVICE_MEMORY
)) ||
163
((sm ==
DEVICE_MEMORY
) && (dm == HOST_MEMORY))) {
memory_types_test.cc
82
EXPECT_EQ(memory_type,
DEVICE_MEMORY
);
constant_folding.cc
470
// 2) If the destination tensor is an int32 tensor, but has
DEVICE_MEMORY
496
(memory_type ==
DEVICE_MEMORY
&& is_int32)) {
function.cc
394
(t == DT_INT32 || t == DT_RESOURCE) ? HOST_MEMORY :
DEVICE_MEMORY
);
398
output_memory_types.push_back(t == DT_INT32 ? HOST_MEMORY :
DEVICE_MEMORY
);
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
assign_op.h
80
1, old_lhs.dtype(), old_lhs.shape(),
DEVICE_MEMORY
, attr);
resource_variable_ops.cc
254
context->forward_input(1, dtype_, value.shape(),
DEVICE_MEMORY
, attr);
Completed in 109 milliseconds