Any roadmap for supporting Einsum operator?


It’s widely used in high performance ViT applications. Just wondering if there is any plan to support this operator officially.

Hi @ysyyork ,
Einsum operator support will be provided in upcoming TRT releases.
Please stay tuned to the forum and TRT site for the updates.