InfiniTensor/pyinfinitensor
Haojie Wang fab08fcfa1
Constroy/fix reducemean (#91)
* feat: 导出 getPerfTime 到 python

Signed-off-by: YdrMaster <ydrml@hotmail.com>

* fix parsing of ReduceMean

* ReduceMean axes defaults to None

* fix ElementWiseCudnn with shape broadcasting

* fix format

---------

Signed-off-by: YdrMaster <ydrml@hotmail.com>
Co-authored-by: YdrMaster <ydrml@hotmail.com>
Co-authored-by: constroy <constroy.li@gmail.com>
2023-06-29 08:27:00 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor Constroy/fix reducemean (#91) 2023-06-29 08:27:00 +08:00
tests fix: 正确导入 conv 的 bias 2023-06-27 17:37:21 +08:00
pyproject.toml Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00