Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp375 const Shape inputToForgetWeightsShape = context->getInputShape(kInputToForgetWeightsTensor); in execute() local
506 inputToForgetWeightsShape.scale * inputShape.scale / forgetIntermediateScale; in execute()
624 -inputShape.offset, inputToForgetWeightsBuffer, inputToForgetWeightsShape, in execute()