tvm_tirx.script.ir_builder.Lambda

tvm_tirx.script.ir_builder.Lambda(parameter_types, function, *, ret_type=None)

Build a shared staging lambda from explicit types and a Python callable.

Scalar constructors such as T.float32 may be used as parameter types. An optional return annotation checks the body type without inserting casts.