xlnstorch.set_default_implementation

xlnstorch.set_default_implementation#

xlnstorch.set_default_implementation(torch_function, impl_key)#

Set the default implementation for a given torch function.

Parameters:

  • torch_function (Callable) – The torch function for which to set the default implementation.

  • impl_key (str) – The key identifying the implementation to be set as default.

Raises:

ValueError – If no implementations are registered for the given torch function. If the specified implementation key is not registered for the torch function.

Return type:

None

Contents