Update python.js

lutzroeder · Dec 12, 2022 · 1053dcf · 1053dcf
1 parent a740115
commit 1053dcf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/source/python.js b/source/python.js
@@ -4270,6 +4270,17 @@ python.Execution = class {
             params.groups = groups;
             return params;
         });
+        this.registerFunction('ops.quantized.conv_transpose1d_prepack', function(weight, bias, stride, padding, output_padding, dilation, groups) {
+            const params = self.invoke('__torch__.torch.classes.quantized.Conv2dPackedParamsBase', []);
+            params.weight = weight;
+            params.bias = bias;
+            params.stride = stride;
+            params.padding =padding;
+            params.output_padding = output_padding;
+            params.dilation = dilation;
+            params.groups = groups;
+            return params;
+        });
         this.registerFunction('ops.quantized.conv_transpose2d_prepack', function(weight, bias, stride, padding, output_padding, dilation, groups) {
             const params = self.invoke('__torch__.torch.classes.quantized.Conv2dPackedParamsBase', []);
             params.weight = weight;