Fix the netInstance.dispose is not a function issue (#209)

webmachinelearning · Mar 27, 2024 · 692b921 · 692b921
1 parent abbdf25
commit 692b921
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/image_classification/resnet50v2_nchw.js b/image_classification/resnet50v2_nchw.js
@@ -163,6 +163,14 @@ export class ResNet50V2Nchw {
     this.graph_ = this.builder_.build({'output': outputOperand});
   }
 
+  // Release the constant tensors of a model
+  dispose() {
+    // dispose() is only available in webnn-polyfill
+    if (this.graph_ !== null && 'dispose' in this.graph_) {
+      this.graph_.dispose();
+    }
+  }
+
   // Release the constant tensors of a model
   async compute(inputBuffer, outputBuffer) {
     const inputs = {'input': inputBuffer};

diff --git a/image_classification/resnet50v2_nhwc.js b/image_classification/resnet50v2_nhwc.js
@@ -191,6 +191,14 @@ export class ResNet50V2Nhwc {
     this.graph_ = this.builder_.build({'output': outputOperand});
   }
 
+  // Release the constant tensors of a model
+  dispose() {
+    // dispose() is only available in webnn-polyfill
+    if (this.graph_ !== null && 'dispose' in this.graph_) {
+      this.graph_.dispose();
+    }
+  }
+
   // Release the constant tensors of a model
   async compute(inputBuffer, outputBuffer) {
     const inputs = {'input': inputBuffer};