Browse Source

Update PyTorch script

Lutz Roeder 2 years ago
parent
commit
2d25f0690c
1 changed files with 1 additions and 0 deletions
  1. 1 0
      tools/pytorch_metadata.py

+ 1 - 0
tools/pytorch_metadata.py

@@ -145,6 +145,7 @@ 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