Home
last modified time | relevance | path

Searched refs:meta_graph (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc39 MetaGraphDef meta_graph; in TEST_F() local
52 TF_CHECK_OK(s.ToGraphDef(meta_graph.mutable_graph_def())); in TEST_F()
70 (*meta_graph.mutable_collection_def())["saved_model_assets"] in TEST_F()
74 *((*meta_graph.mutable_collection_def())["train_op"] in TEST_F()
82 GrapplerItemFromMetaGraphDef("0", meta_graph, cfg); in TEST_F()
98 MetaGraphDef meta_graph; in TEST_F() local
116 TF_CHECK_OK(s.ToGraphDef(meta_graph.mutable_graph_def())); in TEST_F()
144 (*meta_graph.mutable_collection_def())["saved_model_assets"] in TEST_F()
148 (*meta_graph.mutable_collection_def())["saved_model_assets"] in TEST_F()
153 *((*meta_graph.mutable_collection_def())["train_op"] in TEST_F()
[all …]
Dgrappler_item_builder.cc291 const string& id, const MetaGraphDef& meta_graph, const ItemConfig& cfg) { in GrapplerItemFromMetaGraphDef() argument
298 new_item->graph = meta_graph.graph_def(); in GrapplerItemFromMetaGraphDef()
311 meta_graph.collection_def().count("train_op") > 0) { in GrapplerItemFromMetaGraphDef()
312 const CollectionDef& nodes = meta_graph.collection_def().at("train_op"); in GrapplerItemFromMetaGraphDef()
324 for (const auto& name_and_signature : meta_graph.signature_def()) { in GrapplerItemFromMetaGraphDef()
442 if (meta_graph.collection_def().count(var_collection) == 0) { in GrapplerItemFromMetaGraphDef()
445 const CollectionDef& vars = meta_graph.collection_def().at(var_collection); in GrapplerItemFromMetaGraphDef()
455 if (meta_graph.collection_def().count("table_initializer") > 0) { in GrapplerItemFromMetaGraphDef()
457 meta_graph.collection_def().at("table_initializer"); in GrapplerItemFromMetaGraphDef()
479 if (meta_graph.collection_def().count("saved_model_assets") > 0) { in GrapplerItemFromMetaGraphDef()
[all …]
Dgrappler_item_builder.h73 const string& id, const MetaGraphDef& meta_graph, const ItemConfig& cfg);
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py33 from tensorflow.python.framework import meta_graph
90 meta_graph_def, var_list = meta_graph.export_scoped_meta_graph(
105 meta_graph.import_scoped_meta_graph(filename)
108 new_meta_graph_def, _ = meta_graph.export_scoped_meta_graph(filename +
135 op_list = meta_graph.stripped_op_list_for_graph(ops.get_default_graph()
141 op_list = meta_graph.stripped_op_list_for_graph(ops.get_default_graph()
161 op_list = meta_graph.stripped_op_list_for_graph(graph)
182 op_list = meta_graph.stripped_op_list_for_graph(graph)
205 meta_graph_def, _ = meta_graph.export_scoped_meta_graph(
215 meta_graph_def, _ = meta_graph.export_scoped_meta_graph(
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dcost_analyzer_tool.py47 meta_graph = saved_model.meta_graphs[0]
51 meta_graph = saved_model.meta_graphs[0]
54 meta_graph = meta_graph_pb2.MetaGraphDef()
55 text_format.Merge(input_data, meta_graph)
58 meta_graph.ParseFromString(input_data)
70 meta_graph = saver.export_meta_graph(
76 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
77 return meta_graph
Ddatasets_test.py26 from tensorflow.python.framework import meta_graph
55 mg = meta_graph.create_meta_graph_def(graph=g)
80 mg = meta_graph.create_meta_graph_def(graph=g)
116 mg = meta_graph.create_meta_graph_def(graph=g)
129 mg = meta_graph.create_meta_graph_def(graph=g)
156 mg = meta_graph.create_meta_graph_def(graph=g)
260 mg = meta_graph.create_meta_graph_def(graph=g)
289 mg = meta_graph.create_meta_graph_def(graph=g)
326 mg = meta_graph.create_meta_graph_def(graph=g)
361 mg = meta_graph.create_meta_graph_def(graph=g)
[all …]
Ditem_test.py24 from tensorflow.python.framework import meta_graph
43 mg = meta_graph.create_meta_graph_def(graph=g)
57 mg = meta_graph.create_meta_graph_def(graph=g)
70 mg = meta_graph.create_meta_graph_def(graph=g)
92 mg = meta_graph.create_meta_graph_def(graph=g)
120 mg = meta_graph.create_meta_graph_def(graph=g)
Dcluster_test.py22 from tensorflow.python.framework import meta_graph
41 mg = meta_graph.create_meta_graph_def(graph=g)
58 mg = meta_graph.create_meta_graph_def(graph=g)
76 mg = meta_graph.create_meta_graph_def(graph=g)
96 mg = meta_graph.create_meta_graph_def(graph=g)
123 mg = meta_graph.create_meta_graph_def(graph=g)
149 mg = meta_graph.create_meta_graph_def(graph=g)
Dmodel_analyzer_test.py22 from tensorflow.python.framework import meta_graph
41 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
62 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
Dcost_analyzer_test.py25 from tensorflow.python.framework import meta_graph
50 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
75 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
106 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
145 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
Dtf_optimizer_test.py24 from tensorflow.python.framework import meta_graph
49 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
78 mg = meta_graph.create_meta_graph_def(graph=g)
123 mg = meta_graph.create_meta_graph_def(graph=g)
Dmemory_optimizer_test.py25 from tensorflow.python.framework import meta_graph
50 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
79 mg = meta_graph.create_meta_graph_def(graph=ops.get_default_graph())
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py63 def get_signature_def(meta_graph, signature_key): argument
76 signature_def_map = meta_graph.signature_def
187 meta_graph = get_meta_graph_def(saved_model_dir, tag_set)
188 signature_def = get_signature_def(meta_graph, signature_key)
192 collection_def = meta_graph.collection_def
198 loader.load(sess, meta_graph.meta_info_def.tags, saved_model_dir)
Dlite.py529 meta_graph = saved_model_proto.meta_graphs[0]
530 for function in meta_graph.graph_def.library.function:
741 meta_graph = saved_model.get_meta_graph_def_from_tags(
746 signature_def = meta_graph.signature_def[
769 _build_debug_info_func(self._funcs[0].graph), meta_graph.graph_def)
773 meta_graph.graph_def)
778 meta_graph.graph_def)
846 meta_graph = saved_model.get_meta_graph_def_from_tags(
848 signature_def = meta_graph.signature_def[
861 self).convert(meta_graph.graph_def, input_tensors,
Dutil.py267 meta_graph = _export_meta_graph(graph_def=graph_def, graph=graph)
280 meta_graph.signature_def["not_used_key"].CopyFrom(signature)
287 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
289 return tf_optimizer.OptimizeGraph(config, meta_graph)
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc243 const MetaGraphDef& meta_graph, in LoadMetagraphIntoSession() argument
248 TF_RETURN_IF_ERROR(ValidateSavedTensors(meta_graph.graph_def())); in LoadMetagraphIntoSession()
249 return (*session)->Create(meta_graph.graph_def()); in LoadMetagraphIntoSession()
402 const MetaGraphDef& meta_graph, const string& export_dir, in RestoreSession() argument
406 TF_RETURN_IF_ERROR(internal::GetAssetFileDefs(meta_graph, &asset_file_defs)); in RestoreSession()
407 if (meta_graph.has_saver_def()) { in RestoreSession()
409 meta_graph.saver_def().restore_op_name(), in RestoreSession()
410 meta_graph.saver_def().filename_tensor_name(), in RestoreSession()
421 internal::GetInitOp(export_dir, meta_graph, &init_op_name)); in RestoreSession()
422 TF_RETURN_IF_ERROR(RunInitOp(run_options, export_dir, meta_graph, in RestoreSession()
Dloader.h104 const MetaGraphDef& meta_graph, const string& export_dir,
111 const MetaGraphDef& meta_graph,
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Dmodel_handler.py120 meta_graph = saved_model_loader.load(
127 meta_graph.signature_def[saved_model_signature_key].outputs.values()
131 sess, meta_graph.graph_def, output_node_names))
132 meta_graph.graph_def.CopyFrom(graph_def)
133 return meta_graph
265 def meta_graph(self) -> meta_graph_pb2.MetaGraphDef: member in ModelHandlerV1
273 return self.meta_graph.signature_def[
278 return self.meta_graph.signature_def[
310 importer.import_graph_def(self.meta_graph.graph_def, name="")
496 self._check_conversion(self.meta_graph.graph_def)
/external/tensorflow/tensorflow/python/keras/tests/
Dgraph_util_test.py72 meta_graph = export_meta_graph(graph_def=graph_def)
76 meta_graph.collection_def["train_op"].CopyFrom(fetch_collection)
83 return tf_optimizer.OptimizeGraph(config, meta_graph)
/external/tensorflow/tensorflow/core/protobuf/
DBUILD52 "meta_graph",
152 "meta_graph.proto",
182 "meta_graph.proto",
Dsaved_model.proto5 import "tensorflow/core/protobuf/meta_graph.proto";
/external/tensorflow/tensorflow/python/saved_model/
Dloader_test.py218 meta_graph = loader.get_meta_graph_def_from_tags(["foo_graph"])
219 self.assertIsNotNone(meta_graph)
220 self.assertIn("foo", meta_graph.signature_def)
221 self.assertIn("bar", meta_graph.signature_def)
/external/tensorflow/tensorflow/python/summary/
Dsummary_test.py31 from tensorflow.python.framework import meta_graph
228 metagraph_def, _ = meta_graph.export_scoped_meta_graph(export_scope='outer')
231 meta_graph.import_scoped_meta_graph(metagraph_def, graph=g,
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint.py32 from tensorflow.python.framework import meta_graph
122 meta_graph_def = meta_graph.create_meta_graph_def(
/external/tensorflow/tensorflow/python/summary/writer/
Dwriter.py30 from tensorflow.python.framework import meta_graph
90 meta_graph.create_meta_graph_def(graph_def=graph_def or

12