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 module

conv

Registers convolution graph ops and dispatches them to the nn.conv kernels.

comptime values

logger

comptime logger = Logger(stdout, prefix=String(""), source_location=False)

Structs

• ​AvgPool: Registers the mo.avg_pool graph op with the graph compiler.
• ​AvgPoolCeilModeTrue: Registers the mo.avg_pool_ceil_mode_true graph op with the graph compiler.
• ​Conv: Registers the mo.conv graph op with the graph compiler.
• ​Conv2dResidualAdd: Fused conv2d + TMA residual add + bias for SM100 (Blackwell).
• ​ConvertE4M3FNToE4M3FNUZ: Registers the mo.convert_e4m3fn_to_e4m3fnuz graph op with the graph compiler.
• ​ConvTranspose: Registers the mo.conv_transpose graph op with the graph compiler.
• ​LayoutTransformFCQRS2FQRSCf: Registers the layout_transform_FCQRS_to_FQRSCf graph op with the graph compiler.
• ​LayoutTransformFCRS2FRSCf: Registers the layout_transform_FCRS_to_FRSCf graph op with the graph compiler.
• ​LayoutTransformQRSCF2FQRSCf: Registers the layout_transform_QRSCF_to_FQRSCf graph op with the graph compiler.
• ​LayoutTransformQRSFC2FQRSCf: Registers the layout_transform_QRSFC_to_FQRSCf graph op with the graph compiler.
• ​LayoutTransformRSCF2FRSCf: Registers the layout_transform_RSCF_to_FRSCf graph op with the graph compiler.
• ​LayoutTransformRSFC2FRSCf: Registers the layout_transform_RSFC_to_FRSCf graph op with the graph compiler.
• ​PackConvFilterShape: Registers the pack_conv_filter_shape graph op with the graph compiler.
• ​PadConstant: Registers the mo.pad.constant graph op with the graph compiler.
• ​PadReflect: Registers the mo.pad.reflect graph op with the graph compiler.
• ​PadRepeat: Registers the mo.pad.repeat graph op with the graph compiler.

Functions

• ​avg_pool_ceil_mode_true_shape: Computes the output shape for the mo.avg_pool_ceil_mode_true graph op.
• ​avg_pool_shape: Computes the output shape for the mo.avg_pool graph op.
• ​conv2d_residual_add_shape: Computes the output shape for the conv2d_residual_add graph op.
• ​convert_e4m3fn_to_e4m3fnuz_shape: Computes the output shape for the mo.convert_e4m3fn_to_e4m3fnuz graph op.
• ​mo_conv_shape: Computes the output shape for the mo.conv graph op.
• ​mo_conv_transpose_shape: Computes the output shape for the mo.conv_transpose graph op.
• ​pack_conv_filter_shape_fn: Compute the output shape of convolution filter packing.
• ​pad_constant_shape: Computes the output shape for the mo.pad.constant graph op.
• ​pad_reflect_shape: Computes the output shape for the mo.pad.reflect graph op.
• ​pad_repeat_shape: Computes the output shape for the mo.pad.repeat graph op.