Searched refs:inputToCellShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceLSTM.cpp | 120 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare() local
121 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2u); in prepare()
122 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numCells); in prepare()
123 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 1), inputSize); in prepare()
|
| D | QLSTM.cpp | 91 const Shape inputToCellShape = context->getInputShape(kInputToCellWeightsTensor); in prepare() local
92 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2u); in prepare()
93 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numUnits); in prepare()
94 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 1), inputSize); in prepare()
|