Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp286 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare() local
287 NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1); in prepare()
288 NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numCells); in prepare()
DQLSTM.cpp266 const Shape inputGateBiasShape = context->getInputShape(kInputGateBiasTensor); in prepare() local
267 NN_RET_CHECK_EQ(getNumberOfDimensions(inputGateBiasShape), 1); in prepare()
268 NN_RET_CHECK_EQ(getSizeOfDimension(inputGateBiasShape, 0), numUnits); in prepare()