Switch to CUDA implementation if batch size >= 65536 for affine_grid (#16403)
Summary:
Changelog:
- Append a condition that switches to the native CUDA implementation for affine_grid
Fixes #16365
Differential Revision: D13832192
Pulled By: soumith
fbshipit-source-id: 3f484e6673d71e3ba7627b170cb8f1611e12b9b2
