Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp300 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare() local
301 NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1); in prepare()
302 NN_RET_CHECK_EQ(getSizeOfDimension(outputGateBiasShape, 0), numCells); in prepare()
DQLSTM.cpp280 const Shape outputGateBiasShape = context->getInputShape(kOutputGateBiasTensor); in prepare() local
281 NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1); in prepare()
282 NN_RET_CHECK_EQ(getSizeOfDimension(outputGateBiasShape, 0), numUnits); in prepare()