.xshape.rename_expressions

class experimental_experiment.xshape.rename_expressions.RenameTransformer(mapping, expr: str | None = None)

class experimental_experiment.xshape.rename_expressions.RenameVariable(mapping, expr: str | None = None)

experimental_experiment.xshape.rename_expressions.parse_expression_tokens(expr: str) → Set[str]

Extracts the token from an expression.

experimental_experiment.xshape.rename_expressions.rename_dynamic_dimensions(constraints: Dict[str, Set[str]], original: Set[str], ban_prefix: str = 'DYN') → Dict[str, str]

Renames dynamic shapes as requested by the user. torch.export.export() uses many names for dynamic dimensions. When building the onnx model, some of them are redundant and can be replaced by the name provided by the user.

Parameters:

• constraints – exhaustive list of used name and all the values equal to it

• original – the names to use if possible

• ban_prefix – avoid any rewriting by a constant starting with this prefix

Returns:

replacement dictionary

experimental_experiment.xshape.rename_expressions.rename_dynamic_expression(expression: str, replacements: Dict[str, str])

Renames variables inside an expression. The function removes any space.

Parameters:

• expression – something like s15 + seq_length

• replacements – replacements to make

Returns:

new string

experimental_experiment.xshape.rename_expressions.rename_expression(expr: str, mapping: Dict[str, str]) → str

Renames variables in a Python expression using AST.

Parameters:

• expr – Python expression as string

• mapping – Mapping from old names to new names

Returns:

rransformed expression