Code

Feature/onnx argmax#1814

Code

Merged

laggui merged 8 commits intotracel-ai:maintracel-ai/burn:mainfrom will-maclean:feature/onnx-argmaxwill-maclean/burn:feature/onnx-argmaxCopy head branch name to clipboardMay 31, 2024

+255-9

Commits

Commits on May 17, 2024

• pre-test

  will-maclean

  committed
  51a8910

  Copy full SHA for 51a8910

Commits on May 26, 2024

• implementing argmax for burn-import from onnx

  will-maclean

  committed
  cfbf5bb

  Copy full SHA for cfbf5bb

• tidying

  will-maclean

  committed
  524541c

  Copy full SHA for 524541c

• fixing return types and tests

  will-maclean

  committed
  bcec44d

  Copy full SHA for bcec44d

• merge

  will-maclean

  committed
  60b5cd1

  Copy full SHA for 60b5cd1

Commits on May 27, 2024

• merge

  will-maclean

  committed
  716e372

  Copy full SHA for 716e372

• addressing feedback

  will-maclean

  committed
  028367b

  Copy full SHA for 028367b

Commits on May 31, 2024

• only warn when select_last_index!=0

  will-maclean

  committed
  f9d8e32

  Copy full SHA for f9d8e32