Home
last modified time | relevance | path

Searched refs:pruned_graph (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_cluster.cc90 std::unique_ptr<GraphDef> pruned_graph(new GraphDef); in MaybeResolveClusters() local
117 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
125 AddNodeToGraph(node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
129 if (pruned_graph->node_size() == 0) { in MaybeResolveClusters()
132 return pruned_graph; in MaybeResolveClusters()
143 std::unique_ptr<GraphDef> pruned_graph = in MaybeReplaceCompositeSubgraph() local
147 if (pruned_graph) { in MaybeReplaceCompositeSubgraph()
148 *(pruned_graph->mutable_library()) = tf_graph.library(); in MaybeReplaceCompositeSubgraph()
150 return pruned_graph; in MaybeReplaceCompositeSubgraph()
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py176 pruned_graph = func_graph.FuncGraph(name)
185 [sink_tensor], pruned_graph, sources=flat_feeds + internal_captures)
189 pruned_graph.outputs.extend(
192 pruned_graph.outputs.append(lift_map[sink_tensor])
194 pruned_graph.captures[external_capture] = lift_map[internal_capture]
195 pruned_graph.inputs.extend(lift_map[x] for x in flat_feeds)
196 pruned_graph.inputs.extend(pruned_graph.captures.values())
198 pruned_graph.variables = self.graph.variables
206 pruned_graph.structured_outputs = nest.map_structure(
209 pruned_graph, variable_holder=self._variable_holder)
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc424 GraphDef* pruned_graph) { in Optimize() argument
488 pruned_graph->Clear(); in Optimize()
489 *pruned_graph->mutable_library() = item.graph.library(); in Optimize()
490 *pruned_graph->mutable_versions() = item.graph.versions(); in Optimize()
493 pruned_graph->mutable_node()->Swap(runnable_item.graph.mutable_node()); in Optimize()
498 pruned_graph->mutable_node()->Reserve(runnable_item.graph.node_size()); in Optimize()
503 NodeDef* new_node = pruned_graph->add_node(); in Optimize()
512 << pruned_graph->node_size() << " nodes."; in Optimize()
513 CHECK_LE(pruned_graph->node_size(), item.graph.node_size()); in Optimize()
519 const GraphDef& pruned_graph, double result) { in Feedback() argument
Dmodel_pruner.h35 GraphDef* pruned_graph) override;
38 const GraphDef& pruned_graph, double result) override;
Dmemory_optimizer.h46 GraphDef* pruned_graph) override;
49 const GraphDef& pruned_graph, double result) override;
Dmeta_optimizer.cc674 const GraphDef& pruned_graph, double result) { in Feedback() argument
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc625 std::unique_ptr<Graph>* pruned_graph,
635 const Graph& graph, std::unique_ptr<Graph>* pruned_graph,
2276 std::unique_ptr<Graph>* pruned_graph, in MakePrunedGraphCopyAndInline() argument
2280 pruned_graph->reset(new Graph(library)); in MakePrunedGraphCopyAndInline()
2281 (*pruned_graph)->set_versions(graph.versions()); in MakePrunedGraphCopyAndInline()
2286 Node* copied = (*pruned_graph)->CopyNode(n); in MakePrunedGraphCopyAndInline()
2299 (*pruned_graph) in MakePrunedGraphCopyAndInline()
2308 for (auto node : (*pruned_graph)->nodes()) { in MakePrunedGraphCopyAndInline()
2333 TF_RETURN_IF_ERROR(InlineFunctionBody(*library, pruned_graph->get(), node, in MakePrunedGraphCopyAndInline()
2342 const Graph& graph, std::unique_ptr<Graph>* pruned_graph, in MakeGraphForOutsideCompilationSends() argument
[all …]
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc159 GraphDef pruned_graph; in PruneGraph() local
161 TF_RETURN_IF_ERROR(pruner.Optimize(cluster, *item, &pruned_graph)); in PruneGraph()
162 item->graph = std::move(pruned_graph); in PruneGraph()
/external/tensorflow/tensorflow/lite/
Darena_planner_test.cc523 TestGraph pruned_graph({0, 1}, in TEST_F() local
529 SwapGraph(&pruned_graph); in TEST_F()
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc2621 std::unique_ptr<GraphDef> pruned_graph = in ImportTensorFlowGraphDef() local
2623 if (pruned_graph) { in ImportTensorFlowGraphDef()
2624 tf_graph = std::move(pruned_graph); in ImportTensorFlowGraphDef()