Browse Source

Update coreml-metadata.json

Lutz Roeder 2 years ago
parent
commit
7b9a6be79b
1 changed files with 9 additions and 0 deletions
  1. 9 0
      source/coreml-metadata.json

+ 9 - 0
source/coreml-metadata.json

@@ -138,6 +138,15 @@
       { "name": "mode", "type": "FlattenLayerParams.FlattenOrder" }
     ]
   },
+  {
+    "name": "gather",
+    "category": "Transform",
+    "description": "Gather layer that gathers elements from the first input, along a specified axis, at indices specified in the second input.",
+    "inputs": [
+      { "name": "input", "type": "Tensor" },
+      { "name": "indices", "type": "Tensor" }
+    ]
+  },
   {
     "name": "gelu",
     "category": "Activation",