Searched refs:recurrentToInputShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 232 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare() local
233 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
234 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numCells); in prepare()
235 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 1), outputSize); in prepare()
|
| D | QLSTM.cpp | 212 const Shape recurrentToInputShape = context->getInputShape(kRecurrentToInputWeightsTensor); in prepare() local
213 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
214 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numUnits); in prepare()
215 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 1), outputSize); in prepare()
|