InfiniTensor/pyinfinitensor
panzezhong 6c4dd7b28b fix(front): 将stub改为可以接收GraphProto作为输入,消除分布式脚本保存额外的onnx文件, 采用int64作为index输入类型 2024-04-07 17:15:40 +08:00
..
docs feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
src/pyinfinitensor fix(front): 将stub改为可以接收GraphProto作为输入,消除分布式脚本保存额外的onnx文件, 采用int64作为index输入类型 2024-04-07 17:15:40 +08:00
tests fix mlu some kernel registration & gather op (#210) 2024-02-01 15:02:02 +08:00
pyproject.toml Change onnx-simplifier to onnxsim to resolve build issue on xpu (#164) 2023-10-21 02:58:32 +08:00