Code

Onnx op topk#2305

Code

Open

oojo12 wants to merge 14 commits intotracel-ai:maintracel-ai/burn:mainfrom oojo12:onnx-op-topkoojo12/burn:onnx-op-topkCopy head branch name to clipboard

+496-30

Commits

Commits on Sep 25, 2024

• feat: add topk operation

  oojo12

  committed
  da34ba5

• fix test

  oojo12

  committed
  2973945

• fix: update IOEntry::Node j in the input_name map

  oojo12

  committed
  578c346

• fix: only run on macos

  oojo12

  committed
  17dce60

• chore: clean up

  oojo12

  committed
  ce1aa89

• chore: updated supported ops

  oojo12

  committed
  5783f76

• cleanup

  oojo12

  committed
  16ce366

• cleanup

  oojo12

  committed
  4ca56c6

Commits on Sep 27, 2024

• address feedback

  oojo12

  committed
  49a6a73

• usize cast in config and add topk_smallest

  oojo12

  committed
  f293ad3

• EOD

  oojo12

  committed
  449f519

• EOD cleanup

  oojo12

  committed
  249d60c

Commits on Sep 28, 2024

• minor update

  oojo12

  committed
  07f79ab

• run checks and other fixes

  oojo12

  committed
  7e221c9