YdrMaster a27391fcdc fix: 修正 batchNorm 实现 ... - onnx 和 pytorch 认为 batchNorm 的 4 个参数是 [c] 形状的，cuDNN 可能认为是 [1,c,1,...]。 优化已改为 [c]，但 cuDNN 推理没有改； Signed-off-by: YdrMaster <ydrml@hotmail.com>		2023-03-15 17:23:32 +08:00
..
test_batch_norm.cc	fix: 修正 batchNorm 实现	2023-03-15 17:23:32 +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_reduce_mean.cc	ADD:reduce_mean operator and cuda kernel. (#47)	2022-10-15 16:53:58 +08:00
test_reshape.cc	ADD: batch norm operator and cuda kernel. (#44)	2022-10-15 16:29:28 +08:00
test_resize.cc	ADD: reconfig ResizeObj, support "tf_crop_and_resize " and cubic coeff kernel. (#59)	2022-12-24 04:02:21 +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