Should relay.nn operators be documented in the language reference

AndrewZhaoLuo · January 28, 2022, 9:21pm

I would suggest printing the keys from the actual op registry via packed func interface (it’s just a Map):

e.g.

AndrewZhaoLuo/TVM-Sandbox/blob/main/relay/example_list_all_ops.py

"""
How to list all ops registered so far for quantization and in a generic registry.
"""

import tvm

op_names = tvm.ir.Op.list_op_names()

print("Ops:")
for name in sorted(op_names):
    print(name)
print()

print("Those with registered FTVMFakeQuantizationToInteger")
for name in sorted(op_names):
    op = tvm.ir.Op.get(name)
    if op.get_attr("FTVMFakeQuantizationToInteger"):
        print(name)