* Fix build error of affine grid op in mac os.
test=develop

* Make function return reference.
test=develop