Add OpInfo for torch.baddbmm (#56502)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56502
Reviewed By: heitorschueroff
Differential Revision: D27890939
Pulled By: anjali411
fbshipit-source-id: 072647a05cf93aedb76df0367af71b534be77258