Home
last modified time | relevance | path

Searched refs:signature_defs (Results 1 – 25 of 48) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_util.cc55 const auto signature_defs = in ReadSignatureDefMap() local
58 for (int i = 0; i < signature_defs.Keys().size(); ++i) { in ReadSignatureDefMap()
59 const std::string key = signature_defs.Keys()[i].AsString().c_str(); in ReadSignatureDefMap()
60 (*map)[key] = signature_defs[key].AsString().c_str(); in ReadSignatureDefMap()
116 SerializedSignatureDefMap signature_defs; in HasSignatureDef() local
117 if (ReadSignatureDefMap(model, metadata, &signature_defs) != in HasSignatureDef()
121 return (signature_defs.find(signature_key) != signature_defs.end()); in HasSignatureDef()
132 SerializedSignatureDefMap signature_defs; in GetSignatureDefMap() local
133 auto status = ReadSignatureDefMap(model, metadata, &signature_defs); in GetSignatureDefMap()
138 for (const auto& entry : signature_defs) { in GetSignatureDefMap()
DBUILD1 # Utilities for signature_defs in TFLite
/external/tensorflow/tensorflow/python/saved_model/model_utils/
Dexport_test.py58 signature_defs = export_utils.build_all_signature_defs(
73 self.assertDictEqual(expected_signature_defs, signature_defs)
98 signature_defs = export_utils.build_all_signature_defs(
121 self.assertDictEqual(expected_signature_defs, signature_defs)
152 signature_defs = export_utils.build_all_signature_defs(
180 self.assertDictEqual(expected_signature_defs, signature_defs)
239 signature_defs = export_utils.build_all_signature_defs(
248 self.assertDictEqual(expected_signature_defs, signature_defs)
250 signature_defs = export_utils.build_all_signature_defs(
259 self.assertDictEqual(expected_signature_defs, signature_defs)
/external/tensorflow/tensorflow/lite/
Dinterpreter_builder.cc497 std::vector<Interpreter::SignatureDef> signature_defs; in ParseSignatureDefs() local
498 signature_defs.reserve(signature_def_list->size()); in ParseSignatureDefs()
521 signature_defs.resize(signature_defs.size() + 1); in ParseSignatureDefs()
522 auto& signature_def = signature_defs.back(); in ParseSignatureDefs()
530 interpreter->SetSignatureDef(std::move(signature_defs)); in ParseSignatureDefs()
773 if (ParseSignatureDefs(model_->signature_defs(), interpreter->get()) != in operator ()()
Dinterpreter.h718 void SetSignatureDef(std::vector<SignatureDef> signature_defs) { in SetSignatureDef() argument
719 signature_defs_ = std::move(signature_defs); in SetSignatureDef()
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py190 signature_defs = interpreter._get_full_signature_list()
191 if signature_def_name not in signature_defs:
193 self._signature_def = signature_defs[signature_def_name]
Dlite_v2_test.py989 signature_defs = interpreter.get_signature_list()
999 self.assertEqual(len(signature_defs), 1)
1000 self.assertEqual(list(signature_defs.keys()), ['mul_add'])
1001 self.assertEqual(len(signature_defs.values()), 1)
1003 list(signature_defs['mul_add'].keys()), ['inputs', 'outputs'])
1004 self.assertCountEqual(signature_defs['mul_add']['inputs'], ['x', 'y'])
1005 self.assertEqual(list(signature_defs['mul_add']['outputs']), ['output_0'])
1029 signature_defs = interpreter.get_signature_list()
1039 self.assertEqual(len(signature_defs), 1)
1040 self.assertEqual(list(signature_defs.keys()), ['mul_add'])
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dmetadata.mlir36 // CHECK-NEXT: signature_defs: [ ]
Dvariant_type_on_func.mlir33 // CHECK-NEXT: signature_defs: [ ]
Dvariant_type_on_op.mlir33 // CHECK-NEXT: signature_defs: [ ]
Dflex_exclusively.mlir50 // CHECK-NEXT: signature_defs: [ ]
Dnn.mlir58 // CHECK-NEXT: signature_defs: [ ]
Dnumeric_verify.mlir60 // CHECK-NEXT: signature_defs: [ ]
Dflex_op_with_f64.mlir63 // CHECK-NEXT: signature_defs: [ ]
Dreshape.mlir62 // CHECK-NEXT: signature_defs: [ ]
Dfake_quant.mlir57 // CHECK-NEXT: signature_defs: [ ]
Dflex_op_with_complex128.mlir63 // CHECK-NEXT: signature_defs: [ ]
Dfully_connected.mlir72 // CHECK-NEXT: signature_defs: [ ]
Dtranspose_conv_optional.mlir73 // CHECK-NEXT: signature_defs: [ ]
Dfully_connected_v2.mlir72 // CHECK-NEXT: signature_defs: [ ]
Dmul_v2.mlir69 // CHECK-NEXT: signature_defs: [ ]
Dmul_v3.mlir69 // CHECK-NEXT: signature_defs: [ ]
Dlogical.mlir90 // CHECK-NEXT: signature_defs: [ ]
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc283 const auto& signature_defs = interpreter->signature_def_names(); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureDefNames() local
285 signature_defs.size(), string_class, env->NewStringUTF(""))); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureDefNames()
286 for (int i = 0; i < signature_defs.size(); ++i) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureDefNames()
288 env->NewStringUTF(signature_defs[i]->c_str())); in Java_org_tensorflow_lite_NativeInterpreterWrapper_getSignatureDefNames()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dadd.pbtxt100 # CHECK-NEXT: signature_defs: [ ]

12