Skip to content

Vit operators

Houssem ROUIS requested to merge (removed):vit_operators into dev
  • [Add] OperatorTensors with Kernel implementaiton and unit-tests
    • Erf
    • Concat
    • Gather
    • ReduceMean
    • Reshape
    • Slice
    • Transpose
  • [Softmax kernel] update to support doing the operation on the attribute axis and not on a fixed one.
Edited by Maxence Naud

Merge request reports