| /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()
179 const int16_t* batch_vector, int n_batch, in VectorBatchVectorCwiseProductAccumulate()
|
| 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()
273 const int n_batch = 1; in TEST() local
377 const int n_batch = 1; in TEST() local
474 const int n_batch = 2; in TEST() local
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 22 n_batch, argument
455 n_batch = 1 variable
503 n_batch = 1 variable
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | quantized_lstm.mod.py | 20 n_batch = 2 variable
107 n_batch = 1 variable
|
| D | layer_norm_lstm.mod.py | 22 n_batch = 2 variable
194 n_batch = 2 variable
|
| D | lstm3_state_float16.mod.py | 21 n_batch = 2 variable
|
| D | lstm3_float16.mod.py | 21 n_batch = 2 variable
|
| D | lstm3_state2_float16.mod.py | 21 n_batch = 2 variable
|
| D | lstm2_float16.mod.py | 21 n_batch = 1 variable
|
| D | lstm2_state_float16.mod.py | 21 n_batch = 1 variable
|
| D | lstm2_state2_float16.mod.py | 21 n_batch = 1 variable
|
| D | lstm3_state3_float16.mod.py | 21 n_batch = 2 variable
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | lstm2.mod.py | 21 n_batch = 1 variable
|
| D | lstm3_state3.mod.py | 21 n_batch = 2 variable
|
| D | lstm3.mod.py | 21 n_batch = 2 variable
|
| D | lstm2_state.mod.py | 21 n_batch = 1 variable
|
| D | lstm3_state2.mod.py | 21 n_batch = 2 variable
|
| D | lstm3_state.mod.py | 21 n_batch = 2 variable
|
| D | lstm2_state2.mod.py | 21 n_batch = 1 variable
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | lstm3_state2_relaxed.mod.py | 21 n_batch = 2 variable
|
| D | lstm3_state_relaxed.mod.py | 21 n_batch = 2 variable
|
| D | lstm2_relaxed.mod.py | 21 n_batch = 1 variable
|
| D | lstm3_relaxed.mod.py | 21 n_batch = 2 variable
|
| D | lstm2_state2_relaxed.mod.py | 21 n_batch = 1 variable
|