forked from jiuyuan/InfiniTensor
a4d6426589
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. |
||
---|---|---|
.. | ||
G2BMM.h | ||
GBMM.h | ||
batch_norm.h | ||
concat.h | ||
conv.h | ||
element_wise.h | ||
extend.h | ||
gather.h | ||
matmul.h | ||
membound.h | ||
pad.h | ||
pooling.h | ||
reshape.h | ||
slice.h | ||
split.h | ||
unary.h |