InfiniTensor/include/operators
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
..
G2BMM.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
GBMM.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
batch_norm.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
concat.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
conv.h Add: ConvTransposed (#33) 2022-09-19 15:05:39 +08:00
element_wise.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
extend.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
gather.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
matmul.h Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
membound.h Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
pad.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
pooling.h ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
reshape.h Add: reshape/flatten/identity OP and cuda kernel (#34) 2022-09-21 14:04:30 +08:00
slice.h ADD:pad/slice operator and cuda kernel. (#39) 2022-09-29 10:29:24 +08:00
split.h ADD:concat/split operator and cuda kernels (#29) 2022-09-29 11:01:30 +08:00
unary.h Add activation operators and kernels 2022-09-16 13:58:57 +08:00