InfiniTensor/test
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
..
core Tensor serialization (#25) 2022-09-13 11:27:41 +08:00
kernels ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
nnet Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
operators ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
script cuDNN conv tuning (#16) 2022-08-29 21:37:07 +08:00