Searched refs:tf_graph (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_cluster.cc | 138 const GraphDef& tf_graph) { in MaybeReplaceCompositeSubgraph() argument
145 MaybeResolveClusters(tf_graph, cluster_factories); in MaybeReplaceCompositeSubgraph()
149 *(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 | 2732 const GraphDef& tf_graph) { in ImportTensorFlowGraphDef() argument
2733 LogDumpGraphDef(kLogLevelModelChanged, "AT IMPORT", tf_graph); in ImportTensorFlowGraphDef()
2735 GraphDef inlined_graph(tf_graph); in ImportTensorFlowGraphDef()
2791 std::unique_ptr<GraphDef> tf_graph(new GraphDef); in ImportTensorFlowGraphDef() local
2792 CHECK(ParseFromStringEitherTextOrBinary(input_file_contents, tf_graph.get())); in ImportTensorFlowGraphDef()
2795 MaybeReplaceCompositeSubgraph(*tf_graph); in ImportTensorFlowGraphDef()
2797 tf_graph = std::move(pruned_graph); in ImportTensorFlowGraphDef()
2799 return ImportTensorFlowGraphDef(model_flags, tf_import_flags, *tf_graph); in ImportTensorFlowGraphDef()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.h | 81 Status SegmentGraph(const Graph* tf_graph,
|
| D | segment.cc | 682 Status SegmentGraph(const Graph* tf_graph, in SegmentGraph() argument
711 auto graph = std::unique_ptr<SimpleGraph>(new SimpleGraph(tf_graph)); in SegmentGraph()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer_report.py | 301 tf_graph = tensor_trace_order.graph_order.graph
302 report.graphdef.CopyFrom(tf_graph.as_graph_def())
|