tvm_tirx.script.ir_builder.module_set_attr

tvm_tirx.script.ir_builder.module_set_attr#

tvm_tirx.script.ir_builder.module_set_attr(attr_key, attr_value, allow_overwrite=False)#

Set the specified attr of the ir_module frame. :param attr_key: The key of the attr to be set. :type attr_key: str :param attr_value: The value of the attr to be set. :type attr_value: Optional[Object] :param allow_overwrite: Whether allow overwrite the existing attr. :type allow_overwrite: bool

Parameters:
  • attr_key (str)

  • attr_value (Object | None)

  • allow_overwrite (bool)

Return type:

None