dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
507 Commits 50 Branches 17 Tags 89 MiB
Commit Graph

44 Commits

Author SHA1 Message Date
Dilshod Tadjibaev 8448611908
License fixes (#648) 2023-08-16 12:45:35 -04:00
Dilshod Tadjibaev 3264b1007c
Changes and fixes required for squeezenet1.1-opset7 onnx (#639) 2023-08-15 15:37:52 -04:00
Louis Fortier-Dubois d659f11639
Perf/wgpu/autotune (#609) 2023-08-15 11:26:00 -04:00
Mitchell Robert Vollger 434975c410
Add conv1d to burn-import (onnx) (#638) 2023-08-14 17:48:02 -04:00
Dilshod Tadjibaev b79fa0748f
Fix bugs with MaxPool2d in ONNX conversation (#623) 2023-08-10 09:35:15 -04:00
Dilshod Tadjibaev c1ba355c26
Softmax OP ONNX support (#620) 2023-08-09 16:48:45 -04:00
Dilshod Tadjibaev 894783f08d
Global avg pool (#611) 2023-08-09 16:15:33 -04:00
Dilshod Tadjibaev 716b7569de
Add Dropout ONNX OP and other refactoring (#606) 2023-08-09 09:53:20 -04:00
Caio Piccirillo 1d3bbaab13
Typos (#608) 2023-08-08 17:57:51 -04:00
Gadersd ed255c5561
Use buffered io for massive performance gains when loading and saving… (#593) 2023-08-06 12:56:27 -04:00
Dilshod Tadjibaev 5cc32cc8cb
Remove topological sort of nodes (#589) 
ONNX nodes should come as topologically sorted, so we are removing it instead, we are making sure the nodes are topologically sorted.
 2023-08-06 10:51:28 -04:00
Dilshod Tadjibaev 1554a3c898
Full support for ONNX scalar operators and Constants (#578) 2023-08-04 16:51:51 -04:00
Nathaniel Simard 8436d4ff66
Feat/tensor/adaptive avg pool2d (#572) 2023-08-04 10:23:59 -04:00
Dilshod Tadjibaev f453f47efb
ONNX Concat op (#561) 2023-07-31 15:46:37 -04:00
Dilshod Tadjibaev 6b459fde82
Update Conv2d source generation to support all attributes (#558) 2023-07-28 12:52:23 -04:00
Luni-4 67ce5ec62c
Implement binary operators for `burn-import` (#532) 2023-07-27 19:37:10 -04:00
Luni-4 2148ac43a4
burn-import: Add transpose operator (#546) 2023-07-27 19:34:43 -04:00
Nathaniel Simard f0a71350cf
refactor: burn-import unary operators (#548) 2023-07-27 13:20:11 -04:00
Luni-4 8e1e29b599
Refactor test code and some missing operators test in `burn-import` (#547) 2023-07-26 14:24:08 -04:00
Luni-4 2af347ca0d
Add `Reshape` operator to `burn-import` (#545) 2023-07-26 14:23:01 -04:00
Nathaniel Simard 0a5a2d729a
chore: bump version for next release (#533) 2023-07-26 09:46:28 -04:00
Luni-4 99af7c900f
burn-import: Implement `Add` operator for onnx (#524) 2023-07-24 12:56:53 -04:00
Luni-4 e066d95d2e
Implement padding for conv2d (#523) 2023-07-24 12:54:45 -04:00
Luni-4 f70ff4dc54
Add MaxPool2d to burn-import (#507) 2023-07-23 19:28:36 -04:00
Luni-4 62ab554df8
burn-import: Add a list of all ONNX operators (#508) 2023-07-19 15:07:56 -04:00
Luni-4 93723205e0
Add Sigmoid operator to burn-import (#501) 2023-07-17 12:45:44 -04:00
Luni-4 ffdb0feb29
Hide crate name in onnx-import crate and onnx-inference example (#493) 2023-07-13 19:39:17 -04:00
Dilshod Tadjibaev 825aaa9977
Add missing documents (#424) 2023-06-23 09:28:34 -04:00
Nathaniel Simard 40743c477a
Feat/onnx/more ops (#410) 2023-06-19 16:10:50 -04:00
Dilshod Tadjibaev 834c7ecc1f
Clean up cargo descriptions and formatting (#403) 2023-06-15 09:20:53 -04:00
Dilshod Tadjibaev 80d145d629
Add run-before-pr.sh script and fix clippy errors (#397) 2023-06-14 09:10:24 -04:00
uint256_t dc5aa41ea5
ONNX: Add `Bool` element type (#389) 2023-06-08 16:48:51 -04:00
Nathaniel Simard 3b6ba1bbab
Refactor/dim inference (#369) 2023-06-01 12:02:12 -04:00
Nathaniel Simard bda5c237b0
refactor: onnx parsing (#368) 2023-05-27 18:44:57 -04:00
Nathaniel Simard 3bd58d87ef
feat: burn import logging (#363) 2023-05-25 17:54:51 -04:00
Nathaniel Simard ba0ac112e1
Refactor: Burn Import use BurnGraph (#355) 2023-05-21 13:09:27 -04:00
Dilshod Tadjibaev 05763e1878
Bump version to the next minor to indicate dev (#344) 2023-05-10 18:02:08 -04:00
nathaniel 844b199dc1 Add burn import description 2023-05-06 10:45:58 -04:00
Nathaniel Simard 29eecd6383
Prepare next release (#335) 2023-05-06 10:32:23 -04:00
Dilshod Tadjibaev b7b4f321eb
Correct the struct name for ModelGen (#328) 2023-05-05 12:57:02 -04:00
Nathaniel Simard 6c9a5c8e58
Refactor/record (#323) 2023-05-04 14:59:16 -04:00
Dilshod Tadjibaev 39297a6479
Readme updates (#325) 
* Update text-generation readme for Mac users

* Update root readme to reference import crate

* Update import's readme

* Update torch backend
 2023-05-04 14:58:44 -04:00
Dilshod Tadjibaev 314db93b7f
Add ability to load onnx state to the generated source code (#319) 2023-05-03 13:05:43 -04:00
Dilshod Tadjibaev df980d534e
Add foundation for importing ONNX files (#297) 2023-04-15 10:44:50 -04:00