Searched refs:self (Results 1 – 2 of 2) sorted by relevance
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/ |
| D | model_extractor.py | 33 def __init__(self, dataset, schema_file, tflite_model): argument
37 self.tflite_model = tflite_model
39 (self.quantized_multiplier, self.quantized_shift) = 0, 0
40 self.is_int16xint8 = False # Only 8-bit supported.
41 self.diff_min, self.input_multiplier, self.input_left_shift = 0, 0, 0
43 …self.supported_ops = ["CONV_2D", "DEPTHWISE_CONV_2D", "FULLY_CONNECTED", "AVERAGE_POOL_2D", "SOFTM…
45 def from_bytes(self, tensor_data, type_size) -> list: argument
69 def tflite_to_json(self, tflite_input, schema): argument
88 def write_c_config_header(self, name_prefix, op_name, op_index) -> None: argument
91 self.generated_header_files.append(filename)
[all …]
|
| D | generate_test_data.py | 105 def __init__(self, argument
134 self.tensor_flow_reference_version = (
147 …self.tensor_flow_reference_version += ("// Interpreter from {} version {} and revision {}.\n".form…
151 self.mins = randmin
152 self.maxs = randmax
154 self.bias_mins = bias_min
155 self.bias_maxs = bias_max
157 self.input_ch = in_ch
158 self.output_ch = out_ch
159 self.x_input = x_in
[all …]
|
