Searched refs:terminal_nodes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | transitive_fanin_test.cc | 69 const std::vector<string> terminal_nodes = {"1"}; in TEST_F() local
70 TF_EXPECT_OK(SetTransitiveFaninGraph(graph, &output_graph, terminal_nodes)); in TEST_F()
88 const std::vector<string> terminal_nodes = {"1"}; in TEST_F() local
89 TF_EXPECT_OK(SetTransitiveFaninGraph(graph, &output_graph, terminal_nodes)); in TEST_F()
109 const std::vector<string> terminal_nodes = {"1", "5"}; in TEST_F() local
110 TF_EXPECT_OK(SetTransitiveFaninGraph(graph, &output_graph, terminal_nodes)); in TEST_F()
131 const std::vector<string> terminal_nodes = {"1", "5"}; in TEST_F() local
132 auto s = SetTransitiveFaninGraph(graph, &output_graph, terminal_nodes); in TEST_F()
|
| D | transitive_fanin.cc | 29 const GraphDef& graph, const std::vector<string>& terminal_nodes, in ComputeTransitiveFanin() argument
43 for (const string& root : terminal_nodes) { in ComputeTransitiveFanin()
89 const std::vector<string>& terminal_nodes, in ComputeTransitiveFanin() argument
91 return ComputeTransitiveFanin(graph, terminal_nodes, nullptr, fanin_nodes); in ComputeTransitiveFanin()
96 const std::vector<string>& terminal_nodes) { in SetTransitiveFaninGraph() argument
101 ComputeTransitiveFanin(input_graph, terminal_nodes, &keep)); in SetTransitiveFaninGraph()
|
| D | transitive_fanin.h | 32 const GraphDef& graph, const std::vector<string>& terminal_nodes,
37 const std::vector<string>& terminal_nodes,
45 const std::vector<string>& terminal_nodes);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | model_pruner.cc | 203 const NodeMap& node_map, const std::vector<string>& terminal_nodes, in IdentityNTerminalPorts() argument
211 absl::flat_hash_set<string> visited(terminal_nodes.begin(), in IdentityNTerminalPorts()
212 terminal_nodes.end()); in IdentityNTerminalPorts()
213 for (const string& terminal_node : terminal_nodes) { in IdentityNTerminalPorts()
388 const std::vector<string>& terminal_nodes, in SplitIdentityNInputs() argument
397 IdentityNTerminalPorts(node_map, terminal_nodes, graph->node_size())) { in SplitIdentityNInputs()
434 std::vector<string> terminal_nodes(nodes_to_preserve.begin(), in Optimize() local
436 std::sort(terminal_nodes.begin(), terminal_nodes.end()); in Optimize()
438 SetTransitiveFaninGraph(item.graph, pruned_graph, terminal_nodes)); in Optimize()
440 TF_RETURN_IF_ERROR(SplitIdentityNInputs(pruned_graph, terminal_nodes, in Optimize()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate.cc | 78 std::vector<std::string> terminal_nodes; in GraphdefToMlirImport() local
79 terminal_nodes.reserve(specs.outputs.size() + specs.inputs.size()); in GraphdefToMlirImport()
81 terminal_nodes.push_back(std::string(ParseTensorName(output).node())); in GraphdefToMlirImport()
84 terminal_nodes.push_back(std::string(control_output)); in GraphdefToMlirImport()
87 terminal_nodes.push_back(input.first); in GraphdefToMlirImport()
90 graphdef, &pruned_graph_def, terminal_nodes)); in GraphdefToMlirImport()
|