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
chentong319
Tung D. Le
Alexandre Eichenberger
Gheorghe-Teodor Bercea
Gheorghe-Teodor Bercea