InfiniTensor/pyinfinitensor/tests
zhangyunze ef672894d0
support mixed dtype (#102)
* feat: support mixed dtype

* feat: support cast op

* test: add test for cast op

* feat: support datatype BFloat16

* feat: support data convert fp32 <-> bfp16

* fix: fix all op's infershape func

* fix as review comment
2023-08-16 21:49:43 +08:00
..
test_onnx.py support mixed dtype (#102) 2023-08-16 21:49:43 +08:00