InfiniTensor/test
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
..
core feat: GraphObj 增加一个拓扑排序方法及其测试 2023-03-15 15:09:12 +08:00
kernels Add ConvTransposedNHWC (#67) 2023-03-01 14:15:02 +08:00
nnet Add ConvTransposedNHWC (#67) 2023-03-01 14:15:02 +08:00
operators fix: 修正 batchNorm 实现 2023-03-15 17:23:32 +08:00
script cuDNN conv tuning (#16) 2022-08-29 21:37:07 +08:00