Lines Matching refs:n_input
32 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate() argument
39 for (int col = 0; col < n_input; ++col) { in MatrixBatchVectorMultiplyAccumulate()
40 int8_t input_val = input[batch * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
41 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
138 int n_batch, int n_input, int16_t* output);
147 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output);
150 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh() argument
154 for (int i = 0; i < n_input; ++i) { in ApplyTanh()
155 const int index = batch * n_input + i; in ApplyTanh()
163 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh() argument
168 ApplyTanh<i>(input, n_batch, n_input, output); \ in ApplyTanh()
184 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift,
187 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output);
191 void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input,
201 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input);
203 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input);