For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo trait

TileTensorable

A DenseTensor that can project to a TileTensor view over its storage.

Lets the kernel work on a non-fused argument directly as a TileTensor (via to_tile_tensor) instead of going through the fused load/store/transform access.

Implemented traits

AnyType, DenseTensor, Tensor

comptime members

alignment

comptime alignment

dtype

comptime dtype

LayoutType

comptime LayoutType

mut

comptime mut

rank

comptime rank

ShapeType

comptime ShapeType

Required methods

to_tile_tensor

def to_tile_tensor(self) -> TileTensor[Self.dtype, Self.LayoutType, UntrackedOrigin[_Self.mut]]

Projects the argument to a TileTensor over its storage.

Returns:

TileTensor[_Self.dtype, _Self.LayoutType, UntrackedOrigin[_Self.mut]]: A TileTensor view with the argument's dtype, layout, and mut.

layout

def layout(self) -> Self.LayoutType

Returns the tensor's layout (shape and strides).

Returns:

_Self.LayoutType: The argument's LayoutType instance.

shape

def shape(self) -> Self.ShapeType

Returns the tensor's runtime shape.

Returns:

_Self.ShapeType: The shape as a Coord of the argument's ShapeType.