oidn-web 0.3.5 → 0.5.0

package/lib/helper.d.ts

@@ -1,4 +0,0 @@
-import * as tfjs from '@tensorflow/tfjs-core';
-export declare function profileAndLogKernelCode(execute: () => void, disabled?: boolean): void;
-export declare function memory(): tfjs.MemoryInfo;
-export declare function tidy(f: () => void): void;

package/lib/helper.js

@@ -1,33 +0,0 @@
-import * as tfjs from '@tensorflow/tfjs-core';
-export function profileAndLogKernelCode(execute, 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) {
-            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) {
-    return tfjs.tidy(f);
-}
-//# sourceMappingURL=helper.js.map

package/lib/helper.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"helper.js","sourceRoot":"","sources":["../src/helper.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,uBAAuB,CAAC;AAC9C,MAAM,UAAU,uBAAuB,CAAC,OAAmB,EAAE,QAAQ,GAAG,IAAI;IAC1E,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,EAAE,CAAC;QACV,OAAO;IACT,CAAC;IACD,IAAI;SACD,OAAO,CAAC,GAAG,EAAE;QACZ,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;QACZ,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAC5B,IAAI,GAAG,CACL,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CACvE,CACF,CAAC;QACF,SAAS,YAAY,CAAC,IAAY;YAChC,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC1D,CAAC;QACD,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,CAChC,CAAC,IAAI,EAAE,EAAE,CACP,YAAY,YAAY,CACtB,IAAI,CACL,yDAAyD,IAAI,IAAI,CACrE,CAAC;QACF,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG;IACf,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;;IAErB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;;GAEtB,CAAC;QACE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,UAAU,MAAM;IACpB,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,CAAa;IAChC,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACtB,CAAC"}

package/lib/kernels.d.ts

@@ -1 +0,0 @@
-export {};

package/lib/kernels.js

@@ -1,26 +0,0 @@
-import { 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 = [
-    mirrorPadConfig,
-    padV2Config,
-    sliceConfig,
-    fusedConv2DConfig,
-    maxPoolConfig,
-    resizeNearestNeighborConfig,
-    concatConfig,
-    identityConfig
-];
-for (const kernelConfig of kernelConfigs) {
-    registerKernel({
-        ...kernelConfig,
-        backendName: 'webgpu-oidn'
-    });
-}
-//# sourceMappingURL=kernels.js.map

package/lib/kernels.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"kernels.js","sourceRoot":"","sources":["../src/kernels.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,cAAc,EACf,MAAM,4CAA4C,CAAC;AAEpD,OAAO,EAAE,eAAe,EAAE,MAAM,wDAAwD,CAAC;AACzF,OAAO,EAAE,WAAW,EAAE,MAAM,oDAAoD,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,oDAAoD,CAAC;AACjF,OAAO,EAAE,iBAAiB,EAAE,MAAM,0DAA0D,CAAC;AAC7F,OAAO,EAAE,aAAa,EAAE,MAAM,sDAAsD,CAAC;AACrF,OAAO,EAAE,2BAA2B,EAAE,MAAM,oEAAoE,CAAC;AACjH,OAAO,EAAE,YAAY,EAAE,MAAM,qDAAqD,CAAC;AACnF,OAAO,EAAE,cAAc,EAAE,MAAM,uDAAuD,CAAC;AAEvF,MAAM,aAAa,GAAmB;IACpC,eAAe;IACf,WAAW;IACX,WAAW;IACX,iBAAiB;IACjB,aAAa;IACb,2BAA2B;IAC3B,YAAY;IACZ,cAAc;CACf,CAAC;AAEF,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;IACzC,cAAc,CAAC;QACb,GAAG,YAAY;QACf,WAAW,EAAE,aAAa;KAC3B,CAAC,CAAC;AACL,CAAC"}

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'
-  });
-}