Searched refs:cellToInputShape (Results 1 – 2 of 2) sorted by relevance
237 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare() local238 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()239 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numUnits); in prepare()373 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in execute() local492 std::pow(2, cellShift) * cellToInputShape.scale / inputIntermediateScale; in execute()
259 const Shape cellToInputShape = context->getInputShape(kCellToInputWeightsTensor); in prepare() local260 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()261 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numCells); in prepare()