InfiniTensor/test/kernels
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
cuda ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00