ソースを参照

Update PyTorch script

Lutz Roeder 2 年 前
コミット
2d25f0690c
1 ファイル変更1 行追加0 行削除
  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