oidn-web 0.3.4 → 0.4.0

package/src/helper.ts

@@ -1,43 +0,0 @@
-import * as tfjs from '@tensorflow/tfjs-core';
-export function profileAndLogKernelCode(execute: () => void, disabled = true) {
-  if (disabled) {
-    execute();
-    return;
-  }
-  tfjs
-    .profile(() => {
-      execute();
-    })
-    .then((res) => {
-      const kernelNames = Array.from(
-        new Set(
-          res.kernels.map((k) => k.name).filter((name) => !name.endsWith('_op'))
-        )
-      );
-      function nameToConfig(name: string) {
-        return `${name[0].toLowerCase()}${name.slice(1)}Config`;
-      }
-      const importCode = kernelNames.map(
-        (name) =>
-          `import { ${nameToConfig(
-            name
-          )} } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/${name}';`
-      );
-      const configCode = kernelNames.map((name) => `${nameToConfig(name)},`);
-      const code = `
-  ${importCode.join('\n')}
-  const kernelConfigs: KernelConfig[] = [
-  ${configCode.join('\n')}
-  ]
-  `;
-      console.log(code);
-    });
-}
-
-export function memory() {
-  return tfjs.memory();
-}
-
-export function tidy(f: () => void) {
-  return tfjs.tidy(f);
-}

package/src/kernels.ts

@@ -1,31 +0,0 @@
-import {
-  KernelConfig,
-  registerKernel
-} from '@tensorflow/tfjs-core/dist/kernel_registry';
-
-import { mirrorPadConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/MirrorPad';
-import { padV2Config } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/PadV2';
-import { sliceConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/Slice';
-import { fusedConv2DConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/FusedConv2D';
-import { maxPoolConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/MaxPool';
-import { resizeNearestNeighborConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/ResizeNearestNeighbor';
-import { concatConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/Concat';
-import { identityConfig } from '@tensorflow/tfjs-backend-webgpu/dist/kernels/Identity';
-
-const kernelConfigs: KernelConfig[] = [
-  mirrorPadConfig,
-  padV2Config,
-  sliceConfig,
-  fusedConv2DConfig,
-  maxPoolConfig,
-  resizeNearestNeighborConfig,
-  concatConfig,
-  identityConfig
-];
-
-for (const kernelConfig of kernelConfigs) {
-  registerKernel({
-    ...kernelConfig,
-    backendName: 'webgpu-oidn'
-  });
-}