[Relax][Frontend][Onnx] Cast Op special handling for ShapeExpr input

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

https://tvm.apache.org/

Apache License 2.0

11.42k stars 3.4k forks source link

Closed tsu-bin closed 1 month ago

tsu-bin commented 1 month ago

The Onnx model I'm working on has Cast Op with static ShapeExpr as its input. This PR updates Cast Op converter to handle this sepcial case.