Method __enter__

python/tvm/tirx/script/builder/frame.py:43–45

(self)

@_register_object("script.ir_builder.tirx.ForFrame")
class ForFrame(TIRFrame):
    def __enter__(self) -> Var | list[Var]:  # type: ignore[override]
        super().__enter__()
        return self.vars if len(self.vars) > 1 else self.vars[0]


@_register_object("script.ir_builder.tirx.AssertFrame")