WarpAffine fixes and enhancements