Code

Add onnx op trilu#2323

Code

Merged

laggui merged 9 commits intotracel-ai:maintracel-ai/burn:mainfrom tiruka:add-onnx-op-trilutiruka/burn:add-onnx-op-triluCopy head branch name to clipboardOct 8, 2024

+295-4

Commits

Commits on Sep 28, 2024

• add trilu.py and onnx model for operators

  tiruka

  committed
  81ed5b8

  Copy full SHA for 81ed5b8

Commits on Sep 30, 2024

• add node/trilu.rs under burn-import and modify related files

  tiruka

  committed
  f0a06e0

  Copy full SHA for f0a06e0

• update trilu codegen tests

  tiruka

  committed
  3c88005

  Copy full SHA for 3c88005

• add op_configuration and to_burn rust files

  tiruka

  committed
  1e9d131

  Copy full SHA for 1e9d131

• update how to get diagonal value from nodes argument

  tiruka

  committed
  d3cb018

  Copy full SHA for d3cb018

• add trilu tests in test_onnx.rs

  tiruka

  committed
  ed9a434

  Copy full SHA for ed9a434

Commits on Oct 1, 2024

• update files to follow clippy and format rules

  tiruka

  committed
  45bbb91

  Copy full SHA for 45bbb91

Commits on Oct 8, 2024

• add tests for both of upper and lower to follow review comment

  tiruka

  committed
  a9c750c

  Copy full SHA for a9c750c

• delete onnx model of trilu.onnx since upper and lower models are added

  tiruka

  committed
  f37f5a6

  Copy full SHA for f37f5a6