| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | file_input_yielder.cc | 73 MetaGraphDef metagraph; in NextItem() local
74 Status s = ReadBinaryProto(Env::Default(), filename, &metagraph); in NextItem()
76 s = ReadTextProto(Env::Default(), filename, &metagraph); in NextItem()
86 if (metagraph.collection_def().count("train_op") == 0 || in NextItem()
87 !metagraph.collection_def().at("train_op").has_node_list() || in NextItem()
88 metagraph.collection_def().at("train_op").node_list().value_size() == 0) { in NextItem()
91 metagraph = MetaGraphDef(); in NextItem()
96 metagraph.collection_def().at("train_op").node_list().value()) { in NextItem()
100 for (auto& node : metagraph.graph_def().node()) { in NextItem()
112 metagraph = MetaGraphDef(); in NextItem()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | graph_analyzer_tool.cc | 31 tensorflow::MetaGraphDef* metagraph) { in LoadModel() argument
34 st = ReadBinaryProto(Env::Default(), filename, metagraph); in LoadModel()
37 st = ReadTextProto(Env::Default(), filename, metagraph); in LoadModel()
46 void MaybePruneGraph(const tensorflow::MetaGraphDef& metagraph, in MaybePruneGraph() argument
50 metagraph.collection_def().at("train_op").node_list().value()) { in MaybePruneGraph()
55 *graph = metagraph.graph_def(); in MaybePruneGraph()
59 metagraph.graph_def(), fetch_nodes, &fanin_nodes)); in MaybePruneGraph()
64 << metagraph.graph_def().node_size() - graph->node_size() in MaybePruneGraph()
66 << metagraph.graph_def().node_size() in MaybePruneGraph()
76 tensorflow::MetaGraphDef metagraph; in GraphAnalyzerTool() local
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer.py | 26 def GenerateCostReport(metagraph, argument
47 return tf_wrap.GenerateCostReport(metagraph.SerializeToString(),
52 def GenerateMemoryReport(metagraph, detailed_report=True, cluster=None): argument
69 item = gitem.Item(metagraph)
|
| D | item.py | 30 metagraph, argument
44 self._metagraph = metagraph
46 self._item_graph.CopyFrom(metagraph)
81 def metagraph(self): member in Item
|
| D | cost_analyzer_tool.py | 81 metagraph = get_metagraph()
86 optimized_graph = tf_optimizer.OptimizeGraph(config, metagraph)
87 metagraph.graph_def.CopyFrom(optimized_graph)
89 report = cost_analyzer.GenerateCostReport(metagraph, FLAGS.per_node_report,
93 report = cost_analyzer.GenerateMemoryReport(metagraph)
|
| D | model_analyzer_wrapper.cc | 31 tensorflow::MetaGraphDef metagraph; in PYBIND11_MODULE() local
32 if (!metagraph.ParseFromString(std::string(serialized_metagraph))) { in PYBIND11_MODULE()
41 "metagraph", metagraph, cfg); in PYBIND11_MODULE()
|
| D | cost_analyzer_wrapper.cc | 34 tensorflow::MetaGraphDef metagraph; in PYBIND11_MODULE() local
35 if (!metagraph.ParseFromString(std::string(serialized_metagraph))) { in PYBIND11_MODULE()
44 "metagraph", metagraph, cfg); in PYBIND11_MODULE()
|
| D | memory_optimizer_test.py | 133 metagraph = train.export_meta_graph()
134 return (metagraph, init_op.name, train_op.name, loss.name)
231 self, config, metagraph, init_op_name, train_op_name, loss_op_name): argument
234 train.import_meta_graph(metagraph)
250 metagraph=original_metagraph,
256 metagraph=original_metagraph,
310 metagraph = train.export_meta_graph()
313 len([node for node in metagraph.graph_def.node
320 rewritten_graph_def = tf_optimizer.OptimizeGraph(config, metagraph)
|
| D | tf_optimizer_wrapper.cc | 74 tensorflow::MetaGraphDef metagraph; in PYBIND11_MODULE() local
75 if (!metagraph.ParseFromString(std::string(serialized_metagraph))) { in PYBIND11_MODULE()
88 graph_id, metagraph, item_config); in PYBIND11_MODULE()
|
| D | model_analyzer.py | 24 def GenerateModelReport(metagraph, assume_valid_feeds=True, debug=False): argument
36 metagraph.SerializeToString(), assume_valid_feeds, debug)
|
| D | tf_optimizer.py | 28 metagraph, argument
57 metagraph.SerializeToString(), verbose,
|
| D | item_wrapper.cc | 131 tensorflow::MetaGraphDef metagraph; in PYBIND11_MODULE() local
132 if (!metagraph.ParseFromString(std::string(serialized_metagraph))) { in PYBIND11_MODULE()
137 if (metagraph.collection_def().count("train_op") == 0) { in PYBIND11_MODULE()
147 "item", metagraph, cfg); in PYBIND11_MODULE()
|
| D | item_test.py | 75 for node in grappler_item.metagraph.graph_def.node:
100 for node in grappler_item.metagraph.graph_def.node:
106 for node in grappler_item.metagraph.graph_def.node:
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.cc | 383 Status PartiallyReviveSavedModelObjects(const MetaGraphDef& metagraph, in PartiallyReviveSavedModelObjects() argument
389 NodeAttrMap node_attr_map = NodeToAttrMap(metagraph.graph_def()); in PartiallyReviveSavedModelObjects()
393 TF_RETURN_IF_ERROR(GetAssetFileDefs(metagraph, &assets)); in PartiallyReviveSavedModelObjects()
399 GetSignaturesMap(metagraph.object_graph_def(), &signatures_map)); in PartiallyReviveSavedModelObjects()
411 internal::FunctionNameToFunctionDefMap(metagraph.graph_def().library()); in PartiallyReviveSavedModelObjects()
417 for (int i = 0; i < metagraph.object_graph_def().nodes_size(); ++i) { in PartiallyReviveSavedModelObjects()
418 const SavedObject& node = metagraph.object_graph_def().nodes(i); in PartiallyReviveSavedModelObjects()
449 metagraph.object_graph_def().concrete_functions().at(function_name); in PartiallyReviveSavedModelObjects()
478 metagraph.object_graph_def().concrete_functions().at(function_name); in PartiallyReviveSavedModelObjects()
503 const SavedObjectGraph& obj_graph = metagraph.object_graph_def(); in PartiallyReviveSavedModelObjects()
|
| D | saved_model_utils.h | 100 Status PartiallyReviveSavedModelObjects(const MetaGraphDef& metagraph,
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | test_util.py | 49 metagraph = saver.export_meta_graph(graph_def=graph.as_graph_def())
50 return tf_optimizer.OptimizeGraph(config_proto, metagraph)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1395 TF_Buffer* metagraph = TF_NewBuffer(); in TEST() local
1400 opt, run_options, saved_model_dir.c_str(), tags, 1, graph, metagraph, s); in TEST()
1404 metagraph_def.ParseFromArray(metagraph->data, metagraph->length); in TEST()
1405 TF_DeleteBuffer(metagraph); in TEST()
|