Home
last modified time | relevance | path

Searched defs:outputStateShape (Results 1 – 5 of 5) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceRNN.cpp190 Shape outputStateShape = context->getInputShape(kHiddenStateTensor); in prepare() local
H A DUnidirectionalSequenceLSTM.cpp320 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare() local
H A DQLSTM.cpp298 const Shape outputStateShape = context->getInputShape(kPrevOutputTensor); in prepare() local
H A DLSTM.cpp304 Shape* scratchShape, Shape* outputStateShape, Shape* cellStateShape, in Prepare()
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp1046 Shape scratchShape, outputStateShape, cellStateShape, outputShape; in executeOperation() local