| /external/tensorflow/tensorflow/python/saved_model/ |
| simple_save.py | 74 signature_def_map = {
82 signature_def_map=signature_def_map,
|
| builder_impl.py | 68 signature_def_map=foo_signatures,
171 def _tag_and_add_meta_graph(self, meta_graph_def, tags, signature_def_map):
180 signature_def_map: The map of signature defs to be added to the meta graph
186 if signature_def_map is not None:
187 for key in signature_def_map:
188 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
218 def _validate_signature_def_map(self, signature_def_map):
226 signature_def_map: The map of signature defs to be validated.
228 if signature_def_map is not None:
229 for signature_def_key in signature_def_map
[all...] |
| simple_save_test.py | 83 # Check that the appropriate signature_def_map is created with the
85 signature_def_map = graph.signature_def
86 self.assertEqual(1, len(signature_def_map))
88 list(signature_def_map.keys())[0])
90 signature_def = signature_def_map[
|
| saved_model_test.py | 108 signature_def_map={"foo_key": foo_signature})
120 signature_def_map={"foo_key": foo_signature})
493 sess, ["foo"], signature_def_map={"foo_key": foo_signature})
509 signature_def_map={
[all...] |
| /external/tensorflow/tensorflow/python/estimator/export/ |
| export.py | 201 signature_def_map = {}
207 signature_def_map[signature_name] = signature
222 signature_def_map[signature_name] = signature
226 _log_signature_report(signature_def_map, excluded_signatures)
232 return {k: v for k, v in signature_def_map.items()
243 def _log_signature_report(signature_def_map, excluded_signatures):
253 for signature_name, sig in signature_def_map.items():
269 if not signature_def_map:
272 not in signature_def_map):
|
| /external/tensorflow/tensorflow/examples/saved_model/ |
| saved_model_half_plus_two.py | 174 signature_def_map = {
188 signature_def_map = {
205 signature_def_map=signature_def_map,
212 signature_def_map=signature_def_map,
|
| /external/tensorflow/tensorflow/contrib/saved_model/python/saved_model/ |
| signature_def_utils_test.py | 34 def _add_to_signature_def_map(self, meta_graph_def, signature_def_map=None):
35 if signature_def_map is not None:
36 for key in signature_def_map:
37 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| loader_test.cc | 71 const auto signature_def_map = bundle.meta_graph_def.signature_def(); local
72 const auto signature_def = signature_def_map.at("regress_x_to_y");
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| bundle_shim_test.cc | 73 const auto& signature_def_map = meta_graph_def.signature_def(); local
75 const auto& regression_entry = signature_def_map.find(signature_def_key);
76 ASSERT_FALSE(regression_entry == signature_def_map.end());
525 const auto& signature_def_map = meta_graph_def.signature_def(); local
527 for (const auto& entry : signature_def_map) {
|
| /external/tensorflow/tensorflow/python/tools/ |
| saved_model_cli.py | 73 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
76 for signature_def_key in sorted(signature_def_map.keys()):
186 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
187 for signature_def_key in sorted(signature_def_map.keys()):
|
| freeze_graph_test.py | 154 signature_def_map={
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| saved_model_export_utils.py | 253 signature_def_map = {('%s:%s' % (input_key, output_key or 'None')):
268 signature_def_map[signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY] = (
272 return signature_def_map
[all...] |
| /external/tensorflow/tensorflow/python/estimator/ |
| estimator.py | 607 signature_def_map = build_all_signature_defs(
638 signature_def_map=signature_def_map,
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| estimator.py | [all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_test.cc | 1353 const auto signature_def_map = metagraph_def.signature_def(); local
[all...] |
