Searched refs:cellToOutputShape (Results 1 – 2 of 2) sorted by relevance
249 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare() local250 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1); in prepare()251 NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numUnits); in prepare()386 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in execute() local570 std::pow(2, cellShift) * cellToOutputShape.scale / outputIntermediateScale; in execute()
271 const Shape cellToOutputShape = context->getInputShape(kCellToOutputWeightsTensor); in prepare() local272 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToOutputShape), 1); in prepare()273 NN_RET_CHECK_EQ(getSizeOfDimension(cellToOutputShape, 0), numCells); in prepare()