Home
last modified time | relevance | path

Searched refs:interpreter (Results 1 – 15 of 15) sorted by relevance

/cmsis-nn-latest/Tests/UnitTest/
Dgenerate_test_data.py69 …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 …]
Dsoftmax_settings.py40 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 …]
Dlstm_settings.py41 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 …]
Dsvdf_settings.py53 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 …]
Dpooling_settings.py49 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"])
Dadd_mul_settings.py42 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"])
Dfully_connected_settings.py51 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 …]
Dtest_settings.py86 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 …]
Dconv_settings.py57 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 …]
Dmodel_extractor.py275 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"])
DREADME.md64 …/1484#issuecomment-1677842603. This interpreter is only partially supported, see *Tests depending …
130 ### Tests depending on TFLM interpreter
132 … be bit-exact to TFLM reference kernels. Hence those operators depends on tflite_micro interpreter.
/cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/
Dtest.py23 from tensorflow.lite.python.interpreter import Interpreter
24 from tensorflow.lite.python.interpreter import OpResolverType
40 from tflite_runtime.interpreter import Interpreter as TfliteRuntimeInterpreter
41 from tflite_runtime.interpreter import OpResolverType as TfliteRuntimeOpResolverType
188 …interpreter = Interpreter(str(tflite_path), experimental_op_resolver_type=OpResolverType.BUILTIN_R…
189 input_index = interpreter.get_input_details()[0]["index"]
190 interpreter.allocate_tensors()
191 interpreter.set_tensor(input_index, input_tensor)
192 interpreter.invoke()
193 output_index = interpreter.get_output_details()[0]["index"]
[all …]
Dop_conv.py20 from tensorflow.lite.python.interpreter import Interpreter
21 from tensorflow.lite.python.interpreter import OpResolverType
112 …interpreter = Interpreter(str(tflite_fname), experimental_op_resolver_type=OpResolverType.BUILTIN_…
113 interpreter.allocate_tensors()
114 tensor_details = interpreter.get_tensor_details()
131 input_details = interpreter.get_input_details()
134 output_details = interpreter.get_output_details()
171 tensors["weights"] = interpreter.get_tensor(filter_layer['index'])
174 tensors["bias"] = interpreter.get_tensor(bias_layer['index'])
Dop_lstm.py22 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()
/cmsis-nn-latest/
DREADME.md7 ….com/ARM-software/CMSIS-NN/blob/main/Tests/UnitTest/README.md#tests-depending-on-tflm-interpreter).