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.cc92 std::unique_ptr<GraphDef> pruned_graph(new GraphDef); in MaybeResolveClusters() local
119 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
127 AddNodeToGraph(node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
131 if (pruned_graph->node_size() == 0) { in MaybeResolveClusters()
134 return pruned_graph; in MaybeResolveClusters()
145 std::unique_ptr<GraphDef> pruned_graph = in MaybeReplaceCompositeSubgraph() local
149 if (pruned_graph) { in MaybeReplaceCompositeSubgraph()
150 *(pruned_graph->mutable_library()) = tf_graph.library(); in MaybeReplaceCompositeSubgraph()
152 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;
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py331 pruned_graph = func_graph.FuncGraph(name)
334 pruned_graph,
341 pruned_graph.outputs.extend(lift_map[x] for x in tensor_fetches)
342 pruned_graph.control_outputs.extend(
344 pruned_graph.inputs.extend(lift_map[x] for x in flat_feeds)
346 pruned_graph.add_capture(external_capture, lift_map[internal_capture])
349 t = pruned_graph.as_graph_element(ti.name)
354 pruned_graph._add_function(f)
357 pruned_graph.variables = self.graph.variables
369 pruned_graph.structured_outputs = nest.map_structure(
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc353 std::unique_ptr<Graph>* pruned_graph,
1012 std::unique_ptr<Graph>* pruned_graph, in MakePrunedGraphCopyAndInline() argument
1016 pruned_graph->reset(new Graph(library)); in MakePrunedGraphCopyAndInline()
1017 (*pruned_graph)->set_versions(graph.versions()); in MakePrunedGraphCopyAndInline()
1022 Node* copied = (*pruned_graph)->CopyNode(n); in MakePrunedGraphCopyAndInline()
1035 (*pruned_graph) in MakePrunedGraphCopyAndInline()
1044 for (auto node : (*pruned_graph)->nodes()) { in MakePrunedGraphCopyAndInline()
1063 TF_RETURN_IF_ERROR(InlineFunctionBody(*library, pruned_graph->get(), node, in MakePrunedGraphCopyAndInline()
/external/tensorflow/tensorflow/lite/
Darena_planner_test.cc605 TestGraph pruned_graph({0, 1}, in TEST_F() local
611 SwapGraph(&pruned_graph); in TEST_F()
647 TestGraph pruned_graph({0, 1}, in TEST_F() local
653 SwapGraph(&pruned_graph); in TEST_F()
/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/core/tfrt/utils/
Dtfrt_graph_execution_state.cc206 auto pruned_graph = in CreatePrunedGraph() local
212 pruned_graph.get())); in CreatePrunedGraph()
213 return pruned_graph; in CreatePrunedGraph()
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py867 pruned_graph = self._func.prune([], [identity_node.name])()[0]
869 numpy=pruned_graph.numpy(),
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc2802 std::unique_ptr<GraphDef> pruned_graph = in ImportTensorFlowGraphDef() local
2804 if (pruned_graph) { in ImportTensorFlowGraphDef()
2805 tf_graph = std::move(pruned_graph); in ImportTensorFlowGraphDef()