IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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

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​

Models

Cookbook

Puzzles

Blog

Community

Contact