Searched refs:pruned_graph (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 90 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/core/grappler/optimizers/ |
| D | model_pruner.cc | 423 GraphDef* pruned_graph; in Optimize() local
426 pruned_graph = pruned_graph_release.get(); in Optimize()
427 pruned_graph->mutable_node()->Reserve(item.graph.node_size()); in Optimize()
432 SetTransitiveFaninGraph(item.graph, pruned_graph, terminal_nodes)); in Optimize()
434 TF_RETURN_IF_ERROR(SplitIdentityNInputs(pruned_graph, terminal_nodes, in Optimize()
439 *pruned_graph, &fanin_split_identity_n_graph, terminal_nodes)); in Optimize()
440 pruned_graph->Swap(&fanin_split_identity_n_graph); in Optimize()
444 pruned_graph = const_cast<GraphDef*>(&item.graph); in Optimize()
447 GraphView graph_view(pruned_graph); in Optimize()
456 for (const auto& node : pruned_graph->node()) { in Optimize()
[all …]
|
| D | memory_optimizer.h | 48 GraphDef* pruned_graph) override;
51 const GraphDef& pruned_graph, double result) override;
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | wrap_function.py | 316 pruned_graph = func_graph.FuncGraph(name)
319 pruned_graph,
326 pruned_graph.outputs.extend(lift_map[x] for x in tensor_fetches)
327 pruned_graph.control_outputs.extend(
329 pruned_graph.inputs.extend(lift_map[x] for x in flat_feeds)
331 pruned_graph.add_capture(external_capture, lift_map[internal_capture])
334 t = pruned_graph.as_graph_element(ti.name)
339 pruned_graph._add_function(f)
342 pruned_graph.variables = self.graph.variables
354 pruned_graph.structured_outputs = nest.map_structure(
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | arena_planner_test.cc | 585 TestGraph pruned_graph({0, 1}, in TEST_F() local
591 SwapGraph(&pruned_graph); in TEST_F()
627 TestGraph pruned_graph({0, 1}, in TEST_F() local
633 SwapGraph(&pruned_graph); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 372 std::unique_ptr<Graph>* pruned_graph,
1041 std::unique_ptr<Graph>* pruned_graph, in MakePrunedGraphCopyAndInline() argument
1045 pruned_graph->reset(new Graph(library)); in MakePrunedGraphCopyAndInline()
1046 (*pruned_graph)->set_versions(graph.versions()); in MakePrunedGraphCopyAndInline()
1051 Node* copied = (*pruned_graph)->CopyNode(n); in MakePrunedGraphCopyAndInline()
1064 (*pruned_graph) in MakePrunedGraphCopyAndInline()
1073 for (auto node : (*pruned_graph)->nodes()) { in MakePrunedGraphCopyAndInline()
1092 TF_RETURN_IF_ERROR(InlineFunctionBody(*library, pruned_graph->get(), node, in MakePrunedGraphCopyAndInline()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 82 GraphDef pruned_graph; in PruneGraph() local
84 TF_RETURN_IF_ERROR(pruner.Optimize(cluster, *item, &pruned_graph)); in PruneGraph()
85 item->graph = std::move(pruned_graph); in PruneGraph()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 2732 std::unique_ptr<GraphDef> pruned_graph = in ImportTensorFlowGraphDef() local
2734 if (pruned_graph) { in ImportTensorFlowGraphDef()
2735 tf_graph = std::move(pruned_graph); in ImportTensorFlowGraphDef()
|