| /external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/ |
| signature_def_utils.py | 22 def get_signature_def_by_key(meta_graph_def, signature_def_key):
26 meta_graph_def: MetaGraphDef protocol buffer with the SignatureDefMap to
39 if signature_def_key not in meta_graph_def.signature_def:
42 return meta_graph_def.signature_def[signature_def_key]
|
| signature_def_utils_test.py | 34 def _add_to_signature_def_map(self, meta_graph_def, signature_def_map=None):
37 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
56 meta_graph_def = meta_graph_pb2.MetaGraphDef()
58 meta_graph_def, {"foo": foo_signature_def,
67 meta_graph_def, missing_key)
71 meta_graph_def, "foo")
84 meta_graph_def, "bar")
99 meta_graph_def = meta_graph_pb2.MetaGraphDef()
100 self._add_to_signature_def_map(meta_graph_def, {
107 meta_graph_def, "my_regression"
[all...] |
| reader.py | 90 for meta_graph_def in saved_model.meta_graphs:
91 all_tags.append(list(meta_graph_def.meta_info_def.tags))
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| signature.h | 39 Status GetSignatures(const tensorflow::MetaGraphDef& meta_graph_def,
44 tensorflow::MetaGraphDef* meta_graph_def);
50 const tensorflow::MetaGraphDef& meta_graph_def,
57 const string& name, const tensorflow::MetaGraphDef& meta_graph_def,
63 Status GetRegressionSignature(const tensorflow::MetaGraphDef& meta_graph_def,
89 const tensorflow::MetaGraphDef& meta_graph_def,
93 Status GetDefaultSignature(const tensorflow::MetaGraphDef& meta_graph_def,
99 const tensorflow::MetaGraphDef& meta_graph_def,
|
| bundle_shim_test.cc | 72 const MetaGraphDef meta_graph_def = saved_model_bundle.meta_graph_def; local
73 const auto& signature_def_map = meta_graph_def.signature_def();
100 void ValidateSignatureDef(const MetaGraphDef& meta_graph_def, const string& key,
105 auto iter = meta_graph_def.signature_def().find(key);
106 ASSERT_TRUE(iter != meta_graph_def.signature_def().end());
144 MetaGraphDef meta_graph_def; local
146 TF_EXPECT_OK(ConvertSignaturesToSignatureDefs(&meta_graph_def));
147 EXPECT_EQ(0, meta_graph_def.signature_def_size());
155 MetaGraphDef meta_graph_def; local
171 MetaGraphDef meta_graph_def; local
199 MetaGraphDef meta_graph_def; local
234 MetaGraphDef meta_graph_def; local
258 MetaGraphDef meta_graph_def; local
312 MetaGraphDef meta_graph_def; local
371 MetaGraphDef meta_graph_def; local
410 MetaGraphDef meta_graph_def; local
431 MetaGraphDef meta_graph_def; local
470 MetaGraphDef meta_graph_def; local
524 const MetaGraphDef meta_graph_def = saved_model_bundle.meta_graph_def; local
[all...] |
| bundle_shim.cc | 82 MetaGraphDef* meta_graph_def) {
86 if (!GetNamedSignature(kPredictInputs, *meta_graph_def, &input_signature)
88 !GetNamedSignature(kPredictOutputs, *meta_graph_def, &output_signature)
120 meta_graph_def->signature_def().find(kDefaultServingSignatureDefKey) !=
121 meta_graph_def->signature_def().end();
126 (*meta_graph_def->mutable_signature_def())[signature_def_key] = signature_def;
163 MetaGraphDef* meta_graph_def) {
168 meta_graph_def->signature_def().find(kDefaultServingSignatureDefKey) !=
169 meta_graph_def->signature_def().end();
178 (*meta_graph_def->mutable_signature_def())[kDefaultServingSignatureDefKey]
[all...] |
| session_bundle.py | 59 meta_graph_def=None):
70 meta_graph_def: optional object of type MetaGraphDef. If this object is
81 if not meta_graph_def:
88 meta_graph_def = meta_graph_pb2.MetaGraphDef()
89 meta_graph_def.ParseFromString(
130 collection_def = meta_graph_def.collection_def
137 meta_graph_def)
141 meta_graph_def.graph_def.CopyFrom(graph_def)
146 saver = saver_lib.import_meta_graph(meta_graph_def)
156 meta_graph_def)
[all...] |
| signature_test.cc | 43 tensorflow::MetaGraphDef meta_graph_def; local
49 (*meta_graph_def.mutable_collection_def())[kSignaturesKey]
55 const Status status = GetClassificationSignature(meta_graph_def, &signature);
61 tensorflow::MetaGraphDef meta_graph_def; local
64 (*meta_graph_def.mutable_collection_def())[kSignaturesKey]
70 const Status status = GetClassificationSignature(meta_graph_def, &signature);
78 tensorflow::MetaGraphDef meta_graph_def; local
81 (*meta_graph_def.mutable_collection_def())[kSignaturesKey]
87 const Status status = GetClassificationSignature(meta_graph_def, &signature);
95 tensorflow::MetaGraphDef meta_graph_def; local
114 tensorflow::MetaGraphDef meta_graph_def; local
131 tensorflow::MetaGraphDef meta_graph_def; local
151 tensorflow::MetaGraphDef meta_graph_def; local
168 tensorflow::MetaGraphDef meta_graph_def; local
185 tensorflow::MetaGraphDef meta_graph_def; local
202 tensorflow::MetaGraphDef meta_graph_def; local
220 tensorflow::MetaGraphDef meta_graph_def; local
469 tensorflow::MetaGraphDef meta_graph_def; local
486 tensorflow::MetaGraphDef meta_graph_def; local
496 tensorflow::MetaGraphDef meta_graph_def; local
512 tensorflow::MetaGraphDef meta_graph_def; local
527 tensorflow::MetaGraphDef meta_graph_def; local
551 tensorflow::MetaGraphDef meta_graph_def; local
597 tensorflow::MetaGraphDef meta_graph_def; local
614 tensorflow::MetaGraphDef meta_graph_def; local
631 tensorflow::MetaGraphDef meta_graph_def; local
[all...] |
| signature.cc | 49 Status GetSignatures(const tensorflow::MetaGraphDef& meta_graph_def,
51 const auto& collection_def = meta_graph_def.collection_def();
56 DebugStringIfAvailable(meta_graph_def)));
63 tensorflow::MetaGraphDef* meta_graph_def) {
64 auto& collection_def = *(meta_graph_def->mutable_collection_def());
77 const tensorflow::MetaGraphDef& meta_graph_def,
80 TF_RETURN_IF_ERROR(GetSignatures(meta_graph_def, &signatures));
96 const string& name, const tensorflow::MetaGraphDef& meta_graph_def,
99 TF_RETURN_IF_ERROR(GetSignatures(meta_graph_def, &signatures));
152 Status GetRegressionSignature(const tensorflow::MetaGraphDef& meta_graph_def,
[all...] |
| bundle_shim.h | 53 Status ConvertSignaturesToSignatureDefs(MetaGraphDef* meta_graph_def);
|
| session_bundle.cc | 66 MetaGraphDef* meta_graph_def) {
69 return ReadBinaryProto(Env::Default(), meta_graph_def_path, meta_graph_def);
165 GetMetaGraphDefFromExport(export_dir, &(bundle->meta_graph_def)));
171 AddDefaultAttrsToGraphDef(bundle->meta_graph_def.mutable_graph_def(),
174 const auto& collection_def_map = bundle->meta_graph_def.collection_def();
175 const auto graph_it = bundle->meta_graph_def.collection_def().find(kGraphKey);
190 const GraphDef& graph_def = bundle->meta_graph_def.graph_def();
209 bundle->meta_graph_def.saver_def().restore_op_name(),
210 bundle->meta_graph_def.saver_def().filename_tensor_name(),
|
| session_bundle.h | 47 MetaGraphDef meta_graph_def; member in struct:tensorflow::serving::SessionBundle
|
| /external/tensorflow/tensorflow/python/tools/ |
| saved_model_utils.py | 44 for meta_graph_def in saved_model.meta_graphs:
45 if set(meta_graph_def.meta_info_def.tags) == set_of_tags:
46 return meta_graph_def
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| loader.cc | 77 for (const MetaGraphDef& meta_graph_def : saved_model_proto.meta_graphs()) {
78 // Get tags from the meta_graph_def.
80 for (const string& tag : meta_graph_def.meta_info_def().tags()) {
85 *meta_graph_def_to_load = meta_graph_def;
96 Status LoadMetaGraphIntoSession(const MetaGraphDef& meta_graph_def,
102 return (*session)->Create(meta_graph_def.graph_def());
125 bool HasMainOp(const MetaGraphDef& meta_graph_def) {
126 const auto& collection_def_map = meta_graph_def.collection_def();
135 const MetaGraphDef& meta_graph_def,
139 const auto& collection_def_map = meta_graph_def.collection_def()
[all...] |
| loader.h | 33 MetaGraphDef meta_graph_def; member in struct:tensorflow::SavedModelBundle
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform.py | 459 def _add_pruned_collection(base_meta_graph_def, meta_graph_def,
464 collection = meta_graph_def.collection_def[collection_name]
484 def _add_pruned_saver(base_meta_graph_def, meta_graph_def, removed_op_names):
494 meta_graph_def: The transformed MetaGraphDef being built.
498 # Note this does surgery on meta_graph_def.graph_def too, so that should have
523 # save_tensor_names_op = _find_op(meta_graph_def.graph_def,
531 # meta_graph_def.graph_def, restore_tensor_names_op_name)
540 meta_graph_def.saver_def.CopyFrom(base_meta_graph_def.saver_def)
551 def _add_pruned_signature(base_meta_graph_def, meta_graph_def,
561 meta_graph_def: The transformed MetaGraphDef being built
[all...] |
| meta_graph_transform_test.py | 346 meta_graph_def = meta_graph_pb2.MetaGraphDef()
351 meta_graph_def.collection_def[
358 meta_graph_transform._add_new_inits_to_collection(meta_graph_def,
361 self.assertEqual(meta_graph_def.collection_def[
456 meta_graph_def = meta_graph_pb2.MetaGraphDef()
459 base_meta_graph_def, meta_graph_def, collection_name, removed_op_names)
461 collection = meta_graph_def.collection_def[collection_name]
472 meta_graph_def = meta_graph_pb2.MetaGraphDef()
475 base_meta_graph_def, meta_graph_def, collection_name, removed_op_names)
477 collection = meta_graph_def.collection_def[collection_name
[all...] |
| /external/tensorflow/tensorflow/contrib/saved_model/cc/saved_model/ |
| signature_def_utils_test.cc | 58 const MetaGraphDef meta_graph_def = MakeSampleMetaGraphDef(); local
62 FindSignatureDefByKey(meta_graph_def, kSignatureKey, &signature_def));
66 FindSignatureDefByKey(meta_graph_def, "nonexistent", &signature_def)
|
| signature_def_utils.h | 31 // Finds the entry in meta_graph_def.signature_def with the given key, or
33 // SignatureDef* points into meta_graph_def and may be invalidated by changes
35 Status FindSignatureDefByKey(const MetaGraphDef& meta_graph_def,
|
| signature_def_utils.cc | 38 Status FindSignatureDefByKey(const MetaGraphDef& meta_graph_def,
41 return FindInProtobufMap("SignatureDef", meta_graph_def.signature_def(),
|
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph.py | 377 def add_collection_def(meta_graph_def, key, graph=None,
383 meta_graph_def: MetaGraphDef protocol buffer.
415 col_def = meta_graph_def.collection_def[key]
445 if key in meta_graph_def.collection_def:
446 del meta_graph_def.collection_def[key]
464 def _strip_graph_default_valued_attrs(meta_graph_def):
471 meta_graph_def: `MetaGraphDef` protocol buffer
478 for function_def in meta_graph_def.graph_def.library.function:
499 for node_def in meta_graph_def.graph_def.node:
503 for function_def in meta_graph_def.graph_def.library.function
[all...] |
| meta_graph_test.py | 89 meta_graph_def, var_list = meta_graph.export_scoped_meta_graph(
94 self.assertTrue(meta_graph_def.HasField("meta_info_def"))
95 self.assertNotEqual(meta_graph_def.meta_info_def.tensorflow_version, "")
96 self.assertNotEqual(meta_graph_def.meta_info_def.tensorflow_git_version,
109 test_util.assert_meta_graph_protos_equal(self, meta_graph_def,
179 meta_graph_def, _ = meta_graph.export_scoped_meta_graph(
183 meta_graph_def.graph_def)
186 self.assertTrue(meta_graph_def.meta_info_def.stripped_default_attrs)
189 meta_graph_def, _ = meta_graph.export_scoped_meta_graph(
193 meta_graph_def.graph_def
[all...] |
| /external/tensorflow/tensorflow/python/saved_model/ |
| builder_impl.py | 171 def _tag_and_add_meta_graph(self, meta_graph_def, tags, signature_def_map):
178 meta_graph_def: The meta graph def to add to the SavedModel.
184 meta_graph_def.meta_info_def.tags.append(tag)
188 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
191 proto_meta_graph_def.CopyFrom(meta_graph_def)
310 meta_graph_def = saver.export_meta_graph(
314 self._tag_and_add_meta_graph(meta_graph_def, tags, signature_def_map)
406 meta_graph_def = saver.export_meta_graph(
410 self._tag_and_add_meta_graph(meta_graph_def, tags, signature_def_map)
|
| /external/tensorflow/tensorflow/python/summary/writer/ |
| writer.py | 83 # Also export the meta_graph_def in this case.
227 def add_meta_graph(self, meta_graph_def, global_step=None):
234 meta_graph_def: A `MetaGraphDef` object, often as returned by
240 TypeError: If both `meta_graph_def` is not an instance of `MetaGraphDef`.
242 if not isinstance(meta_graph_def, meta_graph_pb2.MetaGraphDef):
243 raise TypeError("meta_graph_def must be type MetaGraphDef, saw type: %s" %
244 type(meta_graph_def))
245 meta_graph_bytes = meta_graph_def.SerializeToString()
246 event = event_pb2.Event(meta_graph_def=meta_graph_bytes)
|
| /external/tensorflow/tensorflow/contrib/predictor/ |
| saved_model_predictor.py | 56 for meta_graph_def in saved_model.meta_graphs:
57 if set(meta_graph_def.meta_info_def.tags) == set_of_tags:
58 return meta_graph_def
|
