Searched refs:CwiseMul (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 184 void CwiseMul(const int16_t* input_1, const int16_t* input_2, int n_batch, int n_input, int shift,
186 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/operations/ |
| D | QLSTM.cpp | 741 CwiseMul(forgetGateBuffer.data(), prevCellStateBuffer, batchSize, numUnits, in execute()
743 CwiseMul(inputGateBuffer.data(), cellGateBuffer.data(), batchSize, numUnits, 30 + cellShift, in execute()
775 CwiseMul(outputGateBuffer.data(), inputGateBuffer.data(), hiddenStateEffectiveScaleA, in execute()
|