Lines Matching refs:n_batch
12 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm() argument
14 for (int i = 0; i < n_batch; ++i) { in ApplyLayerNorm()
86 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplySigmoid() argument
87 for (int batch = 0; batch < n_batch; ++batch) { in ApplySigmoid()
99 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift, in CwiseMul() argument
101 for (int batch = 0; batch < n_batch; ++batch) { in CwiseMul()
113 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output) { in CwiseMul() argument
114 for (int batch = 0; batch < n_batch; ++batch) { in CwiseMul()
138 void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, in CwiseAdd() argument
140 for (int batch = 0; batch < n_batch; ++batch) { in CwiseAdd()
150 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping() argument
151 for (int batch = 0; batch < n_batch; ++batch) { in CwiseClipping()
164 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping() argument
165 for (int batch = 0; batch < n_batch; ++batch) { in CwiseClipping()
179 const int16_t* batch_vector, int n_batch, in VectorBatchVectorCwiseProductAccumulate() argument
181 for (int b = 0; b < n_batch; b++) { in VectorBatchVectorCwiseProductAccumulate()