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 module

gather_scatter

Registers gather, scatter, and indexed-access graph ops backed by the nn.gather_scatter kernels.

`comptime` values

`logger`

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

Structs

AdvancedIndexingGetItem: Registers the advanced_indexing_getitem graph op with the graph compiler.
AdvancedIndexingSetItem: Registers the advanced_indexing_setitem graph op with the graph compiler.
AdvancedIndexingSetItemInplace: Registers the advanced_indexing_setitem_inplace graph op with the graph compiler.
ApplyPackedBitmask: Registers the mo.apply_packed_bitmask graph op with the graph compiler.
BroadcastShape: Registers the mo.broadcast_shape graph op with the graph compiler.
BroadcastTo: Registers the mo.broadcast_to graph op with the graph compiler.
Concat: Registers the mo.concat graph op with the graph compiler.
DualFusedConcatSlice: Registers the mo.dual_fused_concat_slice graph op with the graph compiler.
FusedConcatSlice: Registers the mo.composite.concat_slice graph op with the graph compiler.
Gather: Registers the mo.gather graph op with the graph compiler.
GatherND: Registers the mo.gather_nd graph op with the graph compiler.
GatherSum: Registers the mo.gather_sum graph op with the graph compiler.
IndexTensor: Registers the index_tensor graph op with the graph compiler.
MutableStore: Registers the mo.mutable.store graph op with the graph compiler.
MutableStoreSlice: Registers the mo.mutable.store.slice graph op with the graph compiler.
Reshape: Registers the mo.reshape graph op with the graph compiler.
Scatter: Registers the mo.scatter graph op with the graph compiler.
ScatterAdd: Registers the mo.scatter.add graph op with the graph compiler.
ScatterMax: Registers the mo.scatter.max graph op with the graph compiler.
ScatterMin: Registers the mo.scatter.min graph op with the graph compiler.
ScatterMul: Registers the mo.scatter.mul graph op with the graph compiler.
ScatterND: Registers the mo.scatter_nd graph op with the graph compiler.
ScatterNDAdd: Registers the mo.scatter_nd.add graph op with the graph compiler.
ScatterNDMax: Registers the mo.scatter_nd.max graph op with the graph compiler.
ScatterNDMin: Registers the mo.scatter_nd.min graph op with the graph compiler.
ScatterNDMul: Registers the mo.scatter_nd.mul graph op with the graph compiler.
ScatterNDSkipNegIndices: Registers the mo.scatter_nd.skip_neg_indices graph op with the graph compiler.
ScatterSetConstant: Registers the mo.scatter_set_constant graph op with the graph compiler.
ShardWeights: Registers the mo.shard_and_stack graph op with the graph compiler.
Slice: Registers the mo.slice graph op with the graph compiler.
Split: Registers the mo.split graph op with the graph compiler.
SplitOutputShapeHelper: Registers the split_ith_output_shape graph op with the graph compiler.
SqueezeShape: Registers the mo.squeeze_shape graph op with the graph compiler.
StaticBroadcastTo: Registers the mo.static.broadcast_to graph op with the graph compiler.
StaticReshape: Registers the mo.static.reshape graph op with the graph compiler.
Struct_sliced_add_ragged: Registers the mo.sliced.add.ragged graph op with the graph compiler.
Tile: Registers the mo.tile graph op with the graph compiler.
Transpose: Registers the mo.transpose graph op with the graph compiler.
UnsqueezeShape: Registers the mo.unsqueeze_shape graph op with the graph compiler.

Functions

advanced_indexing_getitem_shape_fn: Computes the output shape for the advanced_indexing_getitem graph op.
broadcast_shape_fn: Computes the output shape for the mo.broadcast_shape graph op.
broadcast_to_shape_fn: Computes the output shape for the mo.broadcast_to graph op.
check_axis_in_range: Indices passed to gather and scatter ops may be negative. This performs a check to see if the axis is valid.
concat_shape_fn: Computes the output shape for the mo.concat graph op.
gather_nd_shape_fn: Computes the output shape for the mo.gather_nd graph op.
gather_shape_fn: Computes the output shape for the mo.gather graph op.
reshape_shape_fn: Computes the output shape for the mo.reshape graph op.
scatter_add_shape_fn: Computes the output shape for the mo.scatter.add graph op.
scatter_max_shape_fn: Computes the output shape for the mo.scatter.max graph op.
scatter_min_shape_fn: Computes the output shape for the mo.scatter.min graph op.
scatter_mul_shape_fn: Computes the output shape for the mo.scatter.mul graph op.
scatter_nd_add_shape: Computes the output shape for the mo.scatter_nd.add graph op.
scatter_nd_max_shape: Computes the output shape for the mo.scatter_nd.max graph op.
scatter_nd_min_shape: Computes the output shape for the mo.scatter_nd.min graph op.
scatter_nd_mul_shape: Computes the output shape for the mo.scatter_nd.mul graph op.
scatter_nd_shape_fn: Computes the output shape for the mo.scatter_nd graph op.
scatter_shape_fn: Computes the output shape for the mo.scatter graph op.
slice_shape_fn: Computes the output shape for the mo.slice graph op.
split_ith_output_shape_fn: Computes the output shape for the split_ith_output_shape graph op.
squeeze_shape_fn: Computes the output shape for the mo.squeeze_shape graph op.
tile_shape_fn: Computes the output shape for the mo.tile graph op.
transpose_shape_fn: Computes the output shape for the mo.transpose graph op.
unsqueeze_shape_fn: Computes the output shape for the mo.unsqueeze_shape graph op.

comptime values​

logger​

Structs​

Functions​

`comptime` values

`logger`

Structs

Functions