Prechádzať zdrojové kódy

Update caffe2-metadata.json

Lutz Roeder 6 rokov pred
rodič
commit
8b988c1530
1 zmenil súbory, kde vykonal 39 pridanie a 0 odobranie
  1. 39 0
      src/caffe2-metadata.json

+ 39 - 0
src/caffe2-metadata.json

@@ -18315,5 +18315,44 @@
       ],
       "support_level": "default"
     }
+  },
+  {
+    "name": "SparseLengthsSumSparseLookup",
+    "schema": {
+      "description": "\nThis op converts compressed indices of SparseLengthsSum*Sparse to\nuncompressed indices of SparseLengthsSum*. For compressed indices that maps\nto -1. It means it will correspond to a zero row in the uncompressed data.\nTherefore we will remove this indices and adjust the lengths.\n",
+      "inputs": [
+        {
+          "description": "Integer vector containing compressed indices of the first dimension of DATA for the slices that are being aggregated",
+          "name": "INDICES"
+        },
+        {
+          "description": "Vector with the same sum of elements as the first dimension of INDICES",
+          "name": "LENGTHS"
+        },
+        {
+          "description": "Integer vector mapping uncompressed indices to compressed indices",
+          "name": "COMPRESSED_INDICES_MAPPING"
+        },
+        {
+          "description": "Vector of weights to scale rows of DATA with before reduction. Same size as INDICES.",
+          "name": "WEIGHTS"
+        }
+      ],
+      "outputs": [
+        {
+          "description": "Uncompressed indices",
+          "name": "output_indices"
+        },
+        {
+          "description": "Adjusted lengths",
+          "name": "output_lengths"
+        },
+        {
+          "description": "Adjusted weights",
+          "name": "output_weights"
+        }
+      ],
+      "support_level": "default"
+    }
   }
 ]