Searched refs:CwiseMul (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 188 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift,
190 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int32_t multiplier, int32_t shift,
|
| D | QuantUtils.cpp | 99 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift, in CwiseMul() function
112 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int32_t multiplier, int32_t shift, in CwiseMul() function
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QLSTM.cpp | 625 CwiseMul(forgetGateBuffer.data(), prevCellStateBuffer, batchSize, numUnits, in execute()
627 CwiseMul(inputGateBuffer.data(), cellGateBuffer.data(), batchSize, numUnits, 30 + cellShift, in execute()
659 CwiseMul(outputGateBuffer.data(), inputGateBuffer.data(), hiddenStateEffectiveScaleA, in execute()
|