Searched refs:cellToInputShape (Results 1 – 2 of 2) sorted by relevance
236 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare() local237 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()238 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numUnits); in prepare()371 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute() local490 std::pow(2, cellShift) * cellToInputShape.scale / inputIntermediateScale; in execute()
256 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare() local257 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()258 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numCells); in prepare()