Searched refs:recurrentToCellShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 242 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare() local
243 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
244 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numCells); in prepare()
245 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 1), outputSize); in prepare()
|
| D | QLSTM.cpp | 222 const Shape recurrentToCellShape = context->getInputShape(kRecurrentToCellWeightsTensor); in prepare() local
223 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
224 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numUnits); in prepare()
225 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 1), outputSize); in prepare()
|