Hi, all. Hope you guys are doing well.
I’m a newbie in TensorRT and trying to learn how to add batch matmul in TensorRT.
Is there any example code that I can follow?
It seems like this function IMatrixMultiplyLayer is the one to use, but I’m not sure yet.
TensorRT Version: 220.127.116.11
GPU Type: V100
Nvidia Driver Version:
CUDA Version: 11.1
Operating System + Version:
Python Version (if applicable):
TensorFlow Version (if applicable):
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):
Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)
- Exact steps/commands to build your repro
- Exact steps/commands to run your repro
- Full traceback of errors encountered