InfiniTensor/include/utils
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
..
data_generator.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
dataloader.h Json perfrecord (#32) 2022-09-22 15:34:34 +08:00
exception.h add code for backtrace (#21) 2022-09-01 20:30:12 +08:00
validation.h Verify wanghailu 0902 (#22) 2022-09-05 15:45:52 +08:00