forked from jiuyuan/InfiniTensor
YdrMaster
a27391fcdc
- onnx 和 pytorch 认为 batchNorm 的 4 个参数是 [c] 形状的,cuDNN 可能认为是 [1,c,1,...]。 优化已改为 [c],但 cuDNN 推理没有改; Signed-off-by: YdrMaster <ydrml@hotmail.com> |
||
---|---|---|
.. | ||
docs | ||
src/pyinfinitensor | ||
tests | ||
pyproject.toml |