wendy12022 a4d6426589 ADD: batch norm operator and cuda kernel. (#44) ... fix numInputs of batchNorm, add new line in file ending. ADD: batch norm operator and cuda kernel. add training remove comments. fix compile error. add batch norm operator and cuda kernel.		2022-10-15 16:29:28 +08:00
..
bang	Support bang c kernel wanghailu 0927 (#43)	2022-09-30 11:01:52 +08:00
core	Add: python API for timing ConvTranspose (#46)	2022-10-07 16:03:11 +08:00
cuda	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
ffi	Add TVM codegen for MemboundOp (#35)	2022-09-22 18:06:45 +08:00
nnet	Fix NNet tests after migration (#27)	2022-09-13 15:17:22 +08:00
operators	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
utils	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
test.h	Add python interface for CUDA operator evaluation (#42)	2022-09-27 10:41:12 +08:00