brainstate.transform.register_prim_handler

brainstate.transform.register_prim_handler#

brainstate.transform.register_prim_handler(prim_name, handler)[source]#

Register a handler for a primitive for automin

Parameters:
  • prim_name (str) – Name of the primitive.

  • handler (Callable) – Handler for the primitive.

Return type:

None