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
..
cuda_common.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
cuda_element_wise.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
cuda_kernel_wihtout_config.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
cuda_pad_slice.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
cuda_runtime.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
cuda_split_concat.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
cuda_unary.h	Add activation operators and kernels	2022-09-16 13:58:57 +08:00
cuda_utility.h	Simplify tensor transfer between CPU and CUDA (#10)	2022-08-25 11:29:16 +08:00
gather.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
gbmm_g2bmm.cuh	Fix CMake USE_CUDA (#36)	2022-09-21 12:28:00 +08:00
gbmm_g2bmm.h	Fix CMake USE_CUDA (#36)	2022-09-21 12:28:00 +08:00
operator_timer.h	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00