Home
last modified time | relevance | path

Searched refs:inputToOutputShape (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp205 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare() local
206 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
207 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
208 const uint32_t numCells = getSizeOfDimension(inputToOutputShape, 0); in prepare()
DQLSTM.cpp185 const Shape inputToOutputShape = context->getInputShape(kInputToOutputWeightsTensor); in prepare() local
186 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
187 NN_RET_CHECK_EQ(getSizeOfDimension(inputToOutputShape, 1), inputSize); in prepare()
188 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare()