Function registerKernel

  • registerKernel(config): void

  • Registers the function (forward pass) for the kernel in a global registry.

    Parameters

    • config: KernelConfig

      A config object with the following properties:

      • kernelName The official name of the kernel.
      • backendName The official name of the backend.
      • kernelFunc The function to run during the forward pass of the kernel.
      • setupFunc Optional. Gets called once, after the backend initializes.
      • disposeFunc Optional. Gets called once, right before the backend is disposed.

    Returns void

Settings

Member Visibility

Theme

Generated using TypeDoc