Searched refs:recurrentToForgetShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 238 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare() local
239 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
240 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numCells); in prepare()
241 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 1), outputSize); in prepare()
|
| D | QLSTM.cpp | 218 const Shape recurrentToForgetShape = context->getInputShape(kRecurrentToForgetWeightsTensor); in prepare() local
219 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
220 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numUnits); in prepare()
221 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 1), outputSize); in prepare()
|