Searched refs:calc_scale (Results 1 – 1 of 1) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_lstm.py | 108 def calc_scale(name, input_scale, tensor_index): function
114 calc_scale("output_gate_hidden", scales["output_scale"], 5)
115 calc_scale("cell_gate_hidden", scales["output_scale"], 6)
116 calc_scale("forget_gate_hidden", scales["output_scale"], 7)
117 calc_scale("input_gate_hidden", scales["output_scale"], 8)
119 calc_scale("output_gate_input", scales["input_scale"], 9)
120 calc_scale("cell_gate_input", scales["input_scale"], 10)
121 calc_scale("forget_gate_input", scales["input_scale"], 11)
122 calc_scale("input_gate_input", scales["input_scale"], 12)
|
