Home
last modified time | relevance | path

Searched defs:forgetLayerNormShape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DQLSTM.cpp198 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in prepare() local
263 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormTensor); in execute() local
DUnidirectionalSequenceLSTM.cpp227 const Shape forgetLayerNormShape = context->getInputShape(kForgetLayerNormWeightsTensor); in prepare() local