| /cmsis-nn-latest/Tests/UnitTest/ |
| D | generate_test_data.py | 69 …ta_sets(regenerate_input, regenerate_weights, regenerate_biases, schema_file, interpreter) -> dict: argument
92 interpreter=interpreter)
109 interpreter=interpreter)
131 interpreter=interpreter)
151 interpreter=interpreter)
172 interpreter=interpreter)
192 interpreter=interpreter)
212 interpreter=interpreter)
231 interpreter=interpreter)
251 interpreter=interpreter)
[all …]
|
| D | softmax_settings.py | 40 interpreter="tensorflow"): argument
59 interpreter=interpreter)
138 interpreter = self.Interpreter(model_path=str(self.model_path_tflite),
140 interpreter.allocate_tensors()
141 all_layers_details = interpreter.get_tensor_details()
145 interpreter.set_tensor(input_layer["index"], tf.cast(input_data, tf.int8))
146 interpreter.invoke()
147 output_data = interpreter.get_tensor(output_layer["index"])
154 … interpreter = self.convert_and_interpret(model, inttype, tf.expand_dims(input_data, axis=0))
155 output_details = interpreter.get_output_details()
[all …]
|
| D | lstm_settings.py | 41 interpreter="tensorflow"): argument
60 interpreter=interpreter)
174 interpreter = self.convert_and_interpret(model, tf.int8, input_data, dataset_shape=shape)
176 all_layers_details = interpreter.get_tensor_details()
213 input_details = interpreter.get_input_details()
214 output_details = interpreter.get_output_details()
215 actual_input_data = interpreter.get_tensor(input_details[0]["index"])
220 …self.generate_c_array(self.input_data_file_prefix, interpreter.get_tensor(input_data_for_index['in…
221 self.generate_c_array("input_to_input_w", interpreter.get_tensor(input_to_input_w['index']))
222 … self.generate_c_array("input_to_forget_w", interpreter.get_tensor(input_to_forget_w['index']))
[all …]
|
| D | svdf_settings.py | 53 interpreter="tensorflow"):
72 interpreter=interpreter)
199 interpreter = self.Interpreter(model_path=str(self.model_path_tflite),
201 interpreter.allocate_tensors()
204 all_layers_details = interpreter.get_tensor_details()
221 self.generate_c_array("weights_feature", interpreter.get_tensor(weights_1_layer['index']))
222 …self.generate_c_array(self.bias_data_file_prefix, interpreter.get_tensor(bias_layer['index']), "in…
225 …self.generate_c_array("weights_time", interpreter.get_tensor(weights_2_layer['index']), datatype='…
226 self.generate_c_array("state", interpreter.get_tensor(state_layer['index']), "int8_t")
228 …self.generate_c_array("weights_time", interpreter.get_tensor(weights_2_layer['index']), datatype='…
[all …]
|
| D | pooling_settings.py | 49 interpreter="tensorflow"):
72 interpreter=interpreter)
106 interpreter = self.convert_and_interpret(model, inttype, input_data)
108 output_details = interpreter.get_output_details()
116 interpreter.invoke()
117 output_data = interpreter.get_tensor(output_details[0]["index"])
|
| D | add_mul_settings.py | 42 interpreter="tensorflow"): argument
66 interpreter=interpreter)
105 interpreter = self.convert_and_interpret(model, inttype_tf)
107 input_details = interpreter.get_input_details()
108 interpreter.set_tensor(input_details[0]["index"], tf.cast(input_data1, inttype_tf))
109 interpreter.set_tensor(input_details[1]["index"], tf.cast(input_data2, inttype_tf))
127 interpreter.invoke()
128 output_details = interpreter.get_output_details()
129 output_data = interpreter.get_tensor(output_details[0]["index"])
|
| D | fully_connected_settings.py | 51 interpreter="tensorflow",
86 interpreter=interpreter,
257 interpreter = self.interpret_model(input_data, inttype)
260 all_layers_details = interpreter.get_tensor_details()
264 if weights_size != interpreter.get_tensor(filter_layer['index']).size or \
265 … (self.generate_bias and biases.numpy().size != interpreter.get_tensor(bias_layer['index']).size):
279 output_details = interpreter.get_output_details()
280 interpreter.invoke()
281 output_data = interpreter.get_tensor(output_details[0]["index"])
286 …self.weight_data_file_prefix, interpreter.get_tensor(filter_layer['index']), pack=self.int4_weight…
[all …]
|
| D | test_settings.py | 86 interpreter="tensorflow",
98 if interpreter == "tflite_runtime":
99 from tflite_runtime.interpreter import Interpreter
100 from tflite_runtime.interpreter import OpResolverType
105 interpreter = "tflite_runtime"
107 elif interpreter == "tensorflow":
108 from tensorflow.lite.python.interpreter import Interpreter
109 from tensorflow.lite.python.interpreter import OpResolverType
113 interpreter = "tensorflow"
115 elif interpreter == "tflite_micro":
[all …]
|
| D | conv_settings.py | 57 interpreter="tensorflow",
88 interpreter=interpreter,
391 interpreter = self.interpret_model(input_data, inttype)
393 all_layers_details = interpreter.get_tensor_details()
403 expected_weight_size = math.ceil(interpreter.get_tensor(filter_layer['index']).size / 2)
405 expected_weight_size = interpreter.get_tensor(filter_layer['index']).size
408 … (self.generate_bias and biases.numpy().size != interpreter.get_tensor(bias_layer['index']).size):
411 output_details = interpreter.get_output_details()
423 …self.weight_data_file_prefix, interpreter.get_tensor(filter_layer['index']), pack=self.int4_weight…
432 … self.bias_data_file_prefix, interpreter.get_tensor(bias_layer['index']), bias_datatype)
[all …]
|
| D | model_extractor.py | 275 interpreter = self.Interpreter(model_path=str(self.tflite_model),
277 interpreter.allocate_tensors()
280 tensor_details = interpreter.get_tensor_details()
282 output_details = interpreter.get_output_details()
285 input_details = interpreter.get_input_details()
290 interpreter.set_tensor(input_details[0]["index"], tf.cast(input_data, tf.int8))
297 interpreter.invoke()
298 output_data = interpreter.get_tensor(output_details[0]["index"])
|
| D | README.md | 57 …/1484#issuecomment-1677842603. This interpreter is only partially supported, see *Tests depending …
135 ### Tests depending on TFLM interpreter
137 … be bit-exact to TFLM reference kernels. Hence those operators depends on tflite_micro interpreter.
|
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_batch_matmul.py | 22 from tensorflow.lite.python.interpreter import Interpreter
23 from tensorflow.lite.python.interpreter import OpResolverType
61 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
62 interpreter.allocate_tensors()
63 tensor_details = interpreter.get_tensor_details()
68 input_details = interpreter.get_input_details()
72 output_details = interpreter.get_output_details()
75 tensors["lhs_input_tensor"] = interpreter.get_tensor(lhs['index'])
76 tensors["rhs_input_tensor"] = interpreter.get_tensor(rhs['index'])
|
| D | op_pad.py | 22 from tensorflow.lite.python.interpreter import Interpreter
23 from tensorflow.lite.python.interpreter import OpResolverType
57 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
58 interpreter.allocate_tensors()
60 output_details = interpreter.get_output_details()
|
| D | op_pooling.py | 22 from tensorflow.lite.python.interpreter import Interpreter
23 from tensorflow.lite.python.interpreter import OpResolverType
62 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
63 interpreter.allocate_tensors()
64 output_details = interpreter.get_output_details()
|
| D | op_maximum_minimum.py | 22 from tensorflow.lite.python.interpreter import Interpreter
23 from tensorflow.lite.python.interpreter import OpResolverType
67 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
68 interpreter.allocate_tensors()
69 tensor_details = interpreter.get_tensor_details()
74 input_details = interpreter.get_input_details()
78 output_details = interpreter.get_output_details()
|
| D | test.py | 28 from tensorflow.lite.python.interpreter import Interpreter
29 from tensorflow.lite.python.interpreter import OpResolverType
45 from tflite_runtime.interpreter import Interpreter as TfliteRuntimeInterpreter
46 from tflite_runtime.interpreter import OpResolverType as TfliteRuntimeOpResolverType
245 …interpreter = Interpreter(str(tflite_path), experimental_op_resolver_type=OpResolverType.BUILTIN_R…
246 interpreter.allocate_tensors()
249 input_index = interpreter.get_input_details()[i]["index"]
250 interpreter.set_tensor(input_index, val)
252 interpreter.invoke()
253 output_index = interpreter.get_output_details()[0]["index"]
[all …]
|
| D | op_conv.py | 20 from tensorflow.lite.python.interpreter import Interpreter
21 from tensorflow.lite.python.interpreter import OpResolverType
40 interpreter = tflite_micro.runtime.Interpreter.from_file(
44 x_output = interpreter.get_output_details(0)["shape"][2]
45 y_output = interpreter.get_output_details(0)["shape"][1]
47 … scales["input_scale"] = interpreter.get_input_details(0)["quantization_parameters"]["scales"][0]
48 …scales["input_zero_point"] = interpreter.get_input_details(0)["quantization_parameters"]["zero_poi…
49 … scales["output_scale"] = interpreter.get_output_details(0)["quantization_parameters"]["scales"][0]
50 …scales["output_zero_point"] = interpreter.get_output_details(0)["quantization_parameters"]["zero_p…
55 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
[all …]
|
| D | op_fully_connected.py | 22 from tensorflow.lite.python.interpreter import Interpreter
23 from tensorflow.lite.python.interpreter import OpResolverType
154 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
155 interpreter.allocate_tensors()
156 tensor_details = interpreter.get_tensor_details()
172 input_details = interpreter.get_input_details()
175 output_details = interpreter.get_output_details()
178 tensors["weights"] = interpreter.get_tensor(filter_layer['index'])
180 tensors["bias"] = interpreter.get_tensor(bias_layer['index'])
|
| D | op_lstm.py | 22 from tensorflow.lite.python.interpreter import Interpreter
23 from tensorflow.lite.python.interpreter import OpResolverType
84 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
85 interpreter.allocate_tensors()
86 tensor_details = interpreter.get_tensor_details()
110 tensors[name + "_weights"] = interpreter.get_tensor(detail["index"]).flatten()
124 tensors["output_gate_bias"] = interpreter.get_tensor(1 + time_major_offset).flatten()
125 tensors["cell_gate_bias"] = interpreter.get_tensor(2 + time_major_offset).flatten()
126 tensors["forget_gate_bias"] = interpreter.get_tensor(3 + time_major_offset).flatten()
127 tensors["input_gate_bias"] = interpreter.get_tensor(4 + time_major_offset).flatten()
|
| D | op_transpose.py | 23 from tensorflow.lite.python.interpreter import Interpreter
24 from tensorflow.lite.python.interpreter import OpResolverType
|
| /cmsis-nn-latest/ |
| D | README.md | 7 ….com/ARM-software/CMSIS-NN/blob/main/Tests/UnitTest/README.md#tests-depending-on-tflm-interpreter).
|
