Преглед изворни кода

CMSIS-NN: Fix fully connected unit test

Use correct quantization during generation of output reference data.
Måns Nilsson пре 5 година
родитељ
комит
89e0dd9773
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2 2
      CMSIS/NN/Tests/UnitTest/generate_test_data.py

+ 2 - 2
CMSIS/NN/Tests/UnitTest/generate_test_data.py

@@ -738,8 +738,8 @@ class FullyConnectedSettings(TestSettings):
         self.generate_c_array("weights", self.convert_tensor(weights, self.quantize2int8,
                                                              [self.weights_min, self.weights_max]))
         self.generate_c_array("biases", self.convert_tensor(biases, self.quantize_bias), "int32_t")
-        self.generate_c_array("output_ref", self.convert_tensor(conv, self.quantize2int8,
-                                                                [self.output_min, self.output_max]))
+        self.generate_c_array("output_ref", self.convert_tensor(conv, self.quantize_output))
+
         self.quantize_multiplier()
 
         self.write_c_config_header()