12                     int n_batch, int n_input, int16_t* output) {  in ApplyLayerNorm()  argument
17         for (int j = 0; j < n_input; ++j) {  in ApplyLayerNorm()
18             const int32_t index = i * n_input + j;  in ApplyLayerNorm()
23         int32_t mean = static_cast<int32_t>(static_cast<int64_t>(sum) * 1024 / n_input);  in ApplyLayerNorm()
25         int32_t temp = kOverflowGuard / n_input;  in ApplyLayerNorm()
36         for (int j = 0; j < n_input; ++j) {  in ApplyLayerNorm()
37             const int32_t index = i * n_input + j;  in ApplyLayerNorm()
86 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) {  in ApplySigmoid()  argument
88         for (int c = 0; c < n_input; c++) {  in ApplySigmoid()
91             const int index = batch * n_input + c;  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
102         for (int i = 0; i < n_input; ++i) {  in CwiseMul()
103             const int index = batch * n_input + i;  in CwiseMul()
113               int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output) {  in CwiseMul()  argument
115         for (int i = 0; i < n_input; ++i) {  in CwiseMul()
116             const int index = batch * n_input + i;  in CwiseMul()
138 void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input,  in CwiseAdd()  argument
141         for (int i = 0; i < n_input; ++i) {  in CwiseAdd()
142             const int index = batch * n_input + i;  in CwiseAdd()
150 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input) {  in CwiseClipping()  argument
152         for (int i = 0; i < n_input; ++i) {  in CwiseClipping()
153             const int index = batch * n_input + i;  in CwiseClipping()
164 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input) {  in CwiseClipping()  argument
166         for (int i = 0; i < n_input; ++i) {  in CwiseClipping()
167             const int index = batch * n_input + i;  in CwiseClipping()