Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp353 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in prepare() local
355 outputShape.dimensions = prevOutputShape.dimensions; in prepare()
389 const Shape prevOutputShape = context->getInputShape(kPrevOutputTensor); in execute() local
512 recurrentToForgetWeightsShape.scale * prevOutputShape.scale / forgetIntermediateScale; in execute()
540 recurrentToCellWeightsShape.scale * prevOutputShape.scale / cellIntermediateScale; in execute()
560 recurrentToOutputWeightsShape.scale * prevOutputShape.scale / outputIntermediateScale; in execute()
591 projectionWeightsShape.scale * hiddenStateScale / prevOutputShape.scale; in execute()
617 -prevOutputShape.offset, recurrentToInputWeightsBuffer, in execute()
628 -prevOutputShape.offset, recurrentToForgetWeightsBuffer, recurrentToForgetWeightsShape, in execute()
637 -prevOutputShape.offset, recurrentToCellWeightsBuffer, recurrentToCellWeightsShape, in execute()
[all …]