code: def run(f):
      with ir.Context() as context:
          mhlo.register_mhlo_dialect(context)
          f()
      return f
  
file_path: tensorflow/third_party/xla/xla/mlir_hlo/tests/python/attributes.py
imports: from mlir import ir
  from mlir.dialects import mhlo
label_desc: Use image node_id 1 for calling a global function with example usage: run(f) and returns: f