InfiniTensor/pyinfinitensor
constroy 579cdbbb81
fix ReduceMean and element_wise (#90)
* 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>
2023-06-29 07:15:07 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor fix ReduceMean and element_wise (#90) 2023-06-29 07:15:07 +08:00
tests Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
pyproject.toml Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00