MCPcopy Index your code
hub / github.com/tensorflow/tfjs / registerKernel

Function registerKernel

tfjs-core/src/kernel_registry.ts:131–140  ·  view source on GitHub ↗
(config: KernelConfig)

Source from the content-addressed store, hash-verified

129 * disposed.
130 */
131export function registerKernel(config: KernelConfig) {
132 const {kernelName, backendName} = config;
133 const key = makeKey(kernelName, backendName);
134 if (kernelRegistry.has(key)) {
135 log.warn(
136 `The kernel '${kernelName}' for backend ` +
137 `'${backendName}' is already registered`);
138 }
139 kernelRegistry.set(key, config);
140}
141
142/**
143 * Registers a gradient function for a given kernel in the global registry,

Calls 3

makeKeyFunction · 0.85
hasMethod · 0.80
setMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…