Code

Feat/Split ONNX Import#2568

Code

Draft

agelas wants to merge 15 commits intotracel-ai:maintracel-ai/burn:mainfrom agelas:feat/split-onnx-importagelas/burn:feat/split-onnx-importCopy head branch name to clipboard

+349-8

Commits

Commits on Nov 28, 2024

• Add a Split node to burn-import

  agelas

  committed
  d3c26a7

  Copy full SHA for d3c26a7

• Register operation in to_burn

  agelas

  committed
  a992d09

  Copy full SHA for a992d09

• Create Split config function

  agelas

  committed
  1fda6dd

  Copy full SHA for 1fda6dd

• Dimension inference for split outputs

  agelas

  committed
  34b50e6

  Copy full SHA for 34b50e6

• Remove unnecessary f-strings from squeeze

  agelas

  committed
  d3a4c0c

  Copy full SHA for d3a4c0c

• ONNX file for Split and scipt that generates it

  agelas

  committed
  52f0d83

  Copy full SHA for 52f0d83

• Add Split node to name function in Node impl

  agelas

  committed
  9003781

  Copy full SHA for 9003781

• Update supported onnx ops list

  agelas

  committed
  ec5fbf7

  Copy full SHA for ec5fbf7

• Add codegen test

  agelas

  committed
  73e5a20

  Copy full SHA for 73e5a20

• Include split onnx model in build

  agelas

  committed
  33e391b

  Copy full SHA for 33e391b

• Split values should be taken from inputs, make sure only num_outputs or split is provided

  agelas

  committed
  662d11d

  Copy full SHA for 662d11d

• Codegen should make a Vec<Tensor<B, D>>

  agelas

  committed
  402def6

  Copy full SHA for 402def6

Commits on Nov 30, 2024

• Fix up split codegen

  agelas

  committed
  b4ae0e5

  Copy full SHA for b4ae0e5

Commits on Dec 14, 2024

• Remove panic if split is not provided

  agelas

  committed
  fd16d82

  Copy full SHA for fd16d82

• Add basic split test

  agelas

  committed
  9ed4834

  Copy full SHA for 9ed4834