(kernelName: string)
| 180 | |
| 181 | /** Removes the registered gradient from the global registry. */ |
| 182 | export function unregisterGradient(kernelName: string): void { |
| 183 | if (!gradRegistry.has(kernelName)) { |
| 184 | throw new Error( |
| 185 | `The gradient '${kernelName}' for backend is not registered`); |
| 186 | } |
| 187 | gradRegistry.delete(kernelName); |
| 188 | } |
| 189 | |
| 190 | /** |
| 191 | * Finds kernels that have already been registered to a backend and re-registers |