Set block dim and grid dim macros within NestedTensorTransformerFunctions.cu (#77199)
Follow up to https://github.com/pytorch/pytorch/pull/76157#discussion_r856668878
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77199
Approved by: https://github.com/ngimel