Skip to main content

function

custom

custom[name: StringLiteral](values: List[Symbol], out_type: Type) -> Symbol

Creates a node to execute a custom graph operation in the graph.

The custom op should be registered by annotating a function with the max.register.op decorator.

Parameters:

  • name (StringLiteral): The op name provided to max.register.op.

Args:

  • values (List[Symbol]): The op function's arguments.
  • out_type (Type): The op function's return type.

Returns:

A symbolic value representing the output of the op in the graph.