g

Author: xadupre

onnx_diagnostic/export/shape_helper.py

@@ -204,10 +204,10 @@ def guess_dynamic_shapes_from_inputs(
 
 
 def make_fake_with_dynamic_dimensions(
-    inputs: Optional[Any],
+    inputs: Any,
     dynamic_shapes: Any,
     fake_mode: Optional["FakeTensorMode"] = None,  # noqa: F821
-) -> Optional[Tuple["FakeTensor", "FakeTensorMode"]]:  # noqa: F821
+) -> Any:  # noqa: F821
     """
     Replaces all tensors by fake tensor respecting the same
     constraints as the following dynamic shapes.

onnx_diagnostic/helpers/fake_tensor_helper.py

@@ -64,17 +64,15 @@ def fake_reshape(
     for i in sh:
         new_shape[i] = fake_tensor.shape[i]
 
-    #
-
     reduced_tensor = fake_mode.from_tensor(true_tensor, static_shapes=True).sum(
         axis=tuple(sorted(sh)), keepdim=True
     )
     return reduced_tensor.expand(*new_shape)
 
 
 def make_fake(
-    x: Optional[Any], fake_mode: Optional["FakeTensorMode"] = None  # noqa: F821
-) -> Optional[Tuple["FakeTensor", "FaleTensorMode"]]:  # noqa: F821
+    x: Any, fake_mode: Optional["FakeTensorMode"] = None  # noqa: F821
+) -> Optional[Tuple["FakeTensor", "FakeTensorMode"]]:  # noqa: F821
     """
     Replaces all tensors by fake tensors.
     This modification happens inplace for caches.