tvm_tirx.tirx.layout.Iter

class tvm_tirx.tirx.layout.Iter(extent, stride, axis)

Bases: Object

One extent-and-stride factor contributing to a layout axis.

Parameters:

• extent (Expr) – Number of positions in this factor.

• stride (Expr) – Increment on the output axis for each position.

• axis (Axis or str) – Output coordinate receiving the contribution, resolved by name for strings.

property extent

property stride

property axis