↔ [Converter] Add support for aten::bmm in TRTorch

[narendasan]

aten::bmm

• Function Schema:

aten::bmm(Tensor self, Tensor mat2) -> (Tensor)

• Original PyTorch API:

https://pytorch.org/docs/stable/generated/torch.bmm.html?highlight=bmm#torch.bmm

• Relevant TensorRT Documentation:

https://docs.nvidia.com/deeplearning/tensorrt/api/python_api/infer/Graph/Layers.html#imatrixmultiplylayer

Alternatives

Additional context