.torch_interpreter.patches.patch_torch

experimental_experiment.torch_interpreter.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]
experimental_experiment.torch_interpreter.patches.patch_torch.patched__broadcast_shapes(*_shapes)[source]

Patches torch._refs._broadcast_shapes.

experimental_experiment.torch_interpreter.patches.patch_torch.patched_infer_size(a, b)[source]

Patches torch._subclasses.fake_impls.infer_size.