Home
last modified time | relevance | path

Searched defs:sink_node (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils_test.cc281 NodeDef sink_node; in TEST() local
293 NodeDef sink_node; in TEST() local
302 NodeDef sink_node; in TEST() local
Dauto_shard.cc243 NodeDef sink_node; in OptimizeGraph() local
Drebatch.cc251 NodeDef sink_node; in OptimizeGraph() local
Dgraph_utils.cc310 Status FindSinkNode(const GraphDef& graph_def, NodeDef* sink_node) { in FindSinkNode()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc246 Node* sink_node = graph->sink_node(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc130 const SimpleNode* sink_node() const { return nodes_[Graph::kSinkId]; } in sink_node() function in tensorflow::tensorrt::segment::SimpleGraph
/external/tensorflow/tensorflow/core/graph/
Dgraph.h605 Node* sink_node() const { return FindNodeId(kSinkId); } in sink_node() function