The GitHub Actions job "tvm-bot" on tvm.git/main has succeeded. Run started by GitHub user MasterJH5574 (triggered by MasterJH5574).
Head commit for run: 10ba3c232bfce87d7dc4f079c7717ac82ce7203a / Kryptonite <[email protected]> [Frontend][ONNX] Support select_last_index for ArgMax and ArgMin (#18969) ### Summary This PR implements the `select_last_index` attribute (introduced in opset 12) for the `ArgMax` and `ArgMin` ONNX operators. Previously, setting `select_last_index=1` raised `OpAttributeUnImplemented`. This closes the limitation tracked in the ONNX frontend issue. ### Implementation When `select_last_index=1`, the input tensor is reversed along the reduction axis using `relax.op.flip`, argmax/argmin is computed on the flipped copy, and the result is remapped back to the original index space via `last_idx = (axis_size - 1) - flipped_idx` Closes part of #18945 --------- Signed-off-by: OmarAzizi <[email protected]> Report URL: https://github.com/apache/tvm/actions/runs/23950071860 With regards, GitHub Actions via GitBox --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
