Skip to content

Instantly share code, notes, and snippets.

@lmoneta
Last active July 11, 2022 15:52
Show Gist options
  • Save lmoneta/340c220017d9313ca69a11fc9999c81a to your computer and use it in GitHub Desktop.
Save lmoneta/340c220017d9313ca69a11fc9999c81a to your computer and use it in GitHub Desktop.
ONNX Operators

List of Missing ONNX operator for TMVA/SOFIE

This is a list of operators that can be used to parse this ONNX model: https://cernbox.cern.ch/index.php/s/Z59tokW0KvBF3sb

  • LeakyRelu
  • ReduceMean
  • ReduceProd
  • ReduceSumSquare
  • Sub
  • Mul
  • Div
  • Shape
  • Cast
  • Gather (used for example for Embedding)
  • Expand

Extra operators useful for GNN networks:

  • Elu

Other operators used in Particle Net

  • TopK
  • Neg
  • Pow
  • Range
  • Tile
  • ConstantOfShape
  • Max
  • Equal
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment