Home
last modified time | relevance | path

Searched refs:datatype (Results 1 – 11 of 11) sorted by relevance

/cmsis-nn-latest/Tests/UnitTest/
Dpooling_settings.py76 datatype = "int16_t"
79 datatype = "int8_t"
83 self.generate_c_array(self.input_data_file_prefix, input_data, datatype=datatype)
120 datatype=datatype)
Dsoftmax_settings.py121 datatype = "int16_t"
124 datatype = "int8_t"
126 self.generate_c_array(self.input_data_file_prefix, input_data, datatype=datatype)
131 datatype = "int16_t"
160 self.generate_c_array(self.output_data_file_prefix, output_data, datatype=datatype)
Dlstm_settings.py233 self.generate_c_array("cell_to_input", [], datatype='int16_t')
234 self.generate_c_array("cell_to_forget", [], datatype='int16_t')
235 self.generate_c_array("cell_to_output", [], datatype='int16_t')
237 …te_c_array("input_gate_bias", interpreter.get_tensor(input_gate_bias['index']), datatype='int32_t')
238 …rate_c_array("cell_gate_bias", interpreter.get_tensor(cell_gate_bias['index']), datatype='int32_t')
239 …_c_array("forget_gate_bias", interpreter.get_tensor(forget_gate_bias['index']), datatype='int32_t')
240 …_c_array("output_gate_bias", interpreter.get_tensor(output_gate_bias['index']), datatype='int32_t')
244 self.generate_c_array("projection_bias", [], datatype='int32_t')
247 …self.generate_c_array("cell_state", interpreter.get_tensor(cell_state['index']), datatype='int16_t…
287 … self.generate_c_array("input_to_input_eff_bias", input_to_input_eff_bias, datatype='int32_t')
[all …]
Dfully_connected_settings.py141 datatype = "int16_t"
145 datatype = "int8_t"
284 self.generate_c_array(self.input_data_file_prefix, input_data, datatype=datatype)
291 self.generate_c_array(self.bias_data_file_prefix, bias, datatype=bias_datatype)
295 datatype=datatype)
Dconv_settings.py208 datatype = "int16_t"
212 datatype = "int8_t"
421 self.generate_c_array(self.input_data_file_prefix, input_data, datatype=datatype)
427 self.generate_c_array("output_mult", per_channel_multiplier, datatype='int32_t')
428 self.generate_c_array("output_shift", per_channel_shift, datatype='int32_t')
442 datatype=datatype)
Dadd_mul_settings.py130 self.generate_c_array("input1", input_data1, datatype=inttype)
131 self.generate_c_array("input2", input_data2, datatype=inttype)
134 datatype=inttype)
Dmodel_extractor.py256 self.generate_c_array(nice_name + "_bias", bias_data, datatype='int32_t')
266 … self.generate_c_array(f"{nice_name}_output_mult", per_channel_multiplier, datatype='int32_t')
267 … self.generate_c_array(f"{nice_name}_output_shift", per_channel_shift, datatype='int32_t')
Dsvdf_settings.py225 …nerate_c_array("weights_time", interpreter.get_tensor(weights_2_layer['index']), datatype='int8_t')
228 …erate_c_array("weights_time", interpreter.get_tensor(weights_2_layer['index']), datatype='int16_t')
Dtest_settings.py366 def generate_c_array(self, name, array, datatype="int8_t", const="const ", pack=False) -> None: argument
392 … f.write(const + datatype + " " + self.testdataset + '_' + name + "[%d] =\n{\n" % size)
398 f.write(const + datatype + " *" + self.testdataset + '_' + name + " = NULL;\n")
/cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/
Dop_utils.py62 def generate_tf_tensor(dims, minval, maxval, decimals=0, datatype=tf.float32): argument
65 tensor = tf.convert_to_tensor(array, dtype=datatype)
Dtest.py103 …nsor = Lib.op_utils.generate_tf_tensor(shapes["input"], minval, maxval, decimals=0, datatype=dtype)