Home
last modified time | relevance | path

Searched refs:prevCellStateShape (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp357 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare() local
359 cellStateOutShape.dimensions = prevCellStateShape.dimensions; in prepare()
390 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute() local
471 NN_RET_CHECK(CheckedLog2(prevCellStateShape.scale, &cellShift)); in execute()
600 std::min(std::max(cellClip / prevCellStateShape.scale, -32768.0f), 32767.0f)); in execute()