| /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}),
83 MemoryTypeVector({HOST_MEMORY, DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY,
84 HOST_MEMORY, HOST_MEMORY, HOST_MEMORY, HOST_MEMORY
[all...] |
| memory_types.cc | 46 // Update args that have been marked as in "HOST_MEMORY".
52 (*memory_types)[j] = HOST_MEMORY;
64 return (dtype == DT_INT32 || DataTypeAlwaysOnHost(dtype)) ? HOST_MEMORY
128 (*inp_mtypes)[i] = HOST_MEMORY;
133 (*out_mtypes)[i] = HOST_MEMORY;
141 (*inp_mtypes)[i] = HOST_MEMORY;
148 (*out_mtypes)[i] = HOST_MEMORY;
|
| types.h | 48 HOST_MEMORY = 1,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| memory_types_test.cc | 81 // float Switch's output on CPU doesn't have HOST_MEMORY constraint.
86 // int Switch's output on GPU has HOST_MEMORY constraint.
87 EXPECT_EQ(memory_type, HOST_MEMORY);
92 // int Switch's output on GPU has HOST_MEMORY constraint.
93 EXPECT_EQ(memory_type, HOST_MEMORY);
|
| 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
162 if (((sm == HOST_MEMORY) && (dm == DEVICE_MEMORY)) ||
163 ((sm == DEVICE_MEMORY) && (dm == HOST_MEMORY))) {
187 Send(g, tensor_name, device_name, (item.sm == HOST_MEMORY), e);
188 recv = Recv(g, tensor_name, device_name, (item.dm == HOST_MEMORY), e);
|
| constant_folding.cc | 468 // 1) If the destination tensor is not an int32 tensor, and has HOST_MEMORY
495 if ((memory_type == HOST_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...] |
| executor.cc | 707 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY;
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| ops_testutil.h | 58 (params->op_kernel->output_memory_types()[index] == HOST_MEMORY);
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.cc | 150 return dst_it->second == HOST_MEMORY;
198 // host_memory = true iff we need to use HostSend/HostCast.
199 bool host_memory = false; local
203 host_memory = (src_it->second == HOST_MEMORY);
209 const string cast_op = (host_memory) ? "_HostCast" : "Cast";
225 const string send_op = (host_memory) ? "_HostSend" : "_Send";
251 // host_memory = true iff we need to use HostRecv/HostCast.
252 bool host_memory = false; local
256 host_memory = (dst_it->second == HOST_MEMORY)
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| runtime.cc | 274 tensorflow::HOST_MEMORY);
|
| c_api.cc | 486 memtypes[i] == tensorflow::HOST_MEMORY ? host_device : op_device;
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_graph_utils.cc | 151 graph, device_type, memory_type == HOST_MEMORY, src_node->name(),
|
| /external/tensorflow/tensorflow/python/grappler/ |
| cluster.i | 255 if (inp_mtypes[i] == tensorflow::HOST_MEMORY) {
|
