Update tf-metadata.json

lutzroeder · Jul 19, 2024 · 3819ffb · 3819ffb
1 parent e3c12f7
commit 3819ffb
Showing 1 changed file with 66 additions and 0 deletions.
diff --git a/source/tf-metadata.json b/source/tf-metadata.json
@@ -65931,6 +65931,72 @@
       }
     ]
   },
+  {
+    "name": "XlaSparseDenseMatmulGradWithCsrInput",
+    "attributes": [
+      {
+        "name": "N",
+        "type": "int64",
+        "minimum": 1
+      },
+      {
+        "name": "M",
+        "type": "int64",
+        "minimum": 1
+      },
+      {
+        "name": "custom_computation",
+        "type": "function"
+      },
+      {
+        "name": "table_name",
+        "type": "string"
+      }
+    ],
+    "inputs": [
+      {
+        "name": "row_pointers",
+        "type": 3
+      },
+      {
+        "name": "sorted_sample_ids",
+        "type": 3
+      },
+      {
+        "name": "sorted_token_ids",
+        "type": 3
+      },
+      {
+        "name": "sorted_gains",
+        "type": 1
+      },
+      {
+        "name": "activation_gradients",
+        "type": 1
+      },
+      {
+        "name": "tables",
+        "numberAttr": "N",
+        "type": 1
+      },
+      {
+        "name": "hyperparameters",
+        "numberAttr": "M",
+        "type": 1
+      },
+      {
+        "name": "num_minibatches_per_physical_sparse_core",
+        "type": 3
+      }
+    ],
+    "outputs": [
+      {
+        "name": "updated_tables",
+        "numberAttr": "N",
+        "type": 1
+      }
+    ]
+  },
   {
     "name": "XlaSparseDenseMatmulGradWithFtrlAndCsrInput",
     "attributes": [