* kleidiai: add support for get_rows * apply fixes based on code review * apply more fixes based on code review