Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp210 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare() local
211 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
212 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numCells); in prepare()
213 const uint32_t outputSize = getSizeOfDimension(recurrentToOutputShape, 1); in prepare()
DQLSTM.cpp190 const Shape recurrentToOutputShape = context->getInputShape(kRecurrentToOutputWeightsTensor); in prepare() local
191 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
192 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numUnits); in prepare()
193 const uint32_t outputSize = getSizeOfDimension(recurrentToOutputShape, 1); in prepare()