The issue is similar to #7029 .
I am not able to use .keys() on tvm.relay.attrs.ArgReduceAttrs object.
I checked in tvm/python/relay/op/op_attrs.py there is no python class ArgReduceAttrs.
I have created https://github.com/apache/tvm/pull/9606 this PR resolving this issue
Thanks,
-Onkar