Searched refs:cellShift (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | QLSTM.cpp | 470 int cellShift; in execute() local
471 NN_RET_CHECK(CheckedLog2(prevCellStateShape.scale, &cellShift)); in execute()
472 NN_RET_CHECK(cellShift <= -9); in execute()
492 std::pow(2, cellShift) * cellToInputShape.scale / inputIntermediateScale; in execute()
522 std::pow(2, cellShift) * cellToForgetShape.scale / forgetIntermediateScale; in execute()
570 std::pow(2, cellShift) * cellToOutputShape.scale / outputIntermediateScale; in execute()
743 CwiseMul(inputGateBuffer.data(), cellGateBuffer.data(), batchSize, numUnits, 30 + cellShift, in execute()
774 ApplyTanh(cellShift + 15, cellStateBuffer, batchSize, numUnits, inputGateBuffer.data()); in execute()
|