diff --git a/source/tf-metadata.json b/source/tf-metadata.json index 477d2bb83f..8eba6c64b9 100644 --- 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": [