Skip to content

[backend_cuda] Add ViT Operators Impl

In order to be able to forward ViT operator, these operators need to be implemented in backend_cuda:

  • Erf
  • MatMul
  • Softmax

TODO

  • Erf
  • MatMul
  • Softmax
Edited by Houssem ROUIS