Searched refs:projectionShape (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceLSTM.cpp | 308 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare() local
309 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2); in prepare()
310 NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 0), outputSize); in prepare()
311 NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 1), numCells); in prepare()
|
| H A D | QLSTM.cpp | 286 const Shape projectionShape = context->getInputShape(kProjectionWeightsTensor); in prepare() local
287 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2); in prepare()
288 NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 0), outputSize); in prepare()
289 NN_RET_CHECK_EQ(getSizeOfDimension(projectionShape, 1), numUnits); in prepare()
|