[quote="Nullko, post:1, topic:11778"] My accelerator runtime’s Gather-Scatter ops require `i32` indices tensors, however, by default Relay uses `i64` indices, is there a simple way to set all indices tensors in a Relay graph to `i32` dtype? [/quote]
I don't see an easy way for this, may be you can write a Relay pass that does the cast? [quote="Nullko, post:1, topic:11778"] However, the indices tensors produced by ops like `argmax/argmin/topk` are guaranteed to have positive values. It would be better to add a rule to check whether the indices tensor is produced by those ops and skip the “normalization” in such scenario. [/quote] Good point, a PR welcome. --- [Visit Topic](https://discuss.tvm.apache.org/t/byoc-onnx-question-about-indices-tensor-of-gather-scatter-ops/11778/2) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/c39b9a182994ff0ca567ed2cb9f4d7d7d5c2bd8582e695dbd3220ab4b707f215).
