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 function

register_shape_function

def register_shape_function(name: StringSlice[ImmStaticOrigin])

Registers a standalone shape function for a MAX Graph op.

The decorated function computes the output shapes of the named op from its input shapes at compile time, without executing the op itself.

Args:

• ​name (StringSlice[ImmStaticOrigin]): The name of the op whose output shapes the function computes.