Commit Graph

12 Commits

Author SHA1 Message Date
qin.chen 9ebddb5452 add op: maxpoolwithargmax2 and maxpoolgrad 2022-07-29 11:11:33 +08:00
MESeraph 11f953b506
Mapped roi_pool & added unit test (#404)
* Mapped roi_pool & added unit test

* modify roialign/roipool unit test
2022-05-30 19:57:50 +08:00
MESeraph 6d0c6b01b5
modify GatherElements (#406) 2022-05-29 22:25:14 +08:00
Antkillerfarm b916e1301a
Add Broadcast op (#365) 2022-04-18 15:45:15 +08:00
Sven 171abb0f1b
Revert "composed Dense & added unit test (#312)" (#340)
This reverts commit f2e71a3deb.
2022-03-31 18:37:45 +08:00
chxin66 f2e71a3deb
composed Dense & added unit test (#312)
if shape is 3D or larger, implement it as reshape + fc + reshape

Signed-off-by: Chen Xin <jack.chen@verisilicon.com>

Co-authored-by: Chen Xin <jack.chen@verisilicon.com>
2022-03-25 10:49:39 +08:00
Zhouheng Zheng b02aa8b8c4
Added customize operator APIs(#315)
Co-authored-by: zhouheng.zheng <zhouheng.zheng@ouotlook.com>
2022-03-09 12:10:08 +08:00
Zongwu.Yang 4229ad88b3
support conv3d (#238)
Signed-off-by: Zongwu Yang <zongwu.yang@verisilicon.com>
2022-01-11 14:13:15 +08:00
liyuenan 7c63ba621e
Map OneHot & unit test (#258)
Signed-off-by: yuenan.li <yuenan.li@verisilicon.com>

Co-authored-by: yuenan.li <yuenan.li@verisilicon.com>
2022-01-05 22:04:49 +08:00
chxin66 cea11422b8
Added RNNCell & unit test (#249)
Signed-off-by: Chen Xin <jack.chen@verisilicon.com>

Co-authored-by: Chen Xin <jack.chen@verisilicon.com>
2021-12-29 11:08:24 +08:00
chxin66 516a914c73
Mapped Erf operation & unit tests (#211)
Signed-off-by: Chen Xin <jack.chen@verisilicon.com>

Co-authored-by: Chen Xin <jack.chen@verisilicon.com>
2021-11-10 20:07:06 +08:00
lilei 073a79f463 add ops.h to contain all operation header file 2021-09-17 22:41:44 +08:00