e32f531546
Add the split marker to test files ( #90 )
...
Co-authored-by: Tian Jin <tjingrant@gmail.com>
2020-04-16 15:17:27 +08:00
fa8962753c
Concat lower ( #82 )
...
* implement shape inference for concat
* better checking of axis being concatenated: constant values only
* lowering of Concat with lit and backend tests
* fixes
Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-04-13 11:40:39 -04:00
f5bed72e13
implement shape inference for concat ( #74 )
...
* implement shape inference for concat
* better checking of axis being concatenated: constant values only
2020-04-07 16:13:41 -04:00
653fa69102
Unify Conv implementation ( #54 )
...
* fixed readme for new git repo
* conv with bias as an optional input
2020-03-26 11:03:19 -04:00
1622b9f161
[NFC] Change ONNF based names to ONNX-MLIR ( #32 )
...
* Rename onnf to onnx-mlir.
* Change workspace name.
2020-03-17 09:16:33 -04:00
d86591d61a
Import all initialized tensors as dense constants ( #30 )
...
* Import initialized tensor as dense attribute
* Import all initialize tensors as dense constants
* Remove unintentional code
* Fix value attribute format in shape inference tests of reshape
* Readd rank check for reshape's shape inference
* Remove a redundant variable
Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-03-16 11:17:28 -04:00
c46880d5c6
Fix reshape output shape inference when a single dynamic shape is given ( #22 )
...
* Fix reshape when a dynamic shape is given.
* Fix default attributes for ConvNoBias.
* Fix comment.
* Resolve comment.
* Improve checks.
* Handle zero dim case.
* Add helper to fetch constants. Add test for dynamic reshape.
* Add test for zero.
* Use shortcut method for size.
2020-03-13 17:18:46 -04:00
6137fc7c17
Fix issues #15 and #16 ( #29 )
...
* fix issue #15 and #16
* fix format
Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-03-13 10:19:27 -04:00
362491553c
Shape inference for ONNXAveragePool ( #21 )
...
* Shape inference for ONNXAveragePool
* Edit comments and puts helper function on top of the file
* Fix template
2020-03-13 09:59:16 -04:00
a65820940c
Lower ConstantOp ( #28 )
...
* Lower ConstantOp
* Refactor the code
* Edit error messages
* Check whether attribute is sparse or dense during shape inference
2020-03-12 10:58:42 -04:00
811b63e031
Inter common pad ( #26 )
...
* common pad handling in shape inference for conv and maxpool
* common pads
Co-authored-by: Gheorghe-Teodor Bercea <gt.bercea@gmail.com>
2020-03-11 18:36:02 -04:00
e8a0b47e10
Fix case for upper and lower padding when strides are present. ( #11 )
...
* Fix case for upper and lower padding when strides are present.
* Address comments.
* Code clean-up.
* Fix tests.
2020-03-10 16:58:05 -04:00
bbdf4e3b4d
Merge remote-tracking branch 'upstream/master' into shapeinference-pad
...
Conflicts:
test/mlir/onnx/onnx_shape_inference.mlir
2020-02-14 15:35:47 -05:00
17d84901b7
Allow 1-D convolutions. ( #86 )
...
* Fix check.
* Fix comment.
2020-02-14 10:54:08 -05:00
c3041bfb43
shape inference for pad with constant pads
2020-02-13 19:56:05 -05:00
f3047943a1
Handle 1-D MATMUL N-D ( #56 )
2020-01-29 10:35:05 -05:00
3f5c543782
Merge branch 'master' into matmul-shape
2020-01-27 11:37:40 -05:00
c74f814f64
Add attributes as operation parameters ( #45 )
...
* add attributes of Op into parameters
* fix rewrite rule for GemmOp with attributes
* use I64Attr instead of I32Attr and modify test cases for the changes in attributes
* add output name (prefixed with o_) to Op definition
* update shape inference for the new attributes
2020-01-27 10:09:14 -05:00
07d28769d3
Merge remote-tracking branch 'origin/master' into matmul-shape
2020-01-23 11:53:53 -05:00
7dda698e7e
Add test with dilations and auto padding set to SAME_UPPER.
2020-01-22 16:40:33 -05:00
68efd21064
Fix dilation formula in the code.
2020-01-22 16:34:59 -05:00
050d7d277d
Fix test.
2020-01-22 16:12:09 -05:00
0bc07ef661
Merge remote-tracking branch 'origin/master' into matmul-shape
2020-01-22 15:29:09 -05:00
ea45cbcca9
Add support for dilations attribute and add tests.
2020-01-22 14:40:10 -05:00
f0b484c0bc
Add test for transpose with permutation.
2020-01-20 14:54:40 -05:00
da0e9b01b1
Fix 1 and 2 dimensional cases. Add test for 1 and 2 dimensional combinations.
2020-01-14 10:47:24 -05:00
642f77abed
Add additional dynamic dimension.
2020-01-14 10:47:24 -05:00
95ebf3e23a
Add test for multypling stacks of matrices.
2020-01-14 10:47:24 -05:00
ae966cdee9
Add tests for matrices and stack of matrices combinations.
2020-01-14 10:47:24 -05:00
a5f1d39c20
Add tests for matrices and stack of matrices combinations.
2020-01-14 10:47:24 -05:00
6478c88cdc
Add test for all one dimensional case.
2020-01-14 10:47:24 -05:00
151f4f8c44
Add the default shape inference for the transposition operation.
2020-01-09 13:50:38 -05:00
Tung D. Le
Alexandre Eichenberger
Gheorghe-Teodor Bercea
chentong319
Gheorghe-Teodor Bercea