Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_cluster.cc91 std::unique_ptr<GraphDef> pruned_graph(new GraphDef); in MaybeResolveClusters() local
118 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
126 AddNodeToGraph(node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
130 if (pruned_graph->node_size() == 0) { in MaybeResolveClusters()
133 return pruned_graph; in MaybeResolveClusters()
144 std::unique_ptr<GraphDef> pruned_graph = in MaybeReplaceCompositeSubgraph() local
148 if (pruned_graph) { in MaybeReplaceCompositeSubgraph()
149 *(pruned_graph->mutable_library()) = tf_graph.library(); in MaybeReplaceCompositeSubgraph()
151 return pruned_graph; in MaybeReplaceCompositeSubgraph()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc429 GraphDef* pruned_graph; in Optimize() local
432 pruned_graph = pruned_graph_release.get(); in Optimize()
433 pruned_graph->mutable_node()->Reserve(item.graph.node_size()); in Optimize()
438 SetTransitiveFaninGraph(item.graph, pruned_graph, terminal_nodes)); in Optimize()
440 TF_RETURN_IF_ERROR(SplitIdentityNInputs(pruned_graph, terminal_nodes, in Optimize()
445 *pruned_graph, &fanin_split_identity_n_graph, terminal_nodes)); in Optimize()
446 pruned_graph->Swap(&fanin_split_identity_n_graph); in Optimize()
450 pruned_graph = const_cast<GraphDef*>(&item.graph); in Optimize()
453 GraphView graph_view(pruned_graph); in Optimize()
462 for (int i = 0; i < pruned_graph->node_size(); ++i) { in Optimize()
[all …]
Dmemory_optimizer.h48 GraphDef* pruned_graph) override;
51 const GraphDef& pruned_graph, double result) override;
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py333 pruned_graph = func_graph.FuncGraph(name)
336 pruned_graph,
343 pruned_graph.outputs.extend(lift_map[x] for x in tensor_fetches)
344 pruned_graph.control_outputs.extend(
346 pruned_graph.inputs.extend(lift_map[x] for x in flat_feeds)
348 pruned_graph.add_capture(external_capture, lift_map[internal_capture])
351 t = pruned_graph.as_graph_element(ti.name)
356 pruned_graph._add_function(f)
359 pruned_graph.variables = self.graph.variables
371 pruned_graph.structured_outputs = nest.map_structure(
[all …]
/external/tensorflow/tensorflow/lite/
Darena_planner_test.cc615 TestGraph pruned_graph({0, 1}, in TEST_F() local
621 SwapGraph(&pruned_graph); in TEST_F()
657 TestGraph pruned_graph({0, 1}, in TEST_F() local
663 SwapGraph(&pruned_graph); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc372 std::unique_ptr<Graph>* pruned_graph,
1038 std::unique_ptr<Graph>* pruned_graph, in MakePrunedGraphCopyAndInline() argument
1042 pruned_graph->reset(new Graph(library)); in MakePrunedGraphCopyAndInline()
1043 (*pruned_graph)->set_versions(graph.versions()); in MakePrunedGraphCopyAndInline()
1048 Node* copied = (*pruned_graph)->CopyNode(n); in MakePrunedGraphCopyAndInline()
1061 (*pruned_graph) in MakePrunedGraphCopyAndInline()
1070 for (auto node : (*pruned_graph)->nodes()) { in MakePrunedGraphCopyAndInline()
1089 TF_RETURN_IF_ERROR(InlineFunctionBody(*library, pruned_graph->get(), node, in MakePrunedGraphCopyAndInline()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc83 GraphDef pruned_graph; in PruneGraph() local
85 TF_RETURN_IF_ERROR(pruner.Optimize(cluster, *item, &pruned_graph)); in PruneGraph()
86 item->graph = std::move(pruned_graph); in PruneGraph()
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py841 pruned_graph = self._func.prune([], [identity_node.name])()[0]
843 numpy=pruned_graph.numpy(),
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc2794 std::unique_ptr<GraphDef> pruned_graph = in ImportTensorFlowGraphDef() local
2796 if (pruned_graph) { in ImportTensorFlowGraphDef()
2797 tf_graph = std::move(pruned_graph); in ImportTensorFlowGraphDef()