Commit Graph

9 Commits

Author SHA1 Message Date
Guillaume Lagrange cdd1fa1672
Refactor tensor data (#1916)
* Move distribution to module

* Add new TensorData with serialization support

* Implement display and from for TensorData

* Add missing Cargo.lock

* Add missing bytemuck feature

* Add zeros, ones, full and random TensorData methods

* Refactor Data -> TensorData usage

* Fix tests

Since TensorData is not generic over the element type anymore no type inference can be done by the compiler. We must explicitly cast the expected results to the expected backend type.

* Remove commented line

* Fix import

* Add record-backward-compat

* Remove dim const generic from TensorData

* Support NestedValue de/serialization with TensorData

* Fix burn-jit tests

* Remove eprinln

* Refactor onnx import to use TensorData

* Fix tch from_data

* Fix nested value serialization for u8

* Fix missing import

* Fix reduce min onnx test

* Fix deprecated attribute

* Remove shape getter

* Remove strict assert in tests

* Add tensor data as_bytes

* Add tensor check for rank mismatch

* Fix typo (dimensions plural)

* Fix error message

* Update book examples with from_data and fix Display impl for TensorData

* Add deprecation note
2024-06-26 20:22:19 -04:00
Mathias Insley 6b0673d4bd
Contributor Book: Onnx to Burn Conversion (#1771)
* Start on steps (1 & 2) for implementing a new operator for onnx conversion

* Add more steps to process

* Touch up

* Fix dimension inferencing instructions

* Fix numbering and other small stuff

* Update w/links

* Add a warning about dimension changes

* Minor link touch-ups and wording

* Add a note on unary/binary operations
2024-05-22 08:30:11 -04:00
Thierry Cantin-Demers 1cde566317
Add indentation to project architecture in contributing book (#1738)
Now reflects the structure of the book
2024-05-06 13:43:21 -04:00
Anton Blomström 6b14bb8f01
Add info about enabling debugging for new contributors (#1719) 2024-05-02 17:42:18 -04:00
Louis Fortier-Dubois e4b80bad5e
Many superficial fixes to the contributor book (#1644)
* wip

* many fixes in the contributor book

* undo candle modif

* oops candle changes shouldnt have been there

* typo

* fix commands
2024-04-16 17:17:11 -04:00
Guillaume Lagrange 35b36bbe62
Add shape ONNX op support (#1639)
* Add shape onnx op support

* Remove cast node from onnx graph

* Fix shape implementation

* Fix shape config error message

* Fix typo

* Fix clippy type complexity for generated code
2024-04-16 09:28:21 -04:00
Mathias Insley 48c61ebb81
Docs/update contributor book (#1622)
* Update links to latest commit off main

* Some pedantry

* Update links and add jit

* Update instructions for burn-jit and wgpu

* Updated import section with more recent links

* Some grammar/typo/styling fixes

* Code added to burn-wgpu too
2024-04-16 08:33:59 -04:00
Dilshod Tadjibaev 42db540db3
Removed duplicate files that are causing problems on Mac (#1266) 2024-02-06 09:13:45 -05:00
Joshua Ferguson 3eab141608
started the burn developer book (#1184)
---------

Co-authored-by: Sylvain Benner <sylvain@benner.online>
Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
2024-02-05 11:03:24 -05:00