onnx_diagnostic.torch_export_patches.patch_inputs¶
- onnx_diagnostic.torch_export_patches.patch_inputs.convert_dynamic_axes_into_dynamic_shapes(model: Module, args: Tuple[Any, ...] | None = None, kwargs: Dict[str, Any] | None = None, dynamic_axes: Dict[str, Dict[int, str]] | None = None, prefix_mapping: Dict[str, str] | None = None, verbose: int = 0) Tuple[Tuple[Any, ...], Dict[str, Any], Dict[str, Any]] [source]¶
Converts the input from an export to something
torch.export.export()
can handle.- Parameters:
model – model to convert (used to extract the signature)
args – positional arguments
kwargs – named arguments
dynamic_axes – dynamic axes
prefix_mapping – prefix mapping
verbose – verbosity
- Returns:
(args, kwargs, dynamic shapes)