InfiniTensor/pyinfinitensor
constroy Li 1e91979c76
add CUDNN impl for Min and Max (#118)
* add cudnn impl for Min and Max

* fix onnx _search_shape with output shape
2023-08-22 16:19:29 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor add CUDNN impl for Min and Max (#118) 2023-08-22 16:19:29 +08:00
tests support mixed dtype (#102) 2023-08-16 21:49:43 +08:00
pyproject.toml Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00