Data Flow now supports GPU shapes