Specify shape and print op to the new gpu back-end