Searched refs:ApplyTanh (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 150 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh() function
163 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh() function
168 ApplyTanh<i>(input, n_batch, n_input, output); \ in ApplyTanh()
|
| /packages/modules/NeuralNetworks/common/operations/ |
| D | QLSTM.cpp | 711 ApplyTanh<3>(cellGateBuffer.data(), batchSize, numUnits, cellGateBuffer.data()); in execute()
774 ApplyTanh(cellShift + 15, cellStateBuffer, batchSize, numUnits, inputGateBuffer.data()); in execute()
|