Description
It’s widely used in high performance ViT applications. Just wondering if there is any plan to support this operator officially.
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.
Thanks!