tvm_tirx.ir.register_op_attr

tvm_tirx.ir.register_op_attr#

tvm_tirx.ir.register_op_attr(op_name, attr_key, value=None, override=False)#

Register an operator property of an operator by name.

Parameters:

  • op_name (str) – The name of operator

  • attr_key (str) – The attribute name.

  • value (object, optional) – The value to set

  • override (bool, optional) – Replace an existing value if True; otherwise duplicate registration raises ValueError. Cached views observe replacements; no priority history is kept.

Returns:

result (object or function) – The registered value when supplied, or a decorator that registers and returns its argument. The named Op must already be registered.