Fix MLTensorUsage is undefined issue

Polyfill MLTensorUsage to make it compatible with old version of Chrome
webmachinelearning · Nov 14, 2024 · 06e90e4 · 06e90e4
1 parent 77a98b9
commit 06e90e4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/common/utils.js b/common/utils.js
@@ -306,8 +306,11 @@ export function getUrlParams() {
 
 export async function isWebNN() {
   if (typeof MLGraphBuilder !== 'undefined') {
-    const context = await navigator.ml.createContext();
-    return !context.tf;
+    if (typeof MLTensorUsage == 'undefined') {
+      // Polyfill MLTensorUsage to make it compatible with old version of Chrome.
+      window.MLTensorUsage = {WEBGPU_INTEROP: 1, READ: 2, WRITE: 4};
+    }
+    return true;
   } else {
     return false;
   }

diff --git a/nnotepad/js/nnotepad.js b/nnotepad/js/nnotepad.js
@@ -52,7 +52,8 @@ class WebNNUtil {
       dataType: isShapeMethod ? operand.dataType() : operand.dataType,
       dimensions: isShapeMethod ? operand.shape() : operand.shape,
       shape: isShapeMethod ? operand.shape() : operand.shape,
-      usage: MLTensorUsage.READ,
+      usage: typeof MLTensorUsage == 'undefined' ?
+          undefined : MLTensorUsage.READ,
       readable: true,
     };
     const tensor = await context.createTensor(desc);