onnx_diagnostic.torch_export_patches.patches.patch_torch¶

onnx_diagnostic.torch_export_patches.patches.patch_torch.patch__check_input_constraints_for_graph(previous_function: Callable, input_placeholders: list[Node], flat_args_with_path, range_constraints, verbose: int = 0) → None[source]¶

class onnx_diagnostic.torch_export_patches.patches.patch_torch.patched_ShapeEnv[source]¶

onnx_diagnostic.torch_export_patches.patches.patch_torch.patched__broadcast_shapes(*_shapes)[source]¶: Patches torch._refs._broadcast_shapes.

onnx_diagnostic.torch_export_patches.patches.patch_torch.patched_infer_size(a, b)[source]¶: Patches torch._subclasses.fake_impls.infer_size.

onnx_diagnostic.torch_export_patches.patches.patch_torch.retrieve_stacktrace()[source]¶: Retrieves and prints the current stack trace, avoids every torch file.