Home
last modified time | relevance | path

Searched refs:output_nodes_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc433 return output_nodes_.contains(node.name()); in IsOutputNode()
438 absl::flat_hash_set<absl::string_view> output_nodes_; member in tensorflow::grappler::__anoneaf86e2d0511::MakeFunctionDefHelper
449 output_nodes_.insert(output_arg.node_name); in Initialize()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc153 std::vector<string> output_nodes_; member in tensorflow::tensorrt::TRTEngineOp