Searched refs:effective_scales (Results 1 – 5 of 5) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_lstm.py | 80 effective_scales = {}
103 … effective_scales["forget_to_cell"] = pow(2, -15) * scales["cell_scale"] / scales["cell_scale"]
104 effective_scales["input_to_cell"] = pow(2, -15) * pow(2, -15) / scales["cell_scale"]
105 effective_scales["output"] = pow(2, -15) * pow(2, -15) / scales["output_scale"]
112 effective_scales[name] = input_scale * scales[name + "_scale"] / pow(2, -12)
134 return Lib.op_utils.Generated_data(generated_params, tensors, scales, effective_scales)
139 effective_scales = {}
152 … effective_scales["forget_to_cell"] = pow(2, -15) * scales["cell_scale"] / scales["cell_scale"]
153 effective_scales["input_to_cell"] = pow(2, -15) * pow(2, -15) / scales["cell_scale"]
154 effective_scales["output"] = pow(2, -15) * pow(2, -15) / scales["output_scale"]
[all …]
|
| D | op_utils.py | 24 def __init__(self, params, tensors, scales, effective_scales, aliases={}): argument
28 self.effective_scales = effective_scales # Scales used by CMSIS-NN
|
| D | op_fully_connected.py | 61 effective_scales = {}
112 … return Lib.op_utils.Generated_data(generated_params, tensors, scales, effective_scales, aliases)
|
| D | op_conv.py | 102 effective_scales = {}
216 … return Lib.op_utils.Generated_data(generated_params, tensors, scales, effective_scales, aliases)
|
| D | test.py | 93 for name, scale in data.effective_scales.items():
|
