HomeSort by relevance Sort by last modified time
    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