Update view.js (#1240)

lutzroeder · Jun 10, 2024 · 65c2c22 · 65c2c22
1 parent 18b7e0a
commit 65c2c22
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/source/view.js b/source/view.js
@@ -3607,6 +3607,15 @@ view.FindSidebar = class extends view.Control {
     }
 };
 
+view.Argument = class {
+
+    constructor(name, value, type) {
+        this.name = name;
+        this.value = value;
+        this.type = type;
+    }
+};
+
 view.Tensor = class {
 
     constructor(tensor) {
@@ -4094,6 +4103,15 @@ view.Tensor = class {
                 return `${indentation}(undefined)`;
         }
     }
+
+    get metrics() {
+        const metrics = Array.from(this._tensor.metrics || []);
+        const keys = new Set(metrics.map((metrics) => metrics.name));
+        if (!keys.has('sparisity')) {
+            // metrics.push(new view.Argument('sparisity', 0, 'float32'));
+        }
+        return metrics;
+    }
 };
 
 view.Quantization = class {