Merge pull request #2 from jlewis-austin/master

Global update dimensions to shape in operand descriptors
sushanthr · Oct 31, 2024 · e58a5b6 · e58a5b6
2 parents cc87d98 + a187132
commit e58a5b6
Show file tree

Hide file tree

Showing 3 changed files with 705 additions and 705 deletions.
diff --git a/SWOps.js b/SWOps.js
@@ -6,7 +6,7 @@ function InstallCpuOps(builder) {
             // float case
             let array_buffer_view = new Float32Array(new ArrayBuffer(4));
             array_buffer_view[0] = val;
-            return builder.constant({type: 'float32', dataType: 'float32', dimensions: [1]}, array_buffer_view);
+            return builder.constant({type: 'float32', dataType: 'float32', shape: [1]}, array_buffer_view);
        }
        throw("GenerateMLOperandFromNumber non float is not yet implemented.");
     }
@@ -79,4 +79,4 @@ function InstallCpuOps(builder) {
     builder.gather = gather(builder.gather);
     builder.unsqueeze = unsqueeze(builder.unsqueeze);
     builder.concat = concat(builder.concat);
-}
+}
diff --git a/index.htm b/index.htm
@@ -89,7 +89,7 @@
                 InstallCpuOps(builder);
 
                 const operandType = {type: 'float32',
-                    dataType: 'float32', dimensions: [1, 3, tensor.height, tensor.width]};
+                    dataType: 'float32', shape: [1, 3, tensor.height, tensor.width]};
                 const input_operand = builder.input('input', operandType);
                 const output_operand = loadModelGraph(input_operand, weights_buffer, builder);
                 const graph = await builder.build({'output': output_operand});