For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/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
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.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!