Searched refs:terminal_nodes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 185 const NodeMap& node_map, const std::vector<string>& terminal_nodes, in IdentityNTerminalPorts() argument
193 absl::flat_hash_set<string> visited(terminal_nodes.begin(), in IdentityNTerminalPorts()
194 terminal_nodes.end()); in IdentityNTerminalPorts()
195 for (string terminal_node : terminal_nodes) { in IdentityNTerminalPorts()
370 const std::vector<string>& terminal_nodes, in SplitIdentityNInputs() argument
379 IdentityNTerminalPorts(node_map, terminal_nodes, graph->node_size())) { in SplitIdentityNInputs()
402 const std::vector<string>& terminal_nodes) { in SetTransitiveFaninGraph() argument
407 ComputeTransitiveFanin(input_graph, terminal_nodes, &ill_formed); in SetTransitiveFaninGraph()
432 std::vector<string> terminal_nodes(nodes_to_preserve.begin(), in Optimize() local
434 std::sort(terminal_nodes.begin(), terminal_nodes.end()); in Optimize()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.cc | 191 const GraphDef& graph, const std::vector<string>& terminal_nodes) { in ComputeTransitiveFanin() argument
194 ComputeTransitiveFanin(graph, terminal_nodes, &ill_formed); in ComputeTransitiveFanin()
200 const GraphDef& graph, const std::vector<string>& terminal_nodes, in ComputeTransitiveFanin() argument
214 for (const string& root : terminal_nodes) { in ComputeTransitiveFanin()
|
| D | grappler_item.h | 134 const GraphDef& graph, const std::vector<string>& terminal_nodes);
140 const GraphDef& graph, const std::vector<string>& terminal_nodes,
|