| /external/tensorflow/tensorflow/python/saved_model/ |
| D | builder_impl.py | 133 def _tag_and_add_meta_graph(self, meta_graph_def, tags, signature_def_map): argument
148 if signature_def_map is not None:
149 for key in signature_def_map:
150 meta_graph_def.signature_def[key].CopyFrom(signature_def_map[key])
186 def _validate_signature_def_map(self, signature_def_map): argument
201 for signature_def_key in signature_def_map:
202 signature_def = signature_def_map[signature_def_key]
209 if constants.INIT_OP_SIGNATURE_KEY in signature_def_map:
213 if constants.TRAIN_OP_SIGNATURE_KEY in signature_def_map:
232 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 | 90 signature_def_map = graph.signature_def
91 self.assertEqual(1, len(signature_def_map))
93 list(signature_def_map.keys())[0])
95 signature_def = signature_def_map[
|
| D | load_v1_in_v2_test.py | 142 signature_def_map={
171 signature_def_map={
178 signature_def_map={
371 signature_def_map={},
392 signature_def_map={
421 signature_def_map={
465 signature_def_map={})
496 signature_def_map={})
679 signature_def_map={
|
| D | saved_model_test.py | 138 signature_def_map={"foo_key": foo_signature})
147 sess, ["foo"], signature_def_map={"foo_key": foo_signature})
159 signature_def_map={"foo_key": foo_signature})
168 sess, ["foo"], signature_def_map={"foo_key": foo_signature})
180 signature_def_map={invalid_key: foo_signature})
574 sess, ["foo"], signature_def_map={"foo_key": foo_signature})
588 signature_def_map={
|
| /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/tools/signature/ |
| D | signature_def_util_wrapper_pybind11.cc | 32 std::map<std::string, tensorflow::SignatureDef> signature_def_map; in WrappedSetSignatureDefMap() local
38 signature_def_map[entry.first] = signature_def; in WrappedSetSignatureDefMap()
40 auto status = tflite::SetSignatureDefMap(model, signature_def_map, &data); in WrappedSetSignatureDefMap()
56 std::map<std::string, tensorflow::SignatureDef> signature_def_map; in WrappedGetSignatureDefMap() local
57 auto status = tflite::GetSignatureDefMap(model, &signature_def_map); in WrappedGetSignatureDefMap()
62 for (const auto& entry : signature_def_map) { in WrappedGetSignatureDefMap()
|
| D | signature_def_utils_test.py | 57 signature_def_map = {serving_key: predict_signature_def}
59 tflite_model, signature_def_map)
68 signature_def_map = signature_def_utils.get_signature_defs(
70 self.assertIsNone(signature_def_map.get(serving_key),
|
| D | signature_def_util.cc | 68 const SignatureDefMap& signature_def_map, in SetSignatureDefMap() argument
73 if (signature_def_map.empty()) { in SetSignatureDefMap()
93 for (const auto& entry : signature_def_map) { in SetSignatureDefMap()
125 SignatureDefMap* signature_def_map) { in GetSignatureDefMap() argument
126 if (!model || !signature_def_map) { in GetSignatureDefMap()
146 *signature_def_map = retrieved_signature_def_map; in GetSignatureDefMap()
|
| D | signature_def_util.h | 45 const std::map<std::string, tensorflow::SignatureDef>& signature_def_map,
61 std::map<std::string, tensorflow::SignatureDef>* signature_def_map);
|
| D | signature_def_utils.py | 25 def set_signature_defs(tflite_model, signature_def_map): argument
45 k: v.SerializeToString() for k, v in signature_def_map.items()}
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model.py | 76 signature_def_map = meta_graph.signature_def
77 signature_def_keys = set(signature_def_map.keys())
85 return signature_def_map[signature_key]
|
| D | convert_saved_model_test.py | 229 signature_def_map = {
236 signature_def_map=signature_def_map)
|
| D | lite_v2_test.py | 842 signature_def_map, init_op, assets_collection = {
857 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/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | common_v1.py | 89 signature_def_map, init_op, assets_collection = create_signature()
96 signature_def_map,
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_aot_compile.py | 273 signature_def_map = meta_graph_def.signature_def
274 if signature_def_key not in signature_def_map:
279 list(signature_def_map.keys())))
280 signature_def = signature_def_map[signature_def_key]
|
| D | saved_model_cli.py | 91 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
94 for signature_def_key in sorted(signature_def_map.keys()):
302 signature_def_map = get_signature_def_map(saved_model_dir, tag_set)
303 for signature_def_key in sorted(signature_def_map.keys()):
|
| D | freeze_graph_test.py | 157 signature_def_map={
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/ |
| D | gen_tftrt_model.py | 122 signature_def_map={
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | freeze.py | 202 signature_def_map={
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saved_model_experimental.py | 319 signature_def_map=_create_signature_def_map(clone, mode),
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 782 signature_def_map=self._grappler_meta_graph_def.signature_def)
|
| D | trt_convert_test.py | 173 signature_def_map={_SAVED_MODEL_SIGNATURE_KEY: signature_def})
|