| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op.cc | 87 MemoryTypeVector output_memory_types(fbody->ret_types.size(), DEVICE_MEMORY);
95 fbody->ret_types, output_memory_types, flr->graph_def_version(), &s); local
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api.cc | 796 const tensorflow::MemoryTypeVector* output_memory_types = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.cc | 96 output_memory_types_(context->output_memory_types().begin(),
97 context->output_memory_types().end()),
188 const MemoryTypeSlice& output_memory_types, int graph_def_version,
199 output_memory_types_(output_memory_types),
1159 MemoryTypeVector output_memory_types; local
[all...] |
| op_kernel.h | 138 const MemoryTypeVector& output_memory_types() const { function in class:tensorflow::OpKernel
240 const MemoryTypeSlice& output_memory_types,
292 const MemoryTypeSlice& output_memory_types() const { function in class:tensorflow::OpKernelConstruction
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.cc | 535 MemoryTypeVector output_memory_types; local
548 &input_memory_types, &output_memory_types));
555 for (size_t i = 0; i < output_memory_types.size(); ++i) {
556 info->output_types[{node_id, i}] = output_memory_types[i];
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 396 MemoryTypeVector output_memory_types; local
398 output_memory_types.push_back(t == DT_INT32 ? HOST_MEMORY : DEVICE_MEMORY);
406 fbody->ret_types, output_memory_types, graph_def_version_, &s);
[all...] |
