Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp383 const Shape inputToOutputWeightsShape = context->getInputShape(kInputToOutputWeightsTensor); in execute() local
394 const uint32_t numUnits = inputToOutputWeightsShape.dimensions[0]; in execute()
554 inputToOutputWeightsShape.scale * inputShape.scale / outputIntermediateScale; in execute()
642 -inputShape.offset, inputToOutputWeightsBuffer, inputToOutputWeightsShape, in execute()