tvm_tirx.tirx.script.grid

Contents

tvm_tirx.tirx.script.grid#

tvm_tirx.tirx.script.grid(*extents, dtype=None)#

The grid For statement.

Parameters:

  • extents (Tuple[Union[Expr, Tuple[Expr, Expr]]]) – If a single Expr is provided, it is used as the extent of the iteration. If a tuple of two Expr is provided, the first is the start of the iteration, and the second is the extent of the iteration.

  • dtype (str, optional) – The dtype of every loop variable, either "int32" or "uint32". When omitted each loop variable takes the dtype of its own extent.

Returns:

res (frame.ForFrame) – The ForFrame.

Return type:

ForFrame

Contents