* add random normal like python code to generate onnx model
* add random normal like node
* modify onnx burn to add new op
* add test on test onnx
* revert commentouts
* fix review points to respond to dynamically shape
* add onnx op of random uniform like
* fix formats
* add random normal like python code to generate onnx model
* add random normal like node
* modify onnx burn to add new op
* add test on test onnx
* revert commentouts
* fix review points to respond to dynamically shape
* add python, onnx model, burn/node for conv_transpose1d
* modify supported-onnx-ops.md about convtranspose1d
* add conv_transpose1d function in op_configuration and to_burn
* add test and modify dim_inference
* apply cargo fmt to formatting
* fix reviewer point-outs by adding symetry checks for padding in 2d,3d
* fix pads initilization and check ways
* add trilu.py and onnx model for operators
* add node/trilu.rs under burn-import and modify related files
* update trilu codegen tests
* add op_configuration and to_burn rust files
* update how to get diagonal value from nodes argument
* add trilu tests in test_onnx.rs
* update files to follow clippy and format rules
* add tests for both of upper and lower to follow review comment
* delete onnx model of trilu.onnx since upper and lower models are added
* renaming repeat to repeat_dim
* implementing repeat function
* renaming repeat files to repeat_dim
* renaming part 2
* renaming part 3
* renaming part 4
* renaming part 5
* adding test file
* adding unit test
* adding rust book documentation
* adding function args doc
* fixing tests
* changing repeat api to match pytorch equivalent
* fixing clippy error
* implementing tile onnx file
* temp
* working implementation and test
* working e2e test
* adding new supported onnx operation to the md file
* Add Hard Sigmoid activation function
* Add ONNX import conversion for HardSigmoid
* Update supported operators list
* Update book
* Make test comparison approximate to eliminate precision issues
* Add burn-candle test
* Fix name in E2E test generator
* make contacts deterministic across Worlds
* add top k acc
* add onnx mean
* fix
* push fix
* format
---------
Co-authored-by: Charles Bournhonesque <cbournhonesque@snapchat.com>
* Feat: burn-import implement ONNX ConstantOfShape
* Introduce shape type and use in ConstantOfShape and Shape
* Add tests for bool and int tensors for ConstantOfShape
* Fix ONNX test generation
* Undo comment
---------
Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
* Implement 3D and transposed 3D convolutions.
* Merge changes from onnx-ir #1921 pr
---------
Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
* Preliminary ReduceProd Support
* Add comma to keep formatter happy
* Give test results a 0.001 tolerance to account for floating-point multiplication
* Reformat assersions
* Correctly mark panic conditions in op_configuration
* Move and redirect GatherElements to new folders/nodes
* Create PyTorch script for gather
* Add onnx file for gather
* Add a gather test to onnx_tests
* Update gather.rs to use select
* Rename codegen test
* Update gather and gather_elements conversion functions
* Validate rank of input node and update output
* Add check for Gather
* pre-test
* implementing argmax for burn-import from onnx
* tidying
* fixing return types and tests
* addressing feedback
* only warn when select_last_index!=0
* Add where onnx op support
* Add broadcasting support
* Remove broadcasting limitation comment
* Fix broadcasting in mask where
* Forgot to reflect changes in codegen test
* Fix clippy
* Mul onnx op already supported
* Add matmul onnx op checks and tests
* Add missing eq derives
* Change supscript symbol
* Remove dead code
* Add support for matmul broadcast
* No more broadcasting restrictions
* Add results comment for mm, mv and vm
* Implement LeakyReLu
* Cargo fmt
* Apply suggestions
* cargo fmt
* Use float_mul_scalar
* Should be grad
* Add to books module
* Move test files
* Update leaky relu to use activation function
* Update tensor.md
* Fix failing test due to approx
* Add back the function comment
* Fix comment per PR feedback
---------
Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
tiruka