Home
last modified time | relevance | path

Searched refs:FindSinkNode (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils_test.cc282 TF_EXPECT_OK(FindSinkNode(graph_def, &sink_node)); in TEST()
294 Status s = FindSinkNode(graph_def, &sink_node); in TEST()
303 Status s = FindSinkNode(graph_def, &sink_node); in TEST()
Dgraph_utils.h148 Status FindSinkNode(const GraphDef& graph_def, NodeDef* sink_node);
Drebatch.cc252 TF_RETURN_IF_ERROR(graph_utils::FindSinkNode(item.graph, &sink_node)); in OptimizeGraph()
Dauto_shard.cc244 TF_RETURN_IF_ERROR(graph_utils::FindSinkNode(item.graph, &sink_node)); in OptimizeGraph()
Dgraph_utils.cc310 Status FindSinkNode(const GraphDef& graph_def, NodeDef* sink_node) { in FindSinkNode() function