Lines Matching refs:n_batch
35 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate() argument
39 for (int batch = 0; batch < n_batch; ++batch) { in MatrixBatchVectorMultiplyAccumulate()
141 int n_batch, int n_input, int16_t* output);
150 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output);
153 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh() argument
156 for (int batch = 0; batch < n_batch; ++batch) { in ApplyTanh()
166 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh() argument
171 ApplyTanh<i>(input, n_batch, n_input, output); \ in ApplyTanh()
187 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift,
190 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output);
194 void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input,
204 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input);
206 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input);
209 const int16_t* batch_vector, int n_batch,