Searched refs:fanin_nodes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 71 std::vector<const NodeDef*> fanin_nodes; in MainOpsFanin() local
72 TF_CHECK_OK(ComputeTransitiveFanin(graph, fetch, &fanin_nodes)); in MainOpsFanin()
73 return fanin_nodes; in MainOpsFanin()
83 std::vector<const NodeDef*> fanin_nodes; in EnqueueOpsFanin() local
84 TF_CHECK_OK(ComputeTransitiveFanin(graph, fetch, &fanin_nodes)); in EnqueueOpsFanin()
85 return fanin_nodes; in EnqueueOpsFanin()
89 std::vector<const NodeDef*> fanin_nodes; in InitOpsFanin() local
90 TF_CHECK_OK(ComputeTransitiveFanin(graph, init_ops, &fanin_nodes)); in InitOpsFanin()
91 return fanin_nodes; in InitOpsFanin()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | transitive_fanin.cc | 31 std::vector<const NodeDef*>* fanin_nodes) { in ComputeTransitiveFanin() argument
61 fanin_nodes->push_back(node); in ComputeTransitiveFanin()
90 std::vector<const NodeDef*>* fanin_nodes) { in ComputeTransitiveFanin() argument
91 return ComputeTransitiveFanin(graph, terminal_nodes, nullptr, fanin_nodes); in ComputeTransitiveFanin()
|
| D | transitive_fanin.h | 34 std::vector<const NodeDef*>* fanin_nodes);
38 std::vector<const NodeDef*>* fanin_nodes);
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | graph_analyzer_tool.cc | 57 std::vector<const tensorflow::NodeDef*> fanin_nodes; in MaybePruneGraph() local
59 metagraph.graph_def(), fetch_nodes, &fanin_nodes)); in MaybePruneGraph()
60 for (const tensorflow::NodeDef* node : fanin_nodes) { in MaybePruneGraph()
|