| /external/tensorflow/tensorflow/core/framework/ |
| memory_types.h | 33 MemoryTypeVector* input_memory_types,
34 MemoryTypeVector* output_memory_types);
|
| memory_types_test.cc | 64 MemoryTypeVector input, output;
71 MemoryTypeVector({DEVICE_MEMORY, DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY,
76 EXPECT_EQ(MemoryTypeVector({DEVICE_MEMORY, DEVICE_MEMORY, DEVICE_MEMORY,
83 MemoryTypeVector({HOST_MEMORY, DEVICE_MEMORY, HOST_MEMORY, HOST_MEMORY,
87 EXPECT_EQ(MemoryTypeVector({HOST_MEMORY, HOST_MEMORY, HOST_MEMORY,
|
| memory_types.cc | 45 MemoryTypeVector* memory_types) {
72 MemoryTypeVector* inp_mtypes,
73 MemoryTypeVector* out_mtypes) {
|
| op_kernel.h | 130 const MemoryTypeVector& input_memory_types() const {
138 const MemoryTypeVector& output_memory_types() const {
173 const MemoryTypeVector input_memory_types_;
175 const MemoryTypeVector output_memory_types_;
[all...] |
| types.h | 99 typedef gtl::InlinedVector<MemoryType, 4> MemoryTypeVector;
|
| op_kernel.cc | [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 | 63 MemoryTypeVector inp_mvec;
64 MemoryTypeVector out_mvec;
206 MemoryTypeVector inp_mvec;
207 MemoryTypeVector out_mvec;
|
| function.cc | 391 MemoryTypeVector input_memory_types;
396 MemoryTypeVector output_memory_types;
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| cluster.i | 249 tensorflow::MemoryTypeVector inp_mtypes;
250 tensorflow::MemoryTypeVector out_mtypes;
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api.cc | 479 const tensorflow::MemoryTypeVector& memtypes = kernel->input_memory_types();
796 const tensorflow::MemoryTypeVector* output_memory_types = nullptr;
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.cc | 534 MemoryTypeVector input_memory_types;
535 MemoryTypeVector output_memory_types;
[all...] |
