InfiniTensor/pyinfinitensor
PanZezhong1725 36ae7b7fb6
Add GatherElements op and cuda kernel (#149)
* Add GatherElements op and cuda kernel

* fix format

* remove print

* remove unused var

* fix spacing

* fix format

---------

Co-authored-by: panzezhong@qiyuanlab.com <panzezhong@zezhongpan>
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2023-10-12 09:18:12 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor Add GatherElements op and cuda kernel (#149) 2023-10-12 09:18:12 +08:00
tests Add GatherElements op and cuda kernel (#149) 2023-10-12 09:18:12 +08:00
pyproject.toml add onnx simplify (#153) 2023-10-10 15:45:27 +08:00