wendy12022 26cee55e81 ADD:extend operator and cuda kernel. (#40) ... Co-authored-by: Haojie Wang <haojie0429@gmail.com>		2022-09-29 14:52:50 +08:00
..
G2BMM.h	Operators g2bmm&gbmm transplantation (#24)	2022-09-08 21:31:35 +08:00
GBMM.h	Operators g2bmm&gbmm transplantation (#24)	2022-09-08 21:31:35 +08:00
concat.h	ADD:concat/split operator and cuda kernels (#29)	2022-09-29 11:01:30 +08:00
conv.h	Add: ConvTransposed (#33)	2022-09-19 15:05:39 +08:00
element_wise.h	ADD add/mul/sub/div/pow operators and CPU/CUDA kernels (#26)	2022-09-09 13:43:59 +08:00
extend.h	ADD:extend operator and cuda kernel. (#40)	2022-09-29 14:52:50 +08:00
gather.h	ADD: Gather operator and cuda kernel. (#41)	2022-09-29 14:44:20 +08:00
matmul.h	Fix NNet tests after migration (#27)	2022-09-13 15:17:22 +08:00
membound.h	Fix NNet tests after migration (#27)	2022-09-13 15:17:22 +08:00
pad.h	ADD:concat/split operator and cuda kernels (#29)	2022-09-29 11:01:30 +08:00
pooling.h	Add maxpool and avgpool operators (#17)	2022-08-31 14:44:53 +08:00
reshape.h	Add: reshape/flatten/identity OP and cuda kernel (#34)	2022-09-21 14:04:30 +08:00
slice.h	ADD:pad/slice operator and cuda kernel. (#39)	2022-09-29 10:29:24 +08:00
split.h	ADD:concat/split operator and cuda kernels (#29)	2022-09-29 11:01:30 +08:00
unary.h	Add activation operators and kernels	2022-09-16 13:58:57 +08:00