Code

Feat/Split Operator#2490

Code

Merged

nathanielsimard merged 35 commits intotracel-ai:maintracel-ai/burn:mainfrom agelas:feat/split-opagelas/burn:feat/split-opCopy head branch name to clipboardNov 21, 2024

+813-17

Commits

Commits on Nov 14, 2024

• Create high-level function signatures for split() and split_with_sizes(), add some clarity vs chunk()

  agelas

  committed
  242094e

  Copy full SHA for 242094e

• Add checks for split() and split_with_sizes()

  agelas

  committed
  64df3d9

  Copy full SHA for 64df3d9

• Remove pointless negative checks, usize is an int

  agelas

  committed
  6e2ca04

  Copy full SHA for 6e2ca04

• Add element-type specific function signatures

  agelas

  committed
  21d06fe

  Copy full SHA for 21d06fe

• Implement split and split_with_sizes for tch backend

  agelas

  committed
  4e30256

  Copy full SHA for 4e30256

• Add split ops to burn-autodiff

  agelas

  committed
  d4cf026

  Copy full SHA for d4cf026

• Add split ops to list of basic operations in burn-book

  agelas

  committed
  3cfdd32

  Copy full SHA for 3cfdd32

• Add fallbacks for split and split_with_size since a few backends dont support them

  agelas

  committed
  5a14745

  Copy full SHA for 5a14745

• Cleanup split_with_sizes and fix arguments

  agelas

  committed
  185703f

  Copy full SHA for 185703f

• Clippy fixes

  agelas

  committed
  2b1c152

  Copy full SHA for 2b1c152

• Finish todos in tensor.rs

  agelas

  committed
  01272f1

  Copy full SHA for 01272f1

• Add documentation to split functions

  agelas

  committed
  334fa0b

  Copy full SHA for 334fa0b

• Small consistency/grammar fixes for chunk

  agelas

  committed
  bd5343f

  Copy full SHA for bd5343f

• Punctuation fix

  agelas

  committed
  236b919

  Copy full SHA for 236b919

• Small doc fixes and change to split_with_sizes

  agelas

  committed
  c5d08ee

  Copy full SHA for c5d08ee

• Add split functions to float, int, and bool BasicOps impls

  agelas

  committed
  824cb30

  Copy full SHA for 824cb30

• Add split functions to int, bool, and quantized tensors

  agelas

  committed
  1deee50

  Copy full SHA for 1deee50

Commits on Nov 15, 2024

• Minor doc + code fixes

  agelas

  committed
  a99a306

  Copy full SHA for a99a306

• Add documentation

  agelas

  committed
  9ac5fc0

  Copy full SHA for 9ac5fc0

• Start adding tests

  agelas

  committed
  3093b5c

  Copy full SHA for 3093b5c

• Try to get this to compile

  agelas

  committed
  36a6ff5

  Copy full SHA for 36a6ff5

• Fix docs and add examples

  agelas

  committed
  3e790df

  Copy full SHA for 3e790df

• Get initial tests to a working state

  agelas

  committed
  d1624f0

  Copy full SHA for d1624f0

• Fix doc tests

  agelas

  committed
  d647a9e

  Copy full SHA for d647a9e

• Add more split tests

  agelas

  committed
  54ef092

  Copy full SHA for 54ef092

• Add more tests for 3D tensor splitting

  agelas

  committed
  2aedd20

  Copy full SHA for 2aedd20

• Check with full dims so no indexing panic

  agelas

  committed
  b3ed77a

  Copy full SHA for b3ed77a

• Correct flow of check

  agelas

  committed
  1775c1a

  Copy full SHA for 1775c1a

• Specify which panic to expect in tests

  agelas

  committed
  7ddab38

  Copy full SHA for 7ddab38

• Add split_with_sizes tests

  agelas

  committed
  3d064f8

  Copy full SHA for 3d064f8

• Small fix to chuk tests

  agelas

  committed
  9fb3aad

  Copy full SHA for 9fb3aad

Commits on Nov 19, 2024

• Resolve merge conflicts

  agelas

  committed
  dfb571a

  Copy full SHA for dfb571a

• Fix redundant closure

  agelas

  committed
  e98240a

  Copy full SHA for e98240a

• Fix redunant closure even more

  agelas

  committed
  86f5d31

  Copy full SHA for 86f5d31

Commits on Nov 20, 2024

• Resolve merge conflict

  agelas

  committed
  fe8bd67

  Copy full SHA for fe8bd67