Move Flatten to the new GPU back-end by reusing reshape.