Sfoglia il codice sorgente

Update executorch.js

Lutz Roeder 9 mesi fa
parent
commit
af677d7975
2 ha cambiato i file con 11 aggiunte e 0 eliminazioni
  1. 10 0
      source/executorch-schema.js
  2. 1 0
      source/executorch.js

+ 10 - 0
source/executorch-schema.js

@@ -596,6 +596,7 @@ fb_xnnpack.XNodeUnion = class {
             case 39: return fb_xnnpack.XNNConvTranspose2d.decode(reader, position);
             case 40: return fb_xnnpack.XNNReciprocalSquareRoot.decode(reader, position);
             case 41: return fb_xnnpack.XNNLog.decode(reader, position);
+            case 42: return fb_xnnpack.XNNGelu.decode(reader, position);
             default: return undefined;
         }
     }
@@ -1182,6 +1183,15 @@ fb_xnnpack.XNNLog = class XNNLog {
     }
 };
 
+fb_xnnpack.XNNGelu = class XNNGelu {
+
+    static decode(reader, position) {
+        const $ = new fb_xnnpack.XNNGelu();
+        fb_xnnpack._XNNNode1x1.decode(reader, position, $);
+        return $;
+    }
+};
+
 vkgraph.OperatorCall = class OperatorCall {
 
     static decode(reader, position) {

+ 1 - 0
source/executorch.js

@@ -564,6 +564,7 @@ xnnpack.Metadata = class {
         this.register('XNNDepthwiseConv2d', 'Layer');
         this.register('XNNELU', 'Activation');
         this.register('XNNFullyConnected', 'Layer');
+        this.register('XNNGelu', 'Activation');
         this.register('XNNGlobalAvgPooling2d', 'Pool');
         this.register('XNNGlobalAvgPooling2d', 'Pool');
         this.register('XNNHardswish', 'Activation');