Update PyTorch script

lutzroeder · Feb 2, 2024 · 3d0e2fc · 3d0e2fc
1 parent a7f8575
commit 3d0e2fc
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tools/pytorch_metadata.py b/tools/pytorch_metadata.py
@@ -145,7 +145,6 @@ def _check_types(types, schemas):
     types.pop('aten::fft')
     types.pop('aten::mul.ScalarT')
     types.pop('aten::classes._nnapi.Compilation')
-    types.pop('aten::quantized_lstm_cell')
     if len(types) > 0:
         raise Exception('\n'.join(list(types.keys()))) # pylint: disable=broad-exception-raised