Mark bucketize as not subject to autograd (#44102)
Summary:
Bucketize returns integers, currently this triggers an internal assert, so we apply the mechanism for this case (also used for argmax etc.).
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44102
Reviewed By: zou3519
Differential Revision: D23500048
Pulled By: albanD
fbshipit-source-id: fdd869cd1feead6616b532b3e188bd5512adedea