Searched refs:tf_graph (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 137 const GraphDef& tf_graph) { in MaybeReplaceCompositeSubgraph() argument
144 MaybeResolveClusters(tf_graph, cluster_factories); in MaybeReplaceCompositeSubgraph()
148 *(pruned_graph->mutable_library()) = tf_graph.library(); in MaybeReplaceCompositeSubgraph()
|
| D | resolve_cluster.h | 59 const tensorflow::GraphDef& tf_graph);
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tensorflow_util.cc | 41 const GraphDef& tf_graph) { in LogDumpGraphDef() argument
46 for (const auto& node : tf_graph.node()) { in LogDumpGraphDef()
55 message, tf_graph.node_size(), ops.size()); in LogDumpGraphDef()
62 for (const auto& node : tf_graph.node()) { in LogDumpGraphDef()
|
| D | tensorflow_util.h | 28 const tensorflow::GraphDef& tf_graph);
|
| D | import_tensorflow.cc | 2560 const GraphDef& tf_graph) { in ImportTensorFlowGraphDef() argument
2561 LogDumpGraphDef(kLogLevelModelChanged, "AT IMPORT", tf_graph); in ImportTensorFlowGraphDef()
2563 GraphDef inlined_graph(tf_graph); in ImportTensorFlowGraphDef()
2618 std::unique_ptr<GraphDef> tf_graph(new GraphDef); in ImportTensorFlowGraphDef() local
2619 CHECK(ParseFromStringEitherTextOrBinary(input_file_contents, tf_graph.get())); in ImportTensorFlowGraphDef()
2622 MaybeReplaceCompositeSubgraph(*tf_graph); in ImportTensorFlowGraphDef()
2624 tf_graph = std::move(pruned_graph); in ImportTensorFlowGraphDef()
2626 return ImportTensorFlowGraphDef(model_flags, tf_import_flags, *tf_graph); in ImportTensorFlowGraphDef()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 54 Status SegmentGraph(const Graph* tf_graph,
|
| D | segment.cc | 404 Status SegmentGraph(const Graph* tf_graph, in SegmentGraph() argument
419 auto graph = std::unique_ptr<SimpleGraph>(new SimpleGraph(tf_graph)); in SegmentGraph()
|