Searched refs:weights_zero_point (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | quantized_lstm.mod.py | 29 weights_zero_point = 100 variable
32 [n_output, n_input], weights_scale, weights_zero_point)
39 [n_output, n_output], weights_scale, weights_zero_point)
116 weights_zero_point = 100 variable
119 [n_output, n_input], weights_scale, weights_zero_point)
130 [n_output, n_output], weights_scale, weights_zero_point)
|
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 67 const Dims<4>& activ_temp_dims, int32_t weights_zero_point, in quantizedLstmStep() argument
116 weights_data_uint8[out_c * fc_accum_depth + d] - weights_zero_point; in quantizedLstmStep()
|