| /external/tensorflow/tensorflow/python/saved_model/ |
| D | builder_impl.py | 132 def _tag_and_add_meta_graph(self, meta_graph_def, tags, signature_def_map): argument
147 if signature_def_map is not None:
148 for key in signature_def_map:
149 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
182 def _validate_signature_def_map(self, signature_def_map): argument
197 for signature_def_key in signature_def_map:
198 signature_def = signature_def_map[signature_def_key]
205 if constants.INIT_OP_SIGNATURE_KEY in signature_def_map:
209 if constants.TRAIN_OP_SIGNATURE_KEY in signature_def_map:
228 signature_def_map=None, argument
[all …]
|
| D | simple_save.py | 79 signature_def_map = {
87 signature_def_map=signature_def_map,
|
| D | simple_save_test.py | 87 signature_def_map = graph.signature_def
88 self.assertEqual(1, len(signature_def_map))
90 list(signature_def_map.keys())[0])
92 signature_def = signature_def_map[
|
| D | load_v1_in_v2_test.py | 119 signature_def_map={
126 signature_def_map={
|
| D | saved_model_test.py | 114 signature_def_map={"foo_key": foo_signature})
125 signature_def_map={"foo_key": foo_signature})
137 signature_def_map={"foo_key": foo_signature})
147 signature_def_map={"foo_key": foo_signature})
159 signature_def_map={invalid_key: foo_signature})
541 sess, ["foo"], signature_def_map={"foo_key": foo_signature})
557 signature_def_map={
|
| D | README.md | 117 signature_def_map=foo_signatures,
|
| /external/tensorflow/tensorflow/python/saved_model/model_utils/ |
| D | export_utils.py | 99 signature_def_map = {}
105 signature_def_map[signature_name] = signature
121 signature_def_map[signature_name] = signature
125 _log_signature_report(signature_def_map, excluded_signatures)
135 signature_def_map = {
137 for k, v in signature_def_map.items()
140 return signature_def_map
152 def _log_signature_report(signature_def_map, excluded_signatures): argument
162 for signature_name, sig in signature_def_map.items():
178 if not signature_def_map:
[all …]
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 77 signature_def_map = meta_graph.signature_def
78 signature_def_keys = set(signature_def_map.keys())
86 return signature_def_map[signature_key]
|
| D | convert_saved_model_test.py | 306 signature_def_map = {
313 signature_def_map=signature_def_map)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.saved_model.builder.-saved-model-builder.pbtxt | 12 …argspec: "args=[\'self\', \'tags\', \'signature_def_map\', \'assets_collection\', \'legacy_init_op…
16 …argspec: "args=[\'self\', \'sess\', \'tags\', \'signature_def_map\', \'assets_collection\', \'lega…
|
| D | tensorflow.saved_model.-builder.pbtxt | 12 …argspec: "args=[\'self\', \'tags\', \'signature_def_map\', \'assets_collection\', \'legacy_init_op…
16 …argspec: "args=[\'self\', \'sess\', \'tags\', \'signature_def_map\', \'assets_collection\', \'lega…
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | bundle_shim_test.cc | 77 const auto& signature_def_map = meta_graph_def.signature_def(); in LoadAndValidateSavedModelBundle() local
79 const auto& regression_entry = signature_def_map.find(signature_def_key); in LoadAndValidateSavedModelBundle()
80 ASSERT_FALSE(regression_entry == signature_def_map.end()); in LoadAndValidateSavedModelBundle()
530 const auto& signature_def_map = meta_graph_def.signature_def(); in TEST() local
532 for (const auto& entry : signature_def_map) { in TEST()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 76 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
79 for signature_def_key in sorted(signature_def_map.keys()):
198 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
199 for signature_def_key in sorted(signature_def_map.keys()):
|
| D | freeze_graph_test.py | 155 signature_def_map={
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | loader_test.cc | 74 const auto signature_def_map = bundle.meta_graph_def.signature_def(); in CheckSavedModelBundle() local
75 const auto signature_def = signature_def_map.at("regress_x_to_y"); in CheckSavedModelBundle()
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils.py | 262 signature_def_map = {('%s:%s' % (input_key, output_key or 'None')):
277 signature_def_map[signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY] = (
281 return signature_def_map
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | estimator.py | 1410 signature_def_map = saved_model_export_utils.build_all_signature_defs(
1438 signature_def_map=signature_def_map,
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saved_model.py | 297 signature_def_map=_create_signature_def_map(clone, mode),
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 166 signature_def_map={"mypredict": signature_def})
|
| D | trt_convert.py | 435 signature_def_map=self._grappler_meta_graph_def.signature_def)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 1374 const auto signature_def_map = metagraph_def.signature_def(); in TEST() local
1375 const auto signature_def = signature_def_map.at("regress_x_to_y"); in TEST()
|