Searched refs:cellToOutputShape (Results 1 – 2 of 2) sorted by relevance
248 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare() local249 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1); in prepare()250 NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numUnits); in prepare()384 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute() local568 std::pow(2, cellShift) * cellToOutputShape.scale / outputIntermediateScale; in execute()
268 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare() local269 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1); in prepare()270 NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numCells); in prepare()