InfiniTensor/pyinfinitensor
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
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor fix: 移除中文注释 2023-03-15 15:16:16 +08:00
tests fix: 修正 batchNorm 实现 2023-03-15 17:23:32 +08:00
pyproject.toml feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00