| /external/tensorflow/tensorflow/core/framework/ |
| memory_types.h | 34 MemoryTypeVector* output_memory_types);
|
| op_kernel.cc | 103 output_memory_types_(context->output_memory_types().begin(),
104 context->output_memory_types().end()),
206 const MemoryTypeSlice& output_memory_types, int graph_def_version,
217 output_memory_types_(output_memory_types),
1317 MemoryTypeVector output_memory_types; local
[all...] |
| op_kernel.h | 168 const MemoryTypeVector& output_memory_types() const { function in class:tensorflow::OpKernel
272 const MemoryTypeSlice& output_memory_types,
324 const MemoryTypeSlice& output_memory_types() const { function in class:tensorflow::OpKernelConstruction
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op.cc | 217 MemoryTypeVector output_memory_types(fbody->ret_types.size(), DEVICE_MEMORY);
220 output_memory_types[i] = HOST_MEMORY;
235 fbody->ret_types, output_memory_types, flr->graph_def_version(), &s); local
|
| create_xla_launch_op_test.cc | 116 EXPECT_EQ(DEVICE_MEMORY, kernel_->output_memory_types()[0]);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| single_threaded_executor.cc | 128 DCHECK_LT(out, op_kernel->output_memory_types().size());
129 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY;
|
| dataset_test_base.cc | 202 (params_->op_kernel->output_memory_types()[index] == HOST_MEMORY);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| kernel_and_device.cc | 218 out_attrs[i].set_on_host(kernel_->output_memory_types()[i] ==
360 if (kernel_->output_memory_types()[idx] == HOST_MEMORY) {
|
| /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 | 551 MemoryTypeVector output_memory_types; local
564 &input_memory_types, &output_memory_types));
571 for (int i = 0; i < output_memory_types.size(); ++i) {
572 info->output_types[{node_id, i}] = output_memory_types[i];
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 582 MemoryTypeVector output_memory_types; local
584 output_memory_types.push_back(MTypeFromDType(t));
592 fbody->ret_types, output_memory_types, graph_def_version_, &s);
[all...] |
| executor.cc | 769 DCHECK_LT(out, op_kernel->output_memory_types().size());
770 bool on_host = op_kernel->output_memory_types()[out] == HOST_MEMORY;
[all...] |
