InfiniTensor/test/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
..
test_batch_norm.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_concat.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_conv.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_conv_transposed_2d.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_element_wise.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_extend.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_gather.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_matmul.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_pad.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_pooling.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_reshape.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_slice.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test_split.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00