[BYOC][ONNX] Question about indices tensor of gather-scatter ops

[masahi]

@Nullko

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?

I don’t see an easy way for this, may be you can write a Relay pass that does the cast?

@Nullko

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.

Good point, a PR welcome.