| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.cpp | 12 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm()
86 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { 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()
113 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output) { in CwiseMul()
138 void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, in CwiseAdd()
150 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping()
164 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping()
|
| D | QuantUtils.h | 32 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate()
150 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh()
163 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh()
|
| /packages/modules/NeuralNetworks/common/operations/ |
| D | LSTMTest.cpp | 78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel()
274 const int n_input = 2; in TEST() local
378 const int n_input = 2; in TEST() local
475 const int n_input = 5; in TEST() local
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | quantized_lstm.mod.py | 21 n_input = 2 variable
108 n_input = 2 variable
|
| D | layer_norm_lstm.mod.py | 23 n_input = 5 variable
195 n_input = 5 variable
|
| D | lstm3_state_float16.mod.py | 22 n_input = 5 variable
|
| D | lstm3_float16.mod.py | 22 n_input = 5 variable
|
| D | lstm3_state2_float16.mod.py | 22 n_input = 5 variable
|
| D | lstm2_float16.mod.py | 22 n_input = 2 variable
|
| D | lstm2_state_float16.mod.py | 22 n_input = 2 variable
|
| D | lstm2_state2_float16.mod.py | 22 n_input = 2 variable
|
| D | lstm3_state3_float16.mod.py | 22 n_input = 5 variable
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | lstm2.mod.py | 22 n_input = 2 variable
|
| D | lstm3_state3.mod.py | 22 n_input = 5 variable
|
| D | lstm3.mod.py | 22 n_input = 5 variable
|
| D | lstm2_state.mod.py | 22 n_input = 2 variable
|
| D | lstm3_state2.mod.py | 22 n_input = 5 variable
|
| D | lstm3_state.mod.py | 22 n_input = 5 variable
|
| D | lstm2_state2.mod.py | 22 n_input = 2 variable
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | lstm3_state2_relaxed.mod.py | 22 n_input = 5 variable
|
| D | lstm3_state_relaxed.mod.py | 22 n_input = 5 variable
|
| D | lstm2_relaxed.mod.py | 22 n_input = 2 variable
|
| D | lstm3_relaxed.mod.py | 22 n_input = 5 variable
|
| D | lstm2_state2_relaxed.mod.py | 22 n_input = 2 variable
|
| D | lstm3_state3_relaxed.mod.py | 22 n_input = 5 variable
|