fix fft2D

Signed-off-by: xavier dupré <[email protected]>

Author: sdpython

tests/backend_test_base.py

@@ -92,10 +92,11 @@ def run_onnxruntime(self, model_path, inputs, output_names, use_custom_ops=False
         import onnx
         with open(model_path, "rb") as f:
             onx = onnx.load(f)
-        print(onx)
-        from mlprodict.onnxrt import OnnxInference
-        oinf = OnnxInference(model_path)
-        oinf.run(inputs, verbose=1, fLOG=print)
+        # with open("debug_%d.onnx" % id(model_path), "wb") as f:
+        #      f.write(onx.SerializeToString())
+        # from mlprodict.onnxrt import OnnxInference
+        # oinf = OnnxInference(model_path)
+        # oinf.run(inputs, verbose=1, fLOG=print)
 
         m = rt.InferenceSession(model_path, opt, providers=providers)
         results = m.run(output_names, inputs)

tests/test_backend.py

@@ -5592,7 +5592,8 @@ def func(x):
 
 
 if __name__ == '__main__':
-    cl = BackendTests()
-    cl.setUp()
-    cl.test_rfft2d_ops_specific_dimension()
+    # cl = BackendTests()
+    # cl.setUp()
+    # cl.test_rfft2d_ops_specific_dimension()
+    # stop
     unittest_main()