tvm_tirx.tirx.layout

Contents

tvm_tirx.tirx.layout#

Represent tensor layouts, iteration axes, and mappings between logical and physical storage.

Classes#

Axis(*args, **kwargs)

A named physical coordinate used by a layout.

Iter(extent, stride, axis)

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

Alias

Canonical API

Layout

tvm_tirx.tirx.Layout

TileLayout

tvm_tirx.tirx.TileLayout

ComposeLayout

tvm_tirx.tirx.ComposeLayout