Home
last modified time | relevance | path

Searched refs:output_attrs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc82 gtl::InlinedVector<AllocatorAttributes, 4> output_attrs; in EvaluateNode() local
87 output_attrs.push_back(attr); in EvaluateNode()
89 params.output_attr_array = output_attrs.data(); in EvaluateNode()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_view.cc187 AllocatorAttributes* output_attrs = item->output_attr_base(); in InitializeNode() local
189 new (&output_attrs[i]) AllocatorAttributes(); in InitializeNode()
223 DCHECK_EQ(output_attrs[i].scope_id, 0); in InitializeNode()
224 output_attrs[i].scope_id = scoped_allocator_attrs[j + 1]; in InitializeNode()
Dgraph_view.h136 const AllocatorAttributes* output_attrs() const { return output_attr_base(); } in output_attrs() function
Dexecutor.cc660 output.alloc_attr = item.output_attrs()[0]; in ProcessConstTensor()
803 params.output_attr_array = item.output_attrs(); in Process()