994f8a9c2a
Fixed layout inference crash(assert) if node have multiply output
...
Signed-off-by: xiang.zhang <xiang.zhang@verisilicon.com>
2021-09-16 10:55:29 +08:00
fae5cede7a
Support layout inference for ops ( #77 )
...
Signed-off-by: yuenan.li <yuenan.li@verisilicon.com>
Co-authored-by: yuenan.li <yuenan.li@verisilicon.com>
2021-05-27 10:33:44 +08:00
410cd8e516
Refine the cmake build ( #63 )
...
Signed-off-by: xiang.zhang <xiang.zhang@verisilicon.com>
2021-05-24 13:40:37 +08:00
b38cad9f1d
Add data layout for kernel to support TVM conv2d ( #40 )
...
Signed-off-by: Zongwu Yang <zongwu.yang@verisilicon.com>
2021-05-14 14:00:02 +08:00
748274143b
support layout inference for operations ( #39 )
...
Add layout inference support for space2depth, depth2space, space2batch, batch2space, pad and
reduce.
Signed-off-by: yuenan.li <yuenan.li@verisilicon.com>
Co-authored-by: yuenan.li <yuenan.li@verisilicon.com>
2021-05-13 22:27:23 +08:00
a42517fdce
Align directory name to namespace for layout inference ( #38 )
...
Signed-off-by: xiang.zhang <xiang.zhang@verisilicon.com>
2021-05-11 09:46:46 +08:00
xiang.zhang
liyuenan
Zongwu.Yang