.torch_dynamo

modules

• .torch_dynamo.backend_helper
• .torch_dynamo.compiled_model
• .torch_dynamo.debug_backend
• .torch_dynamo.fast_backend
• experimental_experiment.torch_dynamo.partition

onnx_custom_backend

experimental_experiment.torch_dynamo.onnx_custom_backend(graph_module: torch.fx.GraphModule, args: List[torch.Tensor], target_opset: int | None = None, backend: str = 'ort', verbose: int | Tuple[int, int] = 0, dump_prefix: str | None = None, dump_patterns: str | None = None, providers: Tuple[str] | None = None, raise_exc: bool = True, storage: Dict[str, Any] | None = None, enable_pattern: str | List[str | type] | None = 'default', disable_pattern: str | List[str | type] | None = None, pre_ort_model_transforms: Callable[[ModelProto], ModelProto] | List[Callable[[ModelProto], ModelProto]] | None = None, ort_optimization_level: str | None = None, dispatcher: Dispatcher | None = None, rename_inputs: bool = True, optimize: bool = True, exporter: str | None = None, processor: str = 'CPU', order_algorithm: str | None = None, options: OptimizationOptions | None = None, export_options: str | ExportOptions | None = None) → Callable

Custom backend to export torch models into onnx (see torch.compiler). This backend relies on onnxruntime and tries to be as efficient as possible.

Parameters:

• graph_module – graph to export

• args – arguments

• target_opset – opset to use for the conversion

• backend – only ‘ort’ is allowed

• verbose – adjust verbosity, if tuple, if gives different verbosity level to the exporter and the runtime

• dump_prefix – to dump the models and the inputs

• dump_patterns – dump the patterns as well

• providers – where to run the model, by default

• raise_exc – raise an exception whenever something goes wrong

• storage – to store any interesting objects during the process

• enable_pattern – optimization patterns to enable

• disable_pattern – optimization patterns to disable

• pre_ort_model_transforms – list of transformations applied on the final ModelProto

• ort_optimization_level – graph optimization level for onnxruntime, the default value is the same as what onnxruntime defines

• dispatcher – see experimental_experiment.torch_interpreter.Dispatcher

• rename_inputs – rename the inputs

• optimize – enable or disable the optimization

• exporter – use a different exporter

• processor – optimization should be made for this processor or this list of processors (comma separated value)

• order_algorithm – algorithm optimizing the order the onnx node, none by default

• options – to define custom Optimization options, in that case, any other optimization parameter is ignored

• export_options – see ExportOptions

Returns:

Callable

See 101: A custom backend for torch for examples. If not empty, storage keeps the memory of the data generated, onnx models, graph module as well the inputs and outputs when the model is run.

The following example shows how to use the custom backend (based on onnxruntime).

<<<

import torch
from experimental_experiment.torch_dynamo import onnx_custom_backend


class MLP(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.layers = torch.nn.Sequential(
            torch.nn.Linear(10, 32),
            torch.nn.Sigmoid(),
            torch.nn.Linear(32, 1),
        )

    def forward(self, x):
        return self.layers(x)


x = torch.randn(3, 10, dtype=torch.float32)

mlp = MLP()
expected = mlp(x)

compiled_model = torch.compile(
    mlp,
    backend=lambda *args, **kwargs: onnx_custom_backend(*args, verbose=1, **kwargs),
    dynamic=False,
    fullgraph=True,
)

try:
    got = compiled_model(x)
    diff = (expected - got).max()
    print(f"discrepancies: {diff}")
except (ImportError, AttributeError) as e:
    print("onnxruntime-training is not installed", e)

>>>

    [onnx_custom_backend] starts conversion to onnx.
    [to_onnx] build the graph module from <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'>, type(args)=<class 'tuple'>
    [to_onnx] build the graph module with input_names=['input0', 'input1', 'input2', 'input3', 'input4']
    [_make_builder_interpreter] use existing <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'>
    [to_onnx] graph module done in 0.0006762920020264573 s
    [to_onnx] start creating the onnx nodes
    [to_onnx] interpreter.function_options=FunctionOptions(export_as_function=True, name='*', domain='*', external_threshold=256, move_initializer_to_constant=True, return_initializer=True, merge_allowed=True, rename_allowed=True)
    [to_onnx] 13 onnx nodes done in 0.0022472019991255365 s
    [to_onnx] start conversion to onnx (before optimization) mask_outputs=None
    [GraphBuilder-TJS._add_shape_information] dynamic shapes replacements={}
    [GraphBuilder-TJS.optimize] start with 13 nodes
    [GraphBuilder-TJS.optimize] #patterns=92
    [GraphBuilder-TJS.optimize] start with subgraphs
    [GraphBuilder-TJS.optimize] done with subgraphs
    [GraphBuilderPatternOptimization-TJS.optimize] start with 7 nodes, 0 initializers, 92 patterns, priorities=[0, 1, 2, 3], max_iter=40
    [GraphBuilderPatternOptimization-TJS.optimize] same children={'SameChildrenFromInputPattern', 'SameChildrenPattern'}
    [GraphBuilderPatternOptimization-TJS.optimize] iteration 0: 7 nodes, priority=0
    [GraphBuilderPatternOptimization-TJS.optimize] increase priority to 1
    [GraphBuilderPatternOptimization-TJS.optimize] iteration 1: 7 nodes, priority=1
    [GraphBuilderPatternOptimization-TJS.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=IdentityPattern:0.000
    [GraphBuilderPatternOptimization-TJS.optimize] iteration 2: 6 nodes, priority=1
    [GraphBuilderPatternOptimization-TJS.optimize] increase priority to 2
    [GraphBuilderPatternOptimization-TJS.optimize] iteration 3: 6 nodes, priority=2
    [GraphBuilderPatternOptimization-TJS.optimize] increase priority to 3
    [GraphBuilderPatternOptimization-TJS.optimize] iteration 4: 6 nodes, priority=3
    [GraphBuilderPatternOptimization-TJS.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=FunctionHalfRotaryEmbeddingPattern:0.000
    [GraphBuilderPatternOptimization-TJS.optimize] iteration 5: 4 nodes, priority=3
    [GraphBuilderPatternOptimization-TJS.optimize] stops current_priority_index=4, priorities=[0, 1, 2, 3]
    [GraphBuilderPatternOptimization-TJS.optimize] done after 6 iterations with 4 nodes in 0.017
    [GraphBuilder-TJS.optimize] done with 4 nodes in 0.019
    [GraphBuilder-TJS.to_onnx] make_model 1 inits 0 params
    [GraphBuilder-TJS.time_evaluation_constants_] 0
    [GraphBuilder-TJS._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024
    [GraphBuilder-TJS._build_initializers] switch low/high order
    [GraphBuilder-TJS._build_initializers] done in 1.2559976312331855e-06s with 1 initializers, 0 large initializers
    [GraphBuilder-TJS._add_shape_information] dynamic shapes replacements={}
    [to_onnx] to_onnx done in 0.021127220999915153s and 4 nodes, 1 initializers, 5 inputs, 1 outputs
    [onnx_custom_backend] to_onnx done in 0.025163596001220867 with 4 nodes and 0 local functions.
    [onnx_custom_backend] starts creating InferenceSession
    [onnx_custom_backend] InferenceSession done in 0.006847856995591428
    discrepancies: 2.9802322387695312e-08

onnx_debug_backend

experimental_experiment.torch_dynamo.onnx_debug_backend(graph_module: torch.fx.GraphModule, args: List[torch.Tensor | torch.SymInt | torch.SymFloat], target_opset: int | None = None, backend: str | Callable[[ModelProto, bool | None], Any] = 'ort', verbose: int | Tuple[int, int] = 0, dump_prefix: str | None = None, dump_patterns: str | None = None, providers: Tuple[str] | None = None, raise_exc: bool = True, storage: Dict[str, Any] | None = None, raise_list: Set[str] | None = None, enable_pattern: str | List[str | type] | None = 'default', disable_pattern: str | List[str | type] | None = None, pre_ort_model_transforms: Callable[[ModelProto], ModelProto] | List[Callable[[ModelProto], ModelProto]] | None = None, ort_optimization_level: str | None = None, dispatcher: Dispatcher | None = None, rename_inputs: bool = True, optimize: bool = True, processor: str = 'CPU', order_algorithm: str | None = None) → Callable

Custom backend to export torch models into onnx (see torch.compiler). This backend is not meant to be efficient, it is more to check the conversion is ok. It relies either on onnxruntime or the python reference implementation.

Parameters:

• graph_module – graph to export

• args – arguments

• target_opset – opset to use for the conversion

• backend – after the conversion, the model is executed with a runtime, onnxruntime or the reference implementation, it must be a value among ‘ort’, ‘ref’ or a class, it can be a function as well which returns an object behaving the same way

• verbose – adjust verbosity, if tuple, if gives different verbosity level to the exporter and the runtime

• dump_prefix – prefix used to dump the model generated by the backend

• dump_patterns – dump the patterns as well

• providers – where to run the model, by default

• raise_exc – raise an exception whenever something goes wrong

• storage – to store any interesting objects during the process

• raise_list – the builder stops any time a name falls into that list, this is a debbuging tool

• enable_pattern – optimization patterns to enable

• disable_pattern – optimization patterns to disable

• pre_ort_model_transforms – list of transformations applied on the final ModelProto

• ort_optimization_level – graph optimization level for onnxruntime, the default value is the same as what onnxruntime defines

• dispatcher – see experimental_experiment.torch_interpreter.Dispatcher

• rename_inputs – rename inputs into input_{i}

• optimize – enable or disable the optimization

• processor – specifies the processor it is optimized for

• order_algorithm – algorithm optimizing the order the onnx node, none by default

Returns:

Callable

If not empty, storage keeps the memory of the data generated, onnx models, graph module as well the inputs and outputs when the model is run.

The following example shows how to use the reference implementation (experimental_experiment.reference.ExtendedReferenceEvaluator) to run the onnx model and display the intermediate results.

<<<

import torch
from experimental_experiment.torch_dynamo import onnx_debug_backend


class MLP(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.layers = torch.nn.Sequential(
            torch.nn.Linear(10, 32),
            torch.nn.Sigmoid(),
            torch.nn.Linear(32, 1),
        )

    def forward(self, x):
        return self.layers(x)


x = torch.randn(3, 10, dtype=torch.float32)

mlp = MLP()
expected = mlp(x)

compiled_model = torch.compile(
    mlp,
    backend=lambda *args, **kwargs: onnx_debug_backend(
        *args, verbose=(1, 10), backend="ref", **kwargs
    ),
    dynamic=False,
    fullgraph=True,
)

got = compiled_model(x)
diff = (expected - got).max()
print(f"discrepancies: {diff}")

>>>

    [to_onnx] build the graph module from <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'>, type(args)=<class 'tuple'>
    [to_onnx] build the graph module with input_names=['input0', 'input1', 'input2', 'input3', 'input4']
    [_make_builder_interpreter] use existing <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'>
    [to_onnx] graph module done in 0.0006333269993774593 s
    [to_onnx] start creating the onnx nodes
    [to_onnx] interpreter.function_options=FunctionOptions(export_as_function=True, name='*', domain='*', external_threshold=256, move_initializer_to_constant=True, return_initializer=True, merge_allowed=True, rename_allowed=True)
    [to_onnx] 13 onnx nodes done in 0.002190330997109413 s
    [to_onnx] start conversion to onnx (before optimization) mask_outputs=None
    [GraphBuilder-TTW._add_shape_information] dynamic shapes replacements={}
    [GraphBuilder-TTW.optimize] start with 13 nodes
    [GraphBuilder-TTW.optimize] #patterns=92
    [GraphBuilder-TTW.optimize] start with subgraphs
    [GraphBuilder-TTW.optimize] done with subgraphs
    [GraphBuilderPatternOptimization-TTW.optimize] start with 7 nodes, 0 initializers, 92 patterns, priorities=[0, 1, 2, 3], max_iter=40
    [GraphBuilderPatternOptimization-TTW.optimize] same children={'SameChildrenFromInputPattern', 'SameChildrenPattern'}
    [GraphBuilderPatternOptimization-TTW.optimize] iteration 0: 7 nodes, priority=0
    [GraphBuilderPatternOptimization-TTW.optimize] increase priority to 1
    [GraphBuilderPatternOptimization-TTW.optimize] iteration 1: 7 nodes, priority=1
    [GraphBuilderPatternOptimization-TTW.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=IdentityPattern:0.000
    [GraphBuilderPatternOptimization-TTW.optimize] iteration 2: 6 nodes, priority=1
    [GraphBuilderPatternOptimization-TTW.optimize] increase priority to 2
    [GraphBuilderPatternOptimization-TTW.optimize] iteration 3: 6 nodes, priority=2
    [GraphBuilderPatternOptimization-TTW.optimize] increase priority to 3
    [GraphBuilderPatternOptimization-TTW.optimize] iteration 4: 6 nodes, priority=3
    [GraphBuilderPatternOptimization-TTW.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000
    [GraphBuilderPatternOptimization-TTW.optimize] iteration 5: 4 nodes, priority=3
    [GraphBuilderPatternOptimization-TTW.optimize] stops current_priority_index=4, priorities=[0, 1, 2, 3]
    [GraphBuilderPatternOptimization-TTW.optimize] done after 6 iterations with 4 nodes in 0.015
    [GraphBuilder-TTW.optimize] done with 4 nodes in 0.016
    [GraphBuilder-TTW.to_onnx] make_model 1 inits 0 params
    [GraphBuilder-TTW.time_evaluation_constants_] 0
    [GraphBuilder-TTW._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024
    [GraphBuilder-TTW._build_initializers] switch low/high order
    [GraphBuilder-TTW._build_initializers] done in 1.1779993656091392e-06s with 1 initializers, 0 large initializers
    [GraphBuilder-TTW._add_shape_information] dynamic shapes replacements={}
    [to_onnx] to_onnx done in 0.017948756998521276s and 4 nodes, 1 initializers, 5 inputs, 1 outputs
     +C init7_s2_32_1: int64:(2,):[32, 1]
     +I input0: float32:(32, 10):-0.3115033507347107,0.15047042071819305,0.31346243619918823,-0.16454416513442993,-0.2865389585494995...
     +I input1: float32:(32,):-0.2084808051586151,-0.23735138773918152,0.270869642496109,0.027419045567512512,-0.30015596747398376...
     +I input2: float32:(3, 10):-1.057297706604004,0.43689560890197754,0.41113561391830444,0.5262078642845154,-2.254220962524414...
     +I input3: float32:(1, 32):-0.12932340800762177,-0.06593326479196548,-0.07751987874507904,-0.08029305189847946,-0.10487064719200134...
     +I input4: float32:(1,):[-0.12014711648225784]
    Gemm(input2, input0, input1) -> input_1
     + input_1: float32:(3, 32):0.9411873817443848,0.5806750059127808,0.09388521313667297,0.01602502539753914,-0.6606051325798035...
    Sigmoid(input_1) -> input_2
     + input_2: float32:(3, 32):0.7193394303321838,0.6412227153778076,0.5234540700912476,0.5040061473846436,0.34060370922088623...
    Reshape(input3, init7_s2_32_1) -> l_self_modules_layers_modules_2_parameters_weight_::T10
     + l_self_modules_layers_modules_2_parameters_weight_::T10: float32:(32, 1):-0.12932340800762177,-0.06593326479196548,-0.07751987874507904,-0.08029305189847946,-0.10487064719200134...
    Gemm(input_2, l_self_modules_layers_modules_2_parameters_weight_::T10, input4) -> output_0
     + output_0: float32:(3, 1):[-0.4515935778617859, -0.4168669283390045, -0.3554365336894989]
    discrepancies: 5.960464477539063e-08

dynger_backend

experimental_experiment.torch_dynamo.dynger_backend(graph_module: GraphModule, args: List[Tensor | SymInt | SymFloat], dynamic_shapes: Dict[str, Any] | Tuple[Any] | None = None, optimize: bool = True, verbose: int | Tuple[int, int] = 0) → Callable

Eager backend for dynamo.

Parameters:

• graph_module – graph to export

• args – arguments

• optimize – optimize or not, those optimization would be done on the graph module itself

• verbose – adjust verbosity, if tuple, if gives different verbosity level to the exporter and the runtime

Returns:

Callable

Next examples shows how to display intermediate results while executing the graph produced by torch dynamo.

<<<

import torch
from experimental_experiment.torch_dynamo import dynger_backend


class MLP(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.layers = torch.nn.Sequential(
            torch.nn.Linear(10, 32),
            torch.nn.Sigmoid(),
            torch.nn.Linear(32, 1),
        )

    def forward(self, x):
        return self.layers(x)


x = torch.randn(3, 10, dtype=torch.float32)

mlp = MLP()
expected = mlp(x)

compiled_model = torch.compile(
    mlp,
    backend=lambda *args, **kwargs: dynger_backend(*args, verbose=10, **kwargs),
    dynamic=False,
    fullgraph=True,
)

got = compiled_model(x)
diff = (expected - got).max()
print(f"discrepancies: {diff}")

>>>

    [dynger_backend] use existing <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'>
    [dynger_backend] begin execution with 9 nodes
    <built-in function linear>((l_x_, l_self_modules_layers_modules_0_parameters_weight_, l_self_modules_layers_modules_0_parameters_bias_)) -> input_1
      + input_1: torch.float32:torch.Size([3, 32]):0.6197245121002197,0.23499664664268494,-0.2097005993127823,0.3765624165534973,0.765796422958374...
    <built-in method sigmoid of type object at 0x77b3ef73e160>((input_1,)) -> input_2
      + input_2: torch.float32:torch.Size([3, 32]):0.6501559019088745,0.5584803223609924,0.44776612520217896,0.5930436849594116,0.6826108694076538...
    <built-in function linear>((input_2, l_self_modules_layers_modules_2_parameters_weight_, l_self_modules_layers_modules_2_parameters_bias_)) -> input_3
      + input_3: torch.float32:torch.Size([3, 1]):-0.6722526550292969,-0.645380437374115,-0.5915858149528503
    [dynger_backend] done
    discrepancies: 0.0

Other functions

experimental_experiment.torch_dynamo.filter_decomposition_table(existing_table: Dict | None = None, filter_fct: Callable[[Any], bool] | None = None) → Dict

Returns the decomposition table when some conversions because their translation in ONNX is less efficient.

Parameters:

• existing_table – dictionary of decompositions, by default, it is torch._decomp.decomposition_table.

• filter_fct – if specified, a decomposition function is remove if the function returns false

Returns:

new table

import torch
from torch._dynamo.backends.common import aot_autograd
from experimental_experiment.torch_dynamo import filter_decomposition_table

aot_compiler = aot_autograd(
    fw_compiler=backend_debug,
    decompositions=filter_decomposition_table()
)

compiled_model = torch.compile(
    model,
    backend=aot_compiler,
    dynamic=dynamic,
    fullgraph=fullgraph,
)

The value is:

<<<

import pprint
from experimental_experiment.torch_dynamo import filter_decomposition_table

pprint.pprint(filter_decomposition_table())

>>>

    {<torch._higher_order_ops.out_dtype.OutDtypeOperator object at 0x77b348c1e570>: <function out_dtype_decomp at 0x77b3487abd80>,
     <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x77b3484187c0>,
     <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x77b3484187c0>,
     <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x77b348419bc0>,
     <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x77b348419e40>,
     <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x77b348419e40>,
     <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x77b34841a520>,
     <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x77b34841a520>,
     <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x77b34841a980>,
     <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x77b34841afc0>,
     <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x77b34841afc0>,
     <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x77b34841ad40>,
     <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x77b34841ad40>,
     <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x77b34841b2e0>,
     <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x77b34841b2e0>,
     <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x77b348448400>,
     <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x77b34841b740>,
     <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x77b348448720>,
     <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x77b34841bce0>,
     <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x77b348448400>,
     <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x77b348448720>,
     <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x77b348448ea0>,
     <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x77b3484491c0>,
     <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x77b348449760>,
     <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x77b348449760>,
     <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x77b348449b20>,
     <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x77b348449b20>,
     <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x77b34841b240>,
     <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x77b34841b240>,
     <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x77b348449da0>,
     <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x77b348449da0>,
     <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x77b34844a660>,
     <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x77b34844a660>,
     <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x77b34844bec0>,
     <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x77b34844a8e0>,
     <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x77b34844ab60>,
     <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x77b34844b1a0>,
     <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x77b34844b1a0>,
     <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x77b34844b420>,
     <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x77b34844b7e0>,
     <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x77b34844ba60>,
     <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x77b34844ba60>,
     <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x77b34844bce0>,
     <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x77b34844bce0>,
     <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x77b34844a7a0>,
     <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x77b34844bb00>,
     <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x77b34844bd80>,
     <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x77b34844bf60>,
     <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x77b348468220>,
     <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x77b348468220>,
     <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x77b348468680>,
     <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x77b348468680>,
     <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x77b348468a40>,
     <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x77b348468a40>,
     <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x77b348468e00>,
     <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x77b348468e00>,
     <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x77b34846a520>,
     <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x77b34846a520>,
     <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x77b3484693a0>,
     <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x77b3484693a0>,
     <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x77b3484698a0>,
     <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x77b3484698a0>,
     <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x77b348469da0>,
     <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x77b348469da0>,
     <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x77b34846a160>,
     <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x77b34846a160>,
     <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x77b34846a660>,
     <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x77b34846a660>,
     <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x77b34846aa20>,
     <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x77b34846aa20>,
     <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x77b34846ade0>,
     <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x77b34846ade0>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x77b34846b420>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x77b34846b420>,
     <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x77b348524ea0>,
     <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x77b348524ea0>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x77b348702480>,
     <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.t', overload='default')>: <function t at 0x77b348526520>,
     <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x77b34854c7c0>,
     <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x77b348525da0>,
     <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x77b348524220>,
     <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x77b348526160>,
     <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x77b348526160>,
     <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x77b3485263e0>,
     <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x77b3485263e0>,
     <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x77b3485260c0>,
     <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x77b348526700>,
     <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x77b3485268e0>,
     <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x77b348526d40>,
     <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x77b348526d40>,
     <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x77b348526fc0>,
     <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x77b34854c7c0>,
     <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x77b34854dc60>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x77b34854d260>,
     <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x77b34854d260>,
     <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x77b34854de40>,
     <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x77b34854de40>,
     <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x77b34854e7a0>,
     <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x77b34854e7a0>,
     <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x77b34854f2e0>,
     <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x77b34854f2e0>,
     <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x77b34854eca0>,
     <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x77b34854eca0>,
     <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x77b34854ef20>,
     <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x77b34854ef20>,
     <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x77b34854e700>,
     <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x77b34854e700>,
     <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x77b34854f560>,
     <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x77b34854f560>,
     <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x77b34854f920>,
     <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x77b34854f920>,
     <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x77b34854fce0>,
     <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x77b34854fce0>,
     <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x77b34857c0e0>,
     <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x77b34857c0e0>,
     <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x77b34857c180>,
     <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x77b34857cae0>,
     <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x77b34857cae0>,
     <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x77b34857cfe0>,
     <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x77b34857cfe0>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x77b34857cb80>,
     <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x77b34857cb80>,
     <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x77b34857d580>,
     <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x77b34857d580>,
     <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x77b34857d800>,
     <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x77b34857ef20>,
     <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x77b34854f880>,
     <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x77b34857e5c0>,
     <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x77b34857d620>,
     <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x77b34857d8a0>,
     <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x77b34857e840>,
     <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x77b34857d9e0>,
     <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x77b34857d9e0>,
     <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x77b34857db20>,
     <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x77b34857df80>,
     <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x77b34857dc60>,
     <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x77b34857eac0>,
     <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x77b34857eac0>,
     <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x77b34857dda0>,
     <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x77b34857dee0>,
     <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x77b34857ed40>,
     <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x77b34857e020>,
     <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x77b34857e160>,
     <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x77b34857ed40>,
     <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x77b34857e2a0>,
     <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x77b34857e3e0>,
     <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x77b34857e3e0>,
     <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x77b34857efc0>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x77b34857e520>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x77b34857e520>,
     <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x77b3485a9da0>,
     <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x77b34857e660>,
     <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x77b34857e480>,
     <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x77b34857e7a0>,
     <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x77b34857e7a0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x77b34857e8e0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x77b34857e8e0>,
     <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x77b34857f240>,
     <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x77b34857ea20>,
     <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x77b34857ea20>,
     <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x77b34857eb60>,
     <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x77b34857eca0>,
     <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x77b34857eca0>,
     <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x77b34857f1a0>,
     <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x77b34857f060>,
     <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x77b34857ede0>,
     <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x77b34857ede0>,
     <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x77b34857f1a0>,
     <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x77b34857ef20>,
     <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x77b34857e980>,
     <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x77b34857da80>,
     <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x77b34857e0c0>,
     <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x77b34857e0c0>,
     <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x77b34857de40>,
     <OpOverload(op='aten.log_', overload='default')>: <function log at 0x77b34857d080>,
     <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x77b34857dbc0>,
     <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x77b34857d940>,
     <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x77b34857f4c0>,
     <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x77b34857d4e0>,
     <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x77b34857f2e0>,
     <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x77b34857f100>,
     <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x77b34857f420>,
     <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x77b34857f740>,
     <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x77b34857f560>,
     <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x77b34857f560>,
     <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x77b34857f9c0>,
     <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x77b34857f6a0>,
     <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x77b34857f7e0>,
     <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x77b34857f7e0>,
     <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x77b34857f920>,
     <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x77b34857f920>,
     <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x77b34857fc40>,
     <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x77b34857fa60>,
     <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x77b34857fba0>,
     <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x77b34857fce0>,
     <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x77b34857fce0>,
     <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x77b34857fd80>,
     <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x77b34857fe20>,
     <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x77b34857fe20>,
     <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x77b34857e700>,
     <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x77b34857fd80>,
     <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x77b34857ff60>,
     <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x77b3485aade0>,
     <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x77b3485a80e0>,
     <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x77b3485a8220>,
     <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x77b3485a8360>,
     <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x77b34857e200>,
     <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x77b3485a84a0>,
     <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x77b34857e200>,
     <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x77b34857fec0>,
     <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x77b34857fec0>,
     <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x77b3485a85e0>,
     <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x77b34857ee80>,
     <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x77b34857ee80>,
     <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x77b3485a8720>,
     <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x77b3485a8720>,
     <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x77b3485a8900>,
     <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x77b3485a8680>,
     <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x77b3485a8400>,
     <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x77b3485a8400>,
     <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x77b3485a9800>,
     <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x77b3485a8180>,
     <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x77b3485a89a0>,
     <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x77b3485a8ae0>,
     <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x77b3485a8ae0>,
     <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x77b3485a9a80>,
     <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x77b3485a8c20>,
     <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x77b3485a9d00>,
     <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x77b3485a8d60>,
     <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x77b3485a8ea0>,
     <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x77b3485a8ea0>,
     <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x77b3485a8fe0>,
     <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x77b3485a9120>,
     <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x77b3485a9f80>,
     <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x77b3485a9260>,
     <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x77b3485a93a0>,
     <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x77b34857fb00>,
     <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x77b3485a94e0>,
     <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x77b3485a9620>,
     <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x77b3485a9760>,
     <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x77b3485a98a0>,
     <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x77b34857f600>,
     <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x77b34857f600>,
     <OpOverload(op='aten.square_', overload='default')>: <function square at 0x77b3485a99e0>,
     <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x77b3485a9b20>,
     <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x77b3485a9b20>,
     <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x77b34857d760>,
     <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x77b3485a9c60>,
     <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x77b3485a9300>,
     <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x77b3485a9300>,
     <OpOverload(op='aten.as_strided_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x77b3485a8cc0>,
     <OpOverload(op='aten.as_strided_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x77b3485a8cc0>,
     <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x77b3485a8540>,
     <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x77b3485a8540>,
     <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x77b3485aa160>,
     <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x77b3485aa160>,
     <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x77b3485aa480>,
     <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x77b3485aa480>,
     <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x77b3485aaac0>,
     <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x77b3485aade0>,
     <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x77b3485ab100>,
     <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x77b3485ab100>,
     <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x77b3485aa840>,
     <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x77b3485aa840>,
     <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x77b3485e04a0>,
     <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x77b3485e04a0>,
     <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x77b3485e0720>,
     <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x77b3485e0720>,
     <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x77b3485a9ee0>,
     <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x77b3485e11c0>,
     <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x77b3485e11c0>,
     <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x77b3485a9ee0>,
     <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x77b3485e0680>,
     <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x77b3485e0680>,
     <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x77b3485e14e0>,
     <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x77b3485e14e0>,
     <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x77b3485e1760>,
     <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x77b3485e1760>,
     <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x77b3485e19e0>,
     <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x77b3485e19e0>,
     <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x77b3485e22a0>,
     <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x77b3485e22a0>,
     <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x77b3485e2520>,
     <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x77b3485e2520>,
     <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x77b3485e27a0>,
     <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x77b3485e27a0>,
     <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x77b3485e2980>,
     <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x77b3485e2a20>,
     <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x77b3485e2a20>,
     <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x77b3485e2980>,
     <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x77b3485e3100>,
     <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x77b3485e3100>,
     <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x77b3485e3380>,
     <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x77b3485e3380>,
     <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x77b3485e1940>,
     <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x77b3485e1940>,
     <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x77b3485e3420>,
     <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x77b3485e3420>,
     <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x77b3485e36a0>,
     <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x77b3485e36a0>,
     <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x77b3485e3920>,
     <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x77b3485e3920>,
     <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x77b3485e3ba0>,
     <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x77b3485e3ba0>,
     <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x77b3485e3c40>,
     <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x77b3485e3ce0>,
     <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x77b348418400>,
     <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x77b348418400>,
     <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x77b3487fb740>,
     <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x77b34866d080>,
     <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x77b34866d080>,
     <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x77b34857f380>,
     <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x77b34866eac0>,
     <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x77b34866eac0>,
     <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x77b34866f4c0>,
     <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x77b34866f4c0>,
     <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x77b34866fec0>,
     <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x77b34866fec0>,
     <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x77b34866db20>,
     <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x77b34866db20>,
     <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x77b3486802c0>,
     <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x77b3486802c0>,
     <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x77b3486807c0>,
     <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x77b3486807c0>,
     <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x77b3486811c0>,
     <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x77b3486811c0>,
     <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x77b348682b60>,
     <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x77b348682b60>,
     <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x77b348694680>,
     <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x77b348694680>,
     <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x77b348694b80>,
     <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x77b348694b80>,
     <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x77b348695f80>,
     <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x77b348695f80>,
     <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x77b34841bce0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x77b3486ca980>,
     <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x77b3486ca980>,
     <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x77b3486c8540>,
     <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x77b3486c9800>,
     <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x77b3486c9800>,
     <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x77b3486ca0c0>,
     <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x77b3486ca0c0>,
     <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x77b3486ca520>,
     <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x77b3486c98a0>,
     <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x77b3486c98a0>,
     <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x77b3486cab60>,
     <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x77b3486cab60>,
     <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x77b3486cafc0>,
     <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x77b3486cafc0>,
     <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x77b3486e1d00>,
     <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x77b3486e1d00>,
     <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x77b3486e20c0>,
     <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x77b3486e20c0>,
     <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.all', overload='default')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dim')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dims')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.any', overload='default')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dim')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dims')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.std', overload='default')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='dim')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x77b348701440>,
     <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x77b348701440>,
     <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x77b348702ca0>,
     <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x77b348702520>,
     <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x77b348702520>,
     <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x77b348703060>,
     <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x77b348702700>,
     <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x77b348703740>,
     <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x77b348702d40>,
     <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x77b348702d40>,
     <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x77b348703740>,
     <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x77b3485244a0>,
     <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x77b348701080>,
     <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x77b348701080>,
     <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x77b348703c40>,
     <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x77b348703c40>,
     <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x77b3485242c0>,
     <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x77b348524540>,
     <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x77b348524540>,
     <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x77b348524860>,
     <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x77b348524860>,
     <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x77b34890a020>,
     <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x77b34890a340>,
     <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x77b34890a3e0>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x77b34890a480>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x77b34890a480>,
     <OpOverload(op='aten.ldexp', overload='out')>: <function ldexp at 0x77b348908860>,
     <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x77b34890ac00>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x77b34890b4c0>,
     <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x77b34890aca0>,
     <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x77b34890aca0>,
     <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x77b34890afc0>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x77b34890b6a0>,
     <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x77b34890b420>,
     <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x77b34890b880>,
     <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x77b34890b380>,
     <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x77b34890ba60>,
     <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x77b34890bce0>,
     <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x77b34890bce0>,
     <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x77b3487340e0>,
     <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x77b3487340e0>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x77b348734400>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x77b348734400>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x77b34890b100>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x77b34890b100>,
     <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x77b348736840>,
     <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x77b3487347c0>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x77b3487363e0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x77b348734860>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x77b348734860>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x77b3487363e0>,
     <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x77b348734e00>,
     <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x77b348734e00>,
     <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x77b348735080>,
     <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x77b348735080>,
     <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x77b348735300>,
     <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x77b348735bc0>,
     <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x77b348735bc0>,
     <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x77b3487359e0>,
     <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x77b348735c60>,
     <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x77b348735da0>,
     <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x77b348735da0>,
     <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x77b348736980>,
     <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x77b348734d60>,
     <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x77b348734d60>,
     <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x77b348736520>,
     <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x77b348736520>,
     <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x77b3487367a0>,
     <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x77b3487367a0>,
     <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x77b348737240>,
     <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x77b348736ac0>,
     <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x77b348736a20>,
     <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x77b348737420>,
     <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x77b348737100>,
     <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x77b348737100>,
     <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x77b3487374c0>,
     <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x77b3487374c0>,
     <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x77b348737740>,
     <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x77b348737740>,
     <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x77b348737920>,
     <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x77b3487379c0>,
     <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x77b3487379c0>,
     <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x77b348737f60>,
     <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x77b348737f60>,
     <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x77b348737c40>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x77b348737ba0>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x77b348737ba0>,
     <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x77b34875c9a0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x77b34875ca40>,
     <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x77b34875c5e0>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x77b34875ccc0>,
     <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x77b34875c900>,
     <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x77b34875c900>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x77b34875cd60>,
     <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x77b34875cfe0>,
     <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x77b34875dbc0>,
     <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x77b34875dc60>,
     <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x77b34875d800>,
     <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x77b34875d800>,
     <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x77b34875dee0>,
     <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x77b34875e160>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x77b34875ede0>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x77b34875ede0>,
     <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x77b34875cc20>,
     <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x77b34875cc20>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x77b34875eca0>,
     <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x77b34875ee80>,
     <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x77b34875e3e0>,
     <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x77b34875e3e0>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x77b34875ef20>,
     <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x77b34875f060>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x77b34875eca0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x77b34875f7e0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x77b34875f7e0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x77b34875fec0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x77b34875fec0>,
     <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x77b3487376a0>,
     <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x77b3487376a0>,
     <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x77b348794720>,
     <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x77b348794220>,
     <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x77b348794220>,
     <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x77b348794040>,
     <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x77b348796a20>,
     <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x77b348796a20>,
     <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x77b348794680>,
     <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x77b348794680>,
     <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x77b348795e40>,
     <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x77b348796700>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x77b348795300>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x77b348795300>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x77b348795620>,
     <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x77b3487954e0>,
     <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x77b348795e40>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x77b348794540>,
     <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x77b3487963e0>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x77b348794540>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x77b3487963e0>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x77b348796700>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x77b348796d40>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x77b348796d40>,
     <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x77b3487a93a0>,
     <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x77b348797560>,
     <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x77b348797740>,
     <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x77b348797920>,
     <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x77b348797b00>,
     <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x77b348797f60>,
     <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x77b3487a8180>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x77b3487a8400>,
     <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x77b3487a84a0>,
     <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x77b3487a8680>,
     <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x77b3487abe20>,
     <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x77b3487a8860>,
     <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x77b3487a8a40>,
     <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x77b3487a9120>,
     <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x77b3487a9120>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x77b348796980>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x77b3487a9440>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x77b3487a8f40>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x77b3487a9440>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x77b348796980>,
     <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x77b348795c60>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x77b3487a8400>,
     <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x77b3487a85e0>,
     <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x77b3487a80e0>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x77b3487ab9c0>,
     <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x77b3487a9b20>,
     <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x77b3487a9b20>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x77b3487aa200>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x77b3487aa200>,
     <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x77b3487aaca0>,
     <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x77b3487aaca0>,
     <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x77b3487ab060>,
     <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x77b3487ab060>,
     <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x77b3487ab380>,
     <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x77b3487ab380>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x77b3487ab600>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x77b3487ab600>,
     <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x77b3487d00e0>,
     <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2a20>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x77b3487a82c0>,
     <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2a20>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d04a0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d04a0>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x77b3487ab920>,
     <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x77b3487d0cc0>,
     <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x77b3487d0cc0>,
     <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x77b3487ab920>,
     <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x77b3487d1080>,
     <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x77b3487d1080>,
     <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x77b3487d1800>,
     <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x77b3487d00e0>,
     <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487abc40>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x77b3487a82c0>,
     <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x77b3487d2660>,
     <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x77b3487d23e0>,
     <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x77b3487d23e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d27a0>,
     <OpOverload(op='aten.take', overload='default')>: <function take at 0x77b3487d2d40>,
     <OpOverload(op='aten.take', overload='out')>: <function take at 0x77b3487d2d40>,
     <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x77b3487d2b60>,
     <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x77b3487d2b60>,
     <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2ca0>,
     <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2de0>,
     <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3240>,
     <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2f20>,
     <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3060>,
     <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2fc0>,
     <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2480>,
     <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2480>,
     <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2200>,
     <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2200>,
     <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1940>,
     <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1120>,
     <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3380>,
     <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3380>,
     <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d34c0>,
     <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d34c0>,
     <OpOverload(op='aten.ldexp_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3600>,
     <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3740>,
     <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3880>,
     <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d39c0>,
     <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3b00>,
     <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3c40>,
     <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3c40>,
     <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3ec0>,
     <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x77b3487f8040>,
     <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487f8180>,
     <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x77b3487d22a0>,
     <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x77b3487d22a0>,
     <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x77b3487013a0>,
     <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x77b3485267a0>,
     <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x77b3487019e0>,
     <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x77b34890ac00>,
     <OpOverload(op='aten.where', overload='self')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x77b3486e3b00>,
     <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x77b3487d2160>,
     <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x77b3486e3b00>,
     <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x77b3487d2160>,
     <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.var', overload='default')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='correction')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='dim')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x77b348735c60>,
     <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x77b3487000e0>,
     <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x77b3487000e0>,
     <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x77b3486e3f60>,
     <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x77b3486e3f60>,
     <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x77b34841a980>,
     <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x77b3487347c0>,
     <OpOverload(op='aten.full', overload='out')>: <function full at 0x77b34854cfe0>,
     <OpOverload(op='aten.full', overload='default')>: <function full at 0x77b34854cfe0>,
     <OpOverload(op='aten.ldexp', overload='Tensor')>: <function ldexp at 0x77b348908860>,
     <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x77b3488ebb00>,
     <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x77b34857f380>,
     <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1120>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x77b3487956c0>,
     <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x77b3487956c0>,
     <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x77b3486a77e0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x77b3486a77e0>,
     <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x77b3486c8540>,
     <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x77b3488eb740>,
     <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x77b3488eb740>,
     <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x77b3488ebb00>,
     <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x77b3488eb6a0>,
     <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x77b3488eb6a0>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x77b3488eb920>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x77b3488eb920>,
     <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x77b3489082c0>,
     <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x77b3489082c0>,
     <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x77b348908cc0>,
     <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x77b348908cc0>,
     <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x77b348908fe0>,
     <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x77b348908fe0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x77b34890a8e0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x77b34890a8e0>,
     <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x77b348909080>,
     <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x77b348909080>,
     <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x77b3489093a0>,
     <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x77b3489093a0>,
     <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x77b348909760>,
     <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x77b348909760>,
     <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x77b348909f80>,
     <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x77b348909f80>,
     <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.round', overload='default')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x77b348694180>,
     <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x77b348695080>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x77b348695080>,
     <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x77b3485242c0>,
     <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x77b348695580>,
     <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x77b348695580>,
     <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x77b348695a80>,
     <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x77b348695a80>,
     <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x77b348683920>,
     <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x77b348683920>,
     <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x77b34854d6c0>,
     <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x77b348694720>,
     <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x77b348694720>,
     <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x77b348696c00>,
     <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x77b348696c00>,
     <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x77b348697100>,
     <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x77b348697100>,
     <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x77b348697600>,
     <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x77b348697600>,
     <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x77b3486ca520>,
     <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.add', overload='out')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x77b348697f60>,
     <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x77b34875c5e0>,
     <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x77b348697f60>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x77b3486c89a0>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.div', overload='out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x77b3486a6980>,
     <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x77b3486a6980>,
     <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x77b3486a6de0>,
     <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x77b3486a6de0>,
     <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x77b34890b420>,
     <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x77b3486a6a20>,
     <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x77b3486a6a20>,
     <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x77b3485267a0>,
     <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x77b3486c89a0>,
     <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x77b3486cb880>,
     <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x77b3486c8e00>,
     <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x77b3486c9260>,
     <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x77b3486c9260>,
     <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x77b348cf77e0>,
     <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x77b3486cb880>,
     <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x77b3486cbce0>,
     <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x77b3486cbce0>,
     <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x77b348908720>,
     <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x77b3486e0a40>,
     <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x77b3486e0a40>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x77b348908720>,
     <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x77b348908220>,
     <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x77b3487f9bc0>,
     <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x77b3487f9bc0>,
     <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x77b3487fa200>,
     <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x77b3487fa200>,
     <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x77b3487fa700>,
     <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x77b3487fa700>,
     <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x77b34890b380>,
     <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x77b3487fac00>,
     <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x77b3487fac00>,
     <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x77b3487fb100>,
     <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x77b3487fb100>,
     <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x77b34866c400>,
     <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x77b34866c400>,
     <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x77b34866c900>,
     <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x77b34866c900>,
     <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x77b3486e3240>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x77b3487fb600>,
     <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x77b3487fb600>,
     <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x77b3487fbb00>,
     <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x77b3487fbb00>,
     <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x77b3487fbec0>,
     <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x77b3487fbec0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x77b3486e3240>,
     <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x77b34866ce00>,
     <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x77b34866ce00>,
     <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x77b348cf77e0>,
     <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x77b3487f9c60>,
     <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x77b348694180>,
     <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x77b34866d580>,
     <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x77b34866d580>,
     <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x77b34866da80>,
     <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x77b34866da80>,
     <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x77b34866df80>,
     <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x77b34866df80>,
     <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x77b34866e480>,
     <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x77b34866e480>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x77b348737c40>,
     <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x77b348908180>,
     <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x77b34866efc0>,
     <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x77b34866efc0>,
     <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x77b3485aaac0>,
     <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x77b34875dda0>,
     <OpOverload(op='aten.log', overload='out')>: <function log at 0x77b348681bc0>,
     <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x77b3486816c0>,
     <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x77b3486816c0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.log', overload='default')>: <function log at 0x77b348681bc0>,
     <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x77b348682ac0>,
     <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x77b3486820c0>,
     <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x77b3486820c0>,
     <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x77b3486825c0>,
     <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x77b3486825c0>,
     <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x77b348682ac0>,
     <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x77b348683880>,
     <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x77b348683880>,
     <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x77b348683240>,
     <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x77b348683240>,
     <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x77b3487f9c60>,
     <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x77b3487d2e80>,
     <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x77b3487d2e80>,
     <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x77b3487f8b80>,
     <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x77b3487f8b80>,
     <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x77b348795e40>,
     <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x77b34890a020>,
     <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x77b348909b20>,
     <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x77b34875cea0>,
     <OpOverload(op='aten.round', overload='out')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x77b3487a8f40>,
     <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x77b348524220>,
     <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x77b34857d800>,
     <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x77b3487d1ee0>,
     <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x77b348795620>,
     <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x77b3487019e0>,
     <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x77b348527060>,
     <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x77b34854d6c0>,
     <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten.view', overload='default')>: <function view at 0x77b348527100>,
     <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x77b34854c540>,
     <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x77b34854d9e0>,
     <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x77b34854d9e0>,
     <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x77b34854c0e0>,
     <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x77b34854c0e0>,
     <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x77b348527ce0>,
     <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x77b348527ce0>,
     <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x77b348527f60>,
     <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x77b348527f60>,
     <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x77b34854c180>,
     <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x77b34854c180>,
     <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x77b348526980>,
     <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x77b348526980>,
     <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x77b348525940>,
     <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x77b348525940>,
     <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.item', overload='default')>: <function item at 0x77b3486e2a20>,
     <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x77b3487a89a0>,
     <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x77b348735760>,
     <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x77b348735760>,
     <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x77b348525da0>,
     <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1da0>,
     <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x77b3486c8e00>,
     <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x77b34854c540>}

experimental_experiment.torch_dynamo.get_decomposition_table()

Returns the decomposition table needed to translate backward graph into onnx. It should used as follows:

import torch
from torch._dynamo.backends.common import aot_autograd
from experimental_experiment.torch_dynamo import get_decomposition_table

aot_compiler = aot_autograd(
    fw_compiler=backend_debug,
    decompositions=get_decomposition_table()
)

compiled_model = torch.compile(
    model,
    backend=aot_compiler,
    dynamic=dynamic,
    fullgraph=fullgraph,
)

The value is:

<<<

import pprint
from experimental_experiment.torch_dynamo import get_decomposition_table

pprint.pprint(get_decomposition_table())

>>>

    {<OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x77b34875ccc0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x77b348737c40>}

experimental_experiment.torch_dynamo.get_decomposition_table_by_name(name: str)

Returns a predefined decomposition table.

Parameters:

name – name see below

Returns:

decomposition table

• ‘none’: do not decompose

• ‘default’: get_decomposition_table()

• ‘onnxscript’: get_decomposition_table_onnxscript()

• ‘dynamo’: get_decomposition_table_dynamo()

experimental_experiment.torch_dynamo.get_decomposition_table_dynamo(onnx_registry=None)

Returns the decomposition table needed for the dynamo exporter.

Parameters:

onnx_registry – sent to create_onnx_friendly_decomposition_table

The value is:

<<<

import pprint
from experimental_experiment.torch_dynamo import get_decomposition_table_dynamo

pprint.pprint(get_decomposition_table_dynamo())

>>>

    {<OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910413a0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>),
     <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0360>, kernel=<OpOverload(op='image.write_file', overload='default')>),
     <OpOverload(op='aten.special_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e480>, kernel=<OpOverload(op='aten.special_softmax', overload='default')>),
     <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463740>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>),
     <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463920>, kernel=<OpOverload(op='aten.orgqr', overload='default')>),
     <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463b00>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>),
     <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460cc0>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>),
     <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462520>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>),
     <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461620>, kernel=<OpOverload(op='aten.arctanh', overload='default')>),
     <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461bc0>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>),
     <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894616c0>, kernel=<OpOverload(op='aten.align_as', overload='default')>),
     <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1620>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>),
     <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e8e0>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>),
     <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5260>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>),
     <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86160>, kernel=<OpOverload(op='aten.dstack', overload='default')>),
     <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4fe0>, kernel=<OpOverload(op='aten.max', overload='names_dim')>),
     <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947dc60>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>),
     <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4360>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>),
     <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e3e0>, kernel=<OpOverload(op='aten.special_erfinv', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5300>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>),
     <OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1440>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>),
     <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894602c0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>),
     <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87c40>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460860>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>),
     <OpOverload(op='aten.max_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7880>, kernel=<OpOverload(op='aten.max_pool2d', overload='default')>),
     <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463e20>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>),
     <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461580>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>),
     <OpOverload(op='aten.linalg_inv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6f20>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>),
     <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b85760>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>),
     <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2480>, kernel=<OpOverload(op='image.decode_png', overload='default')>),
     <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2660>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>),
     <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e160>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>),
     <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461f80>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>),
     <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463600>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>),
     <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4a40>, kernel=<OpOverload(op='aten.size', overload='Dimname')>),
     <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463880>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>),
     <OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947dee0>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f420>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a51c0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>),
     <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fe20>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>),
     <OpOverload(op='aten.max_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894623e0>, kernel=<OpOverload(op='aten.max_pool3d', overload='default')>),
     <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86980>, kernel=<OpOverload(op='aten.inverse', overload='default')>),
     <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462b60>, kernel=<OpOverload(op='aten.argsort', overload='stable')>),
     <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a79c0>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>),
     <OpOverload(op='aten.embedding_bag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d800>, kernel=<OpOverload(op='aten.embedding_bag', overload='default')>),
     <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1bc0>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>),
     <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c720>, kernel=<OpOverload(op='aten.arctan2', overload='default')>),
     <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894620c0>, kernel=<OpOverload(op='aten.negative', overload='default')>),
     <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64eac0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>),
     <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2020>, kernel=<OpOverload(op='image.read_file', overload='default')>),
     <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f6a0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>),
     <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947df80>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>),
     <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e840>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>),
     <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1ee0>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>),
     <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6700>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>),
     <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ede0>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>),
     <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a67a0>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>),
     <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460e00>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>),
     <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291043380>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>),
     <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87740>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>),
     <OpOverload(op='aten.einsum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462200>, kernel=<OpOverload(op='aten.einsum', overload='default')>),
     <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7f60>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>),
     <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87060>, kernel=<OpOverload(op='aten.concatenate', overload='names')>),
     <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f2e0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>),
     <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86ca0>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>),
     <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fce0>, kernel=<OpOverload(op='aten.smm', overload='default')>),
     <OpOverload(op='aten.divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460fe0>, kernel=<OpOverload(op='aten.divide', overload='Scalar')>),
     <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87880>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>),
     <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86f20>, kernel=<OpOverload(op='aten.combinations', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894893a0>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>),
     <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84040>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>),
     <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6660>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>),
     <OpOverload(op='aten.instance_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461e40>, kernel=<OpOverload(op='aten.instance_norm', overload='default')>),
     <OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e7a0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>),
     <OpOverload(op='aten.special_erfc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c0e0>, kernel=<OpOverload(op='aten.special_erfc', overload='default')>),
     <OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947cc20>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>),
     <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e480>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>),
     <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c7c0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>),
     <OpOverload(op='aten.scaled_dot_product_attention', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f880>, kernel=<OpOverload(op='aten.scaled_dot_product_attention', overload='default')>),
     <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a58a0>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>),
     <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5c60>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a59e0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>),
     <OpOverload(op='aten.cross_entropy_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7a60>, kernel=<OpOverload(op='aten.cross_entropy_loss', overload='default')>),
     <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7100>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>),
     <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7560>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>),
     <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64d9e0>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>),
     <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f7e0>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>),
     <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fec0>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>),
     <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f9c0>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>),
     <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d4e0>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>),
     <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d760>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>),
     <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460400>, kernel=<OpOverload(op='aten.trapz', overload='dx')>),
     <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86b60>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>),
     <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b85bc0>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>),
     <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c540>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86a20>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>),
     <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86520>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>),
     <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463240>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>),
     <OpOverload(op='aten.affine_grid_generator_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d300>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>),
     <OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ef20>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fec0>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>),
     <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894605e0>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>),
     <OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64d8a0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910098a0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>),
     <OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2700>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>),
     <OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5b20>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>),
     <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7ab740>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>),
     <OpOverload(op='aten.conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910407c0>, kernel=<OpOverload(op='aten.conv3d', overload='default')>),
     <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042480>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>),
     <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009b20>, kernel=<OpOverload(op='aten.special_logit', overload='default')>),
     <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009ee0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7380>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>),
     <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947ef20>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>),
     <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462480>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>),
     <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461940>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>),
     <OpOverload(op='aten.conv1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86fc0>, kernel=<OpOverload(op='aten.conv1d', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462ca0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>),
     <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463a60>, kernel=<OpOverload(op='aten.is_neg', overload='default')>),
     <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463c40>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>),
     <OpOverload(op='aten.tile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e700>, kernel=<OpOverload(op='aten.tile', overload='default')>),
     <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894604a0>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>),
     <OpOverload(op='aten.max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f9c0>, kernel=<OpOverload(op='aten.max_pool1d', overload='default')>),
     <OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461120>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>),
     <OpOverload(op='aten.isclose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e0c0>, kernel=<OpOverload(op='aten.isclose', overload='default')>),
     <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b879c0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>),
     <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f560>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>),
     <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4680>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>),
     <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c680>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>),
     <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f420>, kernel=<OpOverload(op='aten.column_stack', overload='default')>),
     <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460a40>, kernel=<OpOverload(op='aten.stft', overload='default')>),
     <OpOverload(op='aten._debug_has_internal_overlap', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b863e0>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>),
     <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0e00>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>),
     <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4e00>, kernel=<OpOverload(op='aten.flipud', overload='default')>),
     <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4f40>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>),
     <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462de0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>),
     <OpOverload(op='aten.log_sigmoid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b874c0>, kernel=<OpOverload(op='aten.log_sigmoid', overload='default')>),
     <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f1a0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.subtract', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461da0>, kernel=<OpOverload(op='aten.subtract', overload='Tensor')>),
     <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c2c0>, kernel=<OpOverload(op='aten.diff', overload='default')>),
     <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f740>, kernel=<OpOverload(op='aten.argsort', overload='default')>),
     <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86c00>, kernel=<OpOverload(op='aten.pad', overload='default')>),
     <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e340>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>),
     <OpOverload(op='aten.layer_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463060>, kernel=<OpOverload(op='aten.layer_norm', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fa60>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>),
     <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4400>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>),
     <OpOverload(op='aten.greater', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64efc0>, kernel=<OpOverload(op='aten.greater', overload='Tensor')>),
     <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462160>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>),
     <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463ba0>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>),
     <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84400>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>),
     <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e980>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>),
     <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2340>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>),
     <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0c20>, kernel=<OpOverload(op='c10d_functional.broadcast', overload='default')>),
     <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b876a0>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>),
     <OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e2a0>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>),
     <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894632e0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>),
     <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84680>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>),
     <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894609a0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>),
     <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291008ea0>, kernel=<OpOverload(op='aten.fix', overload='default')>),
     <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b844a0>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>),
     <OpOverload(op='aten.group_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4cc0>, kernel=<OpOverload(op='aten.group_norm', overload='default')>),
     <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f880>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.atleast_2d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac863e0>, kernel=<OpOverload(op='aten.atleast_2d', overload='Sequence')>),
     <OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b865c0>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>),
     <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460f40>, kernel=<OpOverload(op='aten.special_i0', overload='default')>),
     <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e840>, kernel=<OpOverload(op='aten.pinverse', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894631a0>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>),
     <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba00e0>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>),
     <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009800>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>),
     <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac849a0>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>),
     <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b847c0>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>),
     <OpOverload(op='aten.log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac86520>, kernel=<OpOverload(op='aten.log_softmax', overload='int')>),
     <OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7ce0>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>),
     <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cae6fa60>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>),
     <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b85940>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>),
     <OpOverload(op='image._is_compiled_against_turbo', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a47c0>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>),
     <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041ee0>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>),
     <OpOverload(op='aten.special_expm1', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291043060>, kernel=<OpOverload(op='aten.special_expm1', overload='default')>),
     <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2a4107a60>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489f80>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>),
     <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87420>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>),
     <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947cfe0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>),
     <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463f60>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f240>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>),
     <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e340>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894625c0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290208040>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>),
     <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042d40>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>),
     <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6a20>, kernel=<OpOverload(op='_test.cat', overload='default')>),
     <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291043740>, kernel=<OpOverload(op='aten.min', overload='names_dim')>),
     <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462700>, kernel=<OpOverload(op='aten.row_stack', overload='default')>),
     <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f6a0>, kernel=<OpOverload(op='aten.is_conj', overload='default')>),
     <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894607c0>, kernel=<OpOverload(op='aten.is_inference', overload='default')>),
     <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463100>, kernel=<OpOverload(op='aten.argwhere', overload='default')>),
     <OpOverload(op='aten.atleast_1d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461800>, kernel=<OpOverload(op='aten.atleast_1d', overload='Sequence')>),
     <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b840e0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488680>, kernel=<OpOverload(op='aten.cov', overload='default')>),
     <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040fe0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009da0>, kernel=<OpOverload(op='aten.to_dense', overload='default')>),
     <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b858a0>, kernel=<OpOverload(op='aten.cdist', overload='default')>),
     <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041e40>, kernel=<OpOverload(op='aten.diagflat', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910436a0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>),
     <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28948a0c0>, kernel=<OpOverload(op='aten.stride', overload='int')>),
     <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291043f60>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>),
     <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040cc0>, kernel=<OpOverload(op='aten.kron', overload='default')>),
     <OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84360>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>),
     <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910428e0>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>),
     <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6340>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>),
     <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100b1a0>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>),
     <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1800>, kernel=<OpOverload(op='image.decode_image', overload='default')>),
     <OpOverload(op='aten.subtract', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947ede0>, kernel=<OpOverload(op='aten.subtract', overload='Scalar')>),
     <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463ce0>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>),
     <OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ff60>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>),
     <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f560>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba23e0>, kernel=<OpOverload(op='image.encode_png', overload='default')>),
     <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ee80>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>),
     <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7740>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>),
     <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e700>, kernel=<OpOverload(op='aten.output_nr', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a42c0>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>),
     <OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7c40>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>),
     <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f380>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>),
     <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894618a0>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>),
     <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ea20>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>),
     <OpOverload(op='aten.logdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f2e0>, kernel=<OpOverload(op='aten.logdet', overload='default')>),
     <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b85800>, kernel=<OpOverload(op='aten.sort', overload='dimname')>),
     <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fc40>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>),
     <OpOverload(op='aten.max_pool1d_with_indices', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a72e0>, kernel=<OpOverload(op='aten.max_pool1d_with_indices', overload='default')>),
     <OpOverload(op='aten.true_divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a62a0>, kernel=<OpOverload(op='aten.true_divide', overload='Scalar')>),
     <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7240>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>),
     <OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a54e0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>),
     <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947ca40>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>),
     <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e020>, kernel=<OpOverload(op='aten.max', overload='other')>),
     <OpOverload(op='aten.concatenate', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5a80>, kernel=<OpOverload(op='aten.concatenate', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291191b20>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462f20>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>),
     <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f920>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>),
     <OpOverload(op='aten.linalg_det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f740>, kernel=<OpOverload(op='aten.linalg_det', overload='default')>),
     <OpOverload(op='aten.cross', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5620>, kernel=<OpOverload(op='aten.cross', overload='default')>),
     <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e8e0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>),
     <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462020>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>),
     <OpOverload(op='aten.avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6fc0>, kernel=<OpOverload(op='aten.avg_pool1d', overload='default')>),
     <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5800>, kernel=<OpOverload(op='aten.bilinear', overload='default')>),
     <OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e660>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042340>, kernel=<OpOverload(op='aten.one_hot', overload='default')>),
     <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7ec0>, kernel=<OpOverload(op='aten.median', overload='names_dim')>),
     <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f1a0>, kernel=<OpOverload(op='aten.gather', overload='dimname')>),
     <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fd80>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>),
     <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947dbc0>, kernel=<OpOverload(op='aten.arcsin', overload='default')>),
     <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042f20>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>),
     <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f100>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>),
     <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7d80>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>),
     <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f600>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>),
     <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e5c0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894613a0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>),
     <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b85c60>, kernel=<OpOverload(op='aten.qr', overload='default')>),
     <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460ea0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>),
     <OpOverload(op='aten.less', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c400>, kernel=<OpOverload(op='aten.less', overload='Tensor')>),
     <OpOverload(op='aten.special_log_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462c00>, kernel=<OpOverload(op='aten.special_log_softmax', overload='default')>),
     <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84860>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>),
     <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ec00>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>),
     <OpOverload(op='aten.is_nonzero', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460040>, kernel=<OpOverload(op='aten.is_nonzero', overload='default')>),
     <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86700>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>),
     <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64ed40>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>),
     <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b860c0>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e7a0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>),
     <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87920>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>),
     <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5f80>, kernel=<OpOverload(op='aten.quantile', overload='default')>),
     <OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4220>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6480>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>),
     <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460540>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>),
     <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b856c0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>),
     <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8aa20>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>),
     <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2c91bfec0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>),
     <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489760>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>),
     <OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461260>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>),
     <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894896c0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>),
     <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7600>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>),
     <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460680>, kernel=<OpOverload(op='aten.min', overload='other')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894882c0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>),
     <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947de40>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6de0>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460360>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>),
     <OpOverload(op='aten.special_erf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460720>, kernel=<OpOverload(op='aten.special_erf', overload='default')>),
     <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac856c0>, kernel=<OpOverload(op='aten.special_round', overload='default')>),
     <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488cc0>, kernel=<OpOverload(op='aten.vstack', overload='default')>),
     <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488f40>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>),
     <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7ab7e0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>),
     <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894627a0>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>),
     <OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87a60>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>),
     <OpOverload(op='aten.grid_sampler', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d260>, kernel=<OpOverload(op='aten.grid_sampler', overload='default')>),
     <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894636a0>, kernel=<OpOverload(op='aten.gradient', overload='array')>),
     <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cae6f1a0>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>),
     <OpOverload(op='aten.det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b842c0>, kernel=<OpOverload(op='aten.det', overload='default')>),
     <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba20c0>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>),
     <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489bc0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>),
     <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28948bf60>, kernel=<OpOverload(op='aten.trapz', overload='x')>),
     <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a700>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>),
     <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e520>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>),
     <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7ab4c0>, kernel=<OpOverload(op='aten.nanmean', overload='default')>),
     <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84a40>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040ae0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>),
     <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f060>, kernel=<OpOverload(op='aten.size', overload='int')>),
     <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7ba0>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>),
     <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910093a0>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>),
     <OpOverload(op='aten.less_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d3a0>, kernel=<OpOverload(op='aten.less_equal', overload='Tensor')>),
     <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894628e0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>),
     <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b868e0>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>),
     <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2c83cb9c0>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894634c0>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>),
     <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100b6a0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>),
     <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a200>, kernel=<OpOverload(op='aten._backward', overload='default')>),
     <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041440>, kernel=<OpOverload(op='aten.msort', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947ee80>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>),
     <OpOverload(op='aten.softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d6c0>, kernel=<OpOverload(op='aten.softmax', overload='int')>),
     <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a56c0>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>),
     <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041620>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>),
     <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041760>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>),
     <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2a4107920>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>),
     <OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488a40>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>),
     <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28948ba60>, kernel=<OpOverload(op='aten.is_leaf', overload='default')>),
     <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a63e0>, kernel=<OpOverload(op='aten.ger', overload='default')>),
     <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba18a0>, kernel=<OpOverload(op='image.decode_gif', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6200>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>),
     <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040680>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>),
     <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489940>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>),
     <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c860>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>),
     <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7e20>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>),
     <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a68e0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>),
     <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947ff60>, kernel=<OpOverload(op='aten.arccosh', overload='default')>),
     <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac85080>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>),
     <OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2c83cba60>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2a4107ba0>, kernel=<OpOverload(op='aten.chalf', overload='default')>),
     <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a71a0>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>),
     <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f4c0>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461c60>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>),
     <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fb00>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>),
     <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4900>, kernel=<OpOverload(op='aten.data', overload='default')>),
     <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7a8cc0>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>),
     <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460220>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>),
     <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8aca0>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>),
     <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5440>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>),
     <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947db20>, kernel=<OpOverload(op='aten.arccos', overload='default')>),
     <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5da0>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>),
     <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009940>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>),
     <OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042980>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>),
     <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7b00>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>),
     <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5580>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>),
     <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87100>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>),
     <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894889a0>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489440>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>),
     <OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461440>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>),
     <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894891c0>, kernel=<OpOverload(op='aten.rrelu', overload='default')>),
     <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fba0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>),
     <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a2a0>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>),
     <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042c00>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>),
     <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460180>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>),
     <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6160>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>),
     <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040e00>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>),
     <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291043420>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>),
     <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042b60>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>),
     <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a8e0>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>),
     <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a980>, kernel=<OpOverload(op='aten.hstack', overload='default')>),
     <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8a160>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488860>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>),
     <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947ea20>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>),
     <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910409a0>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>),
     <OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7a9940>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84540>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>),
     <OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7920>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84220>, kernel=<OpOverload(op='aten.fliplr', overload='default')>),
     <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947e200>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>),
     <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291191d00>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>),
     <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488360>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>),
     <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2200>, kernel=<OpOverload(op='image.decode_webp', overload='default')>),
     <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fba0>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>),
     <OpOverload(op='aten.concat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947cea0>, kernel=<OpOverload(op='aten.concat', overload='default')>),
     <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fd80>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>),
     <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910419e0>, kernel=<OpOverload(op='aten.outer', overload='default')>),
     <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a020>, kernel=<OpOverload(op='aten.is_signed', overload='default')>),
     <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910404a0>, kernel=<OpOverload(op='aten.vander', overload='default')>),
     <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041260>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>),
     <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a65c0>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>),
     <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c5e0>, kernel=<OpOverload(op='aten.concat', overload='names')>),
     <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5760>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041300>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>),
     <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910416c0>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>),
     <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009e40>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>),
     <OpOverload(op='aten.type_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910432e0>, kernel=<OpOverload(op='aten.type_as', overload='default')>),
     <OpOverload(op='aten.atleast_3d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6ca0>, kernel=<OpOverload(op='aten.atleast_3d', overload='Sequence')>),
     <OpOverload(op='aten.greater_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7a8d60>, kernel=<OpOverload(op='aten.greater_equal', overload='Tensor')>),
     <OpOverload(op='aten.special_sinc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fce0>, kernel=<OpOverload(op='aten.special_sinc', overload='default')>),
     <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289488ea0>, kernel=<OpOverload(op='aten._version', overload='default')>),
     <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910422a0>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a0c0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>),
     <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910427a0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>),
     <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64e2a0>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>),
     <OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a60c0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>),
     <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894600e0>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='other_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f380>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a7060>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>),
     <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4c20>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2a41077e0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>),
     <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4ea0>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>),
     <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461d00>, kernel=<OpOverload(op='aten.inner', overload='default')>),
     <OpOverload(op='aten.sym_size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894639c0>, kernel=<OpOverload(op='aten.sym_size', overload='int')>),
     <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894637e0>, kernel=<OpOverload(op='aten.kl_div', overload='default')>),
     <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463420>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>),
     <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d120>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>),
     <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84720>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>),
     <OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0ea0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>),
     <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b84900>, kernel=<OpOverload(op='aten.less', overload='Scalar')>),
     <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1c60>, kernel=<OpOverload(op='_test.get_first', overload='default')>),
     <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f600>, kernel=<OpOverload(op='aten._remove_batch_dim', overload='default')>),
     <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c900>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>),
     <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0860>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d580>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>),
     <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b845e0>, kernel=<OpOverload(op='aten.isreal', overload='default')>),
     <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7aa0c0>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>),
     <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b877e0>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>),
     <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489e40>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>),
     <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910099e0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>),
     <OpOverload(op='aten.multiply', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8a660>, kernel=<OpOverload(op='aten.multiply', overload='Tensor')>),
     <OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7ab880>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042ac0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>),
     <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c180>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>),
     <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86340>, kernel=<OpOverload(op='aten.absolute', overload='default')>),
     <OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1940>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>),
     <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462a20>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>),
     <OpOverload(op='aten.embedding_bag', overload='padding_idx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460d60>, kernel=<OpOverload(op='aten.embedding_bag', overload='padding_idx')>),
     <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947dd00>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac85e40>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>),
     <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040a40>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>),
     <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba02c0>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>),
     <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463380>, kernel=<OpOverload(op='aten.arctan', overload='default')>),
     <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8ad40>, kernel=<OpOverload(op='aten.special_psi', overload='default')>),
     <OpOverload(op='aten.divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291191bc0>, kernel=<OpOverload(op='aten.divide', overload='Tensor')>),
     <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040360>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>),
     <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cabdc720>, kernel=<OpOverload(op='aten.square', overload='default')>),
     <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8ab60>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>),
     <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8a0c0>, kernel=<OpOverload(op='aten.clip', overload='default')>),
     <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a840>, kernel=<OpOverload(op='aten.diag', overload='default')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8a480>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>),
     <OpOverload(op='aten.true_divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7a8e00>, kernel=<OpOverload(op='aten.true_divide', overload='Tensor')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291190ea0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>),
     <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x77b3484187c0>,
     <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x77b3484187c0>,
     <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x77b348419bc0>,
     <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x77b348419e40>,
     <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x77b348419e40>,
     <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x77b34841a520>,
     <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x77b34841a520>,
     <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x77b34841a980>,
     <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x77b34841afc0>,
     <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x77b34841afc0>,
     <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x77b34841ad40>,
     <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x77b34841ad40>,
     <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x77b34841b2e0>,
     <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x77b34841b2e0>,
     <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x77b348448400>,
     <OpOverload(op='aten.selu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b872e0>, kernel=<OpOverload(op='aten.selu', overload='default')>),
     <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x77b348448720>,
     <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x77b34841bce0>,
     <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x77b348448400>,
     <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x77b348448720>,
     <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460b80>, kernel=<OpOverload(op='aten.margin_ranking_loss', overload='default')>),
     <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462fc0>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>),
     <OpOverload(op='aten.nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894614e0>, kernel=<OpOverload(op='aten.nll_loss', overload='default')>),
     <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x77b348449760>,
     <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x77b348449b20>,
     <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x77b348449b20>,
     <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x77b34841b240>,
     <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x77b34841b240>,
     <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x77b348449da0>,
     <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x77b348449da0>,
     <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x77b34844a660>,
     <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x77b34844a660>,
     <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x77b34844bec0>,
     <OpOverload(op='aten.prelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947efc0>, kernel=<OpOverload(op='aten.prelu', overload='default')>),
     <OpOverload(op='aten.relu6', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d440>, kernel=<OpOverload(op='aten.relu6', overload='default')>),
     <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x77b34844b1a0>,
     <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x77b34844b1a0>,
     <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d1c0>, kernel=<OpOverload(op='aten.pairwise_distance', overload='default')>),
     <OpOverload(op='aten.pdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c360>, kernel=<OpOverload(op='aten.pdist', overload='default')>),
     <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x77b34844ba60>,
     <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x77b34844ba60>,
     <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x77b34844bce0>,
     <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x77b34844bce0>,
     <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x77b34844a7a0>,
     <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x77b34844bb00>,
     <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x77b34844bd80>,
     <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x77b34844bf60>,
     <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x77b348468220>,
     <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x77b348468220>,
     <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x77b348468680>,
     <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x77b348468680>,
     <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x77b348468a40>,
     <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x77b348468a40>,
     <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x77b348468e00>,
     <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x77b348468e00>,
     <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x77b34846a520>,
     <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x77b34846a520>,
     <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x77b3484693a0>,
     <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x77b3484693a0>,
     <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x77b3484698a0>,
     <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x77b3484698a0>,
     <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x77b348469da0>,
     <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x77b348469da0>,
     <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x77b34846a160>,
     <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x77b34846a160>,
     <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x77b34846a660>,
     <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x77b34846a660>,
     <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b871a0>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>),
     <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x77b34846aa20>,
     <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x77b34846ade0>,
     <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x77b34846ade0>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x77b34846b420>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x77b34846b420>,
     <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x77b348524ea0>,
     <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x77b348524ea0>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100b4c0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>),
     <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a7a0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>),
     <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x77b348702480>,
     <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.t', overload='default')>: <function t at 0x77b348526520>,
     <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x77b34854c7c0>,
     <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x77b348525da0>,
     <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x77b348524220>,
     <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x77b348526160>,
     <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x77b348526160>,
     <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x77b3485263e0>,
     <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x77b3485263e0>,
     <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x77b3485260c0>,
     <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x77b348526700>,
     <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x77b3485268e0>,
     <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x77b348526d40>,
     <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x77b348526d40>,
     <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894885e0>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>),
     <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289489b20>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>),
     <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x77b348526fc0>,
     <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x77b34854c7c0>,
     <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x77b34854dc60>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910423e0>, kernel=<OpOverload(op='aten.meshgrid', overload='default')>),
     <OpOverload(op='aten.meshgrid', overload='indexing')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042520>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>),
     <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x77b34854de40>,
     <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x77b34854de40>,
     <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291009760>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>),
     <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x77b34854e7a0>,
     <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x77b34854e7a0>,
     <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x77b34854f2e0>,
     <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x77b34854f2e0>,
     <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x77b34854eca0>,
     <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x77b34854eca0>,
     <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x77b34854ef20>,
     <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x77b34854ef20>,
     <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x77b34854e700>,
     <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x77b34854e700>,
     <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x77b34854f560>,
     <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x77b34854f560>,
     <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x77b34854f920>,
     <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x77b34854f920>,
     <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x77b34854fce0>,
     <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x77b34854fce0>,
     <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x77b34857c0e0>,
     <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x77b34857c0e0>,
     <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x77b34857c180>,
     <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x77b34857cae0>,
     <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x77b34857cae0>,
     <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x77b34857cfe0>,
     <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x77b34857cfe0>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x77b34857cb80>,
     <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x77b34857cb80>,
     <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x77b34857d580>,
     <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x77b34857d580>,
     <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x77b34857d800>,
     <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x77b34857ef20>,
     <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x77b34854f880>,
     <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x77b34857e5c0>,
     <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x77b34857d620>,
     <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x77b34857d8a0>,
     <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x77b34857e840>,
     <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x77b34857d9e0>,
     <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x77b34857d9e0>,
     <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x77b34857db20>,
     <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x77b34857df80>,
     <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x77b34857dc60>,
     <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x77b34857eac0>,
     <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x77b34857eac0>,
     <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x77b34857dda0>,
     <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x77b34857dee0>,
     <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x77b34857ed40>,
     <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x77b34857e020>,
     <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x77b34857e160>,
     <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x77b34857ed40>,
     <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x77b34857e2a0>,
     <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x77b34857e3e0>,
     <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x77b34857e3e0>,
     <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x77b34857efc0>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x77b34857e520>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x77b34857e520>,
     <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x77b3485a9da0>,
     <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x77b34857e660>,
     <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x77b34857e480>,
     <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x77b34857e7a0>,
     <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x77b34857e7a0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x77b34857e8e0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x77b34857e8e0>,
     <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x77b34857f240>,
     <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x77b34857ea20>,
     <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x77b34857ea20>,
     <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x77b34857eb60>,
     <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x77b34857eca0>,
     <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x77b34857eca0>,
     <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x77b34857f1a0>,
     <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x77b34857f060>,
     <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x77b34857ede0>,
     <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x77b34857ede0>,
     <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x77b34857f1a0>,
     <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x77b34857ef20>,
     <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x77b34857e980>,
     <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x77b34857da80>,
     <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x77b34857e0c0>,
     <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x77b34857e0c0>,
     <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x77b34857de40>,
     <OpOverload(op='aten.log_', overload='default')>: <function log at 0x77b34857d080>,
     <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x77b34857dbc0>,
     <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x77b34857d940>,
     <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x77b34857f4c0>,
     <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x77b34857d4e0>,
     <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x77b34857f2e0>,
     <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x77b34857f100>,
     <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x77b34857f420>,
     <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x77b34857f740>,
     <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x77b34857f560>,
     <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x77b34857f560>,
     <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x77b34857f9c0>,
     <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x77b34857f6a0>,
     <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x77b34857f7e0>,
     <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x77b34857f7e0>,
     <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x77b34857f920>,
     <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x77b34857f920>,
     <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x77b34857fc40>,
     <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x77b34857fa60>,
     <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x77b34857fba0>,
     <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x77b34857fce0>,
     <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x77b34857fce0>,
     <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x77b34857fd80>,
     <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x77b34857fe20>,
     <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x77b34857fe20>,
     <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x77b34857e700>,
     <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x77b34857fd80>,
     <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x77b34857ff60>,
     <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x77b3485aade0>,
     <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x77b3485a80e0>,
     <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x77b3485a8220>,
     <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x77b3485a8360>,
     <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x77b34857e200>,
     <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x77b3485a84a0>,
     <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x77b34857e200>,
     <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x77b34857fec0>,
     <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x77b34857fec0>,
     <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x77b3485a85e0>,
     <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x77b34857ee80>,
     <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x77b34857ee80>,
     <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x77b3485a8720>,
     <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x77b3485a8720>,
     <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x77b3485a8900>,
     <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x77b3485a8680>,
     <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x77b3485a8400>,
     <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x77b3485a8400>,
     <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x77b3485a9800>,
     <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x77b3485a8180>,
     <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x77b3485a89a0>,
     <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x77b3485a8ae0>,
     <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x77b3485a8ae0>,
     <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x77b3485a9a80>,
     <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x77b3485a8c20>,
     <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x77b3485a9d00>,
     <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x77b3485a8d60>,
     <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x77b3485a8ea0>,
     <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x77b3485a8ea0>,
     <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x77b3485a8fe0>,
     <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x77b3485a9120>,
     <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x77b3485a9f80>,
     <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x77b3485a9260>,
     <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x77b3485a93a0>,
     <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x77b34857fb00>,
     <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x77b3485a94e0>,
     <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x77b3485a9620>,
     <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x77b3485a9760>,
     <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x77b3485a98a0>,
     <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x77b34857f600>,
     <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x77b34857f600>,
     <OpOverload(op='aten.square_', overload='default')>: <function square at 0x77b3485a99e0>,
     <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x77b3485a9b20>,
     <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x77b3485a9b20>,
     <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x77b34857d760>,
     <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x77b3485a9c60>,
     <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x77b3485a9300>,
     <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x77b3485a9300>,
     <OpOverload(op='aten.as_strided_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x77b3485a8cc0>,
     <OpOverload(op='aten.as_strided_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x77b3485a8cc0>,
     <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x77b3485a8540>,
     <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x77b3485a8540>,
     <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x77b3485aa160>,
     <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x77b3485aa160>,
     <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x77b3485aa480>,
     <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x77b3485aa480>,
     <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x77b3485aaac0>,
     <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x77b3485aade0>,
     <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x77b3485ab100>,
     <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x77b3485ab100>,
     <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x77b3485aa840>,
     <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x77b3485aa840>,
     <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x77b3485e04a0>,
     <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x77b3485e04a0>,
     <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x77b3485e0720>,
     <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x77b3485e0720>,
     <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x77b3485a9ee0>,
     <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b859e0>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>),
     <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x77b3485e11c0>,
     <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894622a0>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>),
     <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462840>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>),
     <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x77b3485e0680>,
     <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463d80>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>),
     <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x77b3485e14e0>,
     <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461080>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>),
     <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x77b3485e1760>,
     <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462980>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>),
     <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x77b3485e19e0>,
     <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462e80>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>),
     <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x77b3485e22a0>,
     <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460ae0>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>),
     <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x77b3485e2520>,
     <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462ac0>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>),
     <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x77b3485e27a0>,
     <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x77b3485e2980>,
     <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894611c0>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>),
     <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x77b3485e2a20>,
     <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947fe20>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>),
     <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463560>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>),
     <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x77b3485e3100>,
     <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947cae0>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>),
     <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x77b3485e3380>,
     <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c220>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>),
     <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x77b3485e1940>,
     <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947c040>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>),
     <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x77b3485e3420>,
     <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f7e0>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>),
     <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x77b3485e36a0>,
     <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947dda0>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>),
     <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x77b3485e3920>,
     <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947f4c0>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>),
     <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x77b3485e3ba0>,
     <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f060>, kernel=<OpOverload(op='aten.fft_fftshift', overload='default')>),
     <OpOverload(op='aten.fft_ifftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64fb00>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>),
     <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x77b348418400>,
     <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x77b348418400>,
     <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291040900>, kernel=<OpOverload(op='aten.is_complex', overload='default')>),
     <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x77b34866d080>,
     <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x77b34866d080>,
     <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x77b34857f380>,
     <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x77b34866eac0>,
     <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x77b34866eac0>,
     <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x77b34866f4c0>,
     <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x77b34866f4c0>,
     <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x77b34866fec0>,
     <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x77b34866fec0>,
     <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x77b34866db20>,
     <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x77b34866db20>,
     <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x77b3486802c0>,
     <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x77b3486802c0>,
     <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x77b3486807c0>,
     <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x77b3486807c0>,
     <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x77b3486811c0>,
     <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x77b3486811c0>,
     <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8ac00>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>),
     <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x77b348682b60>,
     <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x77b348682b60>,
     <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x77b348694680>,
     <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x77b348694680>,
     <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x77b348694b80>,
     <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x77b348694b80>,
     <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x77b348695f80>,
     <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x77b348695f80>,
     <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x77b34841bce0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x77b3486ca980>,
     <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x77b3486ca980>,
     <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x77b3486c8540>,
     <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x77b3486c9800>,
     <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x77b3486c9800>,
     <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x77b3486ca0c0>,
     <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x77b3486ca0c0>,
     <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x77b3486ca520>,
     <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x77b3486c98a0>,
     <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x77b3486c98a0>,
     <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x77b3486cab60>,
     <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x77b3486cab60>,
     <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x77b3486cafc0>,
     <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x77b3486cafc0>,
     <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x77b3486e1d00>,
     <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x77b3486e1d00>,
     <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x77b3486e20c0>,
     <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x77b3486e20c0>,
     <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.all', overload='default')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dim')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dims')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac85120>, kernel=<OpOverload(op='aten.std_mean', overload='default')>),
     <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac862a0>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>),
     <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac84cc0>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>),
     <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.any', overload='default')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dim')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dims')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2cac86200>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>),
     <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64f920>, kernel=<OpOverload(op='aten.std', overload='default')>),
     <OpOverload(op='aten.std', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6020>, kernel=<OpOverload(op='aten.std', overload='dim')>),
     <OpOverload(op='aten.std', overload='out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4ae0>, kernel=<OpOverload(op='aten.std', overload='names_dim')>),
     <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6e80>, kernel=<OpOverload(op='aten.std', overload='correction_names')>),
     <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a40e0>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>),
     <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x77b348701b20>,
     <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291043ce0>, kernel=<OpOverload(op='aten.var_mean', overload='default')>),
     <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042160>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>),
     <OpOverload(op='aten.var_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2910431a0>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>),
     <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291041120>, kernel=<OpOverload(op='aten.var_mean', overload='correction_names')>),
     <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x77b348701440>,
     <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x77b348701440>,
     <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x77b348702ca0>,
     <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x77b348702520>,
     <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x77b348702520>,
     <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x77b348703060>,
     <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x77b348702700>,
     <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x77b348703740>,
     <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x77b348702d40>,
     <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x77b348702d40>,
     <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x77b348703740>,
     <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x77b348703880>,
     <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x77b3485244a0>,
     <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x77b348701080>,
     <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x77b348701080>,
     <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x77b348703c40>,
     <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x77b348703c40>,
     <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x77b3485242c0>,
     <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x77b348524540>,
     <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x77b348524540>,
     <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b290a8ade0>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>),
     <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x77b348524860>,
     <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x77b348524860>,
     <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x77b34890a020>,
     <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x77b34890a340>,
     <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x77b34890a3e0>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x77b34890a480>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x77b34890a480>,
     <OpOverload(op='aten.ldexp', overload='out')>: <function ldexp at 0x77b348908860>,
     <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x77b34890ac00>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x77b34890b4c0>,
     <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x77b34890aca0>,
     <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x77b34890aca0>,
     <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x77b34890afc0>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x77b34890b6a0>,
     <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x77b34890b420>,
     <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x77b34890b880>,
     <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x77b34890b380>,
     <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x77b34890ba60>,
     <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x77b34890bce0>,
     <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x77b34890bce0>,
     <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x77b3487340e0>,
     <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x77b3487340e0>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x77b348734400>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x77b348734400>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x77b34890b100>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x77b34890b100>,
     <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x77b348736840>,
     <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x77b3487347c0>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x77b3487363e0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x77b348734860>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x77b348734860>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x77b3487363e0>,
     <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x77b348734e00>,
     <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x77b348734e00>,
     <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x77b348735080>,
     <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x77b348735080>,
     <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x77b348735300>,
     <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x77b348735300>,
     <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x77b348735bc0>,
     <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x77b348735bc0>,
     <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x77b3487359e0>,
     <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x77b3487359e0>,
     <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x77b348735c60>,
     <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x77b348735da0>,
     <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x77b348735da0>,
     <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x77b348736980>,
     <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x77b348734d60>,
     <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x77b348734d60>,
     <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x77b348736520>,
     <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x77b348736520>,
     <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x77b3487367a0>,
     <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x77b3487367a0>,
     <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x77b348737240>,
     <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x77b348736ac0>,
     <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x77b348736a20>,
     <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x77b348737420>,
     <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x77b348737100>,
     <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x77b348737100>,
     <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x77b3487374c0>,
     <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x77b3487374c0>,
     <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x77b348737740>,
     <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x77b348737740>,
     <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x77b348737920>,
     <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x77b3487379c0>,
     <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x77b3487379c0>,
     <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x77b348737f60>,
     <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x77b348737f60>,
     <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x77b348737c40>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x77b348737ba0>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x77b348737ba0>,
     <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x77b34875c9a0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x77b34875ca40>,
     <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x77b34875c5e0>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x77b34875ccc0>,
     <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x77b34875c900>,
     <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x77b34875c900>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x77b34875cd60>,
     <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x77b34875cfe0>,
     <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x77b34875dbc0>,
     <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x77b34875dc60>,
     <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x77b34875d800>,
     <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x77b34875d800>,
     <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x77b34875dee0>,
     <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x77b34875e160>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x77b34875ede0>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x77b34875ede0>,
     <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x77b34875cc20>,
     <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x77b34875cc20>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x77b34875eca0>,
     <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x77b34875ee80>,
     <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x77b34875e3e0>,
     <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x77b34875e3e0>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x77b34875ef20>,
     <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x77b34875f060>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x77b34875eca0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x77b34875f7e0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x77b34875f7e0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x77b34875fec0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x77b34875fec0>,
     <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x77b3487376a0>,
     <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x77b3487376a0>,
     <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x77b348794720>,
     <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x77b348794220>,
     <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x77b348794220>,
     <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x77b348794040>,
     <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x77b348796a20>,
     <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289463ec0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>),
     <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x77b348796a20>,
     <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x77b348794680>,
     <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x77b348794680>,
     <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x77b348795e40>,
     <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28947d080>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>),
     <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x77b348796700>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x77b348795300>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x77b348795300>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x77b348795620>,
     <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x77b3487954e0>,
     <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x77b348795e40>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x77b348794540>,
     <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x77b3487963e0>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x77b348794540>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x77b3487963e0>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x77b348796700>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x77b348796d40>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x77b348796d40>,
     <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x77b3487a93a0>,
     <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x77b348797560>,
     <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x77b348797740>,
     <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x77b348797920>,
     <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x77b348797b00>,
     <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x77b348797f60>,
     <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x77b3487a8180>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x77b3487a8400>,
     <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x77b3487a84a0>,
     <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x77b3487a8680>,
     <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x77b3487abe20>,
     <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x77b3487a8860>,
     <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x77b3487a8a40>,
     <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x77b3487a9120>,
     <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x77b3487a9120>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x77b348796980>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x77b3487a9440>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x77b3487a8f40>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x77b3487a9440>,
     <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x77b3487a8f40>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x77b348796980>,
     <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x77b348795c60>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x77b3487a8400>,
     <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x77b3487a85e0>,
     <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x77b3487a80e0>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x77b3487ab9c0>,
     <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x77b3487a9b20>,
     <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x77b3487a9b20>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x77b3487aa200>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x77b3487aa200>,
     <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x77b3487aaca0>,
     <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x77b3487aaca0>,
     <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x77b3487ab060>,
     <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x77b3487ab060>,
     <OpOverload(op='aten.matmul', overload='default')>: <function matmul at 0x77b3487aafc0>,
     <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x77b3487ab380>,
     <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x77b3487ab380>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x77b3487ab600>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x77b3487ab600>,
     <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x77b3487d00e0>,
     <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2a20>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x77b3487a82c0>,
     <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2a20>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d04a0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d04a0>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x77b3487ab920>,
     <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x77b3487d0cc0>,
     <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x77b3487d0cc0>,
     <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x77b3487ab920>,
     <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x77b3487d1080>,
     <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x77b3487d1080>,
     <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x77b3487d1800>,
     <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x77b3487d00e0>,
     <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487abc40>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x77b3487a82c0>,
     <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x77b3487d2660>,
     <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x77b3487d23e0>,
     <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x77b3487d23e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d27a0>,
     <OpOverload(op='aten.take', overload='default')>: <function take at 0x77b3487d2d40>,
     <OpOverload(op='aten.take', overload='out')>: <function take at 0x77b3487d2d40>,
     <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x77b3487d2b60>,
     <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x77b3487d2b60>,
     <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2ca0>,
     <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2de0>,
     <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3240>,
     <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2f20>,
     <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3060>,
     <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2fc0>,
     <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2480>,
     <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2480>,
     <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2200>,
     <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2200>,
     <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1940>,
     <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1120>,
     <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3380>,
     <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3380>,
     <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d34c0>,
     <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d34c0>,
     <OpOverload(op='aten.ldexp_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3600>,
     <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3740>,
     <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3880>,
     <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d39c0>,
     <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3b00>,
     <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3c40>,
     <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3c40>,
     <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3ec0>,
     <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x77b3487f8040>,
     <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487f8180>,
     <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x77b3487d22a0>,
     <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x77b3487d22a0>,
     <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x77b3487013a0>,
     <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x77b3485267a0>,
     <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x77b3487019e0>,
     <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c7ab100>, kernel=<OpOverload(op='aten.cat', overload='names')>),
     <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x77b34890ac00>,
     <OpOverload(op='aten.where', overload='self')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.where', overload='ScalarOther')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894894e0>, kernel=<OpOverload(op='aten.where', overload='ScalarOther')>),
     <OpOverload(op='aten.where', overload='ScalarSelf')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28948bc40>, kernel=<OpOverload(op='aten.where', overload='ScalarSelf')>),
     <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28948b380>, kernel=<OpOverload(op='aten.where', overload='default')>),
     <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x77b3486e3b00>,
     <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x77b3487d2160>,
     <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x77b3486e3b00>,
     <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b87e20>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>),
     <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x77b3487d2160>,
     <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460c20>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>),
     <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289462340>, kernel=<OpOverload(op='aten.var', overload='default')>),
     <OpOverload(op='aten.var', overload='correction')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289460900>, kernel=<OpOverload(op='aten.var', overload='names_dim')>),
     <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461300>, kernel=<OpOverload(op='aten.var', overload='dim')>),
     <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x77b348735c60>,
     <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b289461b20>, kernel=<OpOverload(op='aten.var', overload='correction_names')>),
     <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x77b3487000e0>,
     <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x77b3487000e0>,
     <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x77b3486e3f60>,
     <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x77b3486e3f60>,
     <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x77b348701f80>,
     <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x77b34841a980>,
     <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x77b3487347c0>,
     <OpOverload(op='aten.full', overload='out')>: <function full at 0x77b34854cfe0>,
     <OpOverload(op='aten.full', overload='default')>: <function full at 0x77b34854cfe0>,
     <OpOverload(op='aten.ldexp', overload='Tensor')>: <function ldexp at 0x77b348908860>,
     <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x77b3488ebb00>,
     <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x77b34857f380>,
     <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b28c64eb60>, kernel=<OpOverload(op='aten.svd', overload='default')>),
     <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1120>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x77b3487956c0>,
     <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x77b3487956c0>,
     <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x77b3486a77e0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x77b3486a77e0>,
     <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x77b3486c8540>,
     <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x77b3488eb740>,
     <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x77b3488eb740>,
     <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x77b3488ebb00>,
     <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x77b3488eb6a0>,
     <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x77b3488eb6a0>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x77b3488eb920>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x77b3488eb920>,
     <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x77b3489082c0>,
     <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x77b3489082c0>,
     <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x77b348908cc0>,
     <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x77b348908cc0>,
     <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x77b348908fe0>,
     <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x77b348908fe0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x77b34890a8e0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x77b34890a8e0>,
     <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x77b348909080>,
     <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x77b348909080>,
     <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x77b3489093a0>,
     <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x77b3489093a0>,
     <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x77b348909760>,
     <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x77b348909760>,
     <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x77b348909f80>,
     <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x77b348909f80>,
     <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.round', overload='default')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x77b348694180>,
     <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x77b348695080>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x77b348695080>,
     <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x77b3485242c0>,
     <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x77b348695580>,
     <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x77b348695580>,
     <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x77b348695a80>,
     <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x77b348695a80>,
     <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x77b348683920>,
     <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x77b348683920>,
     <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x77b34854d6c0>,
     <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x77b348694720>,
     <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x77b348694720>,
     <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x77b348696c00>,
     <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x77b348696c00>,
     <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x77b348697100>,
     <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x77b348697100>,
     <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x77b348697600>,
     <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x77b348697600>,
     <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x77b3486ca520>,
     <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.add', overload='out')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x77b348697f60>,
     <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x77b34875c5e0>,
     <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x77b348697f60>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x77b3486c89a0>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.div', overload='out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x77b3486a6980>,
     <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x77b3486a6980>,
     <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x77b3486a6de0>,
     <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x77b3486a6de0>,
     <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x77b34890b420>,
     <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x77b3486a6a20>,
     <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x77b3486a6a20>,
     <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x77b3485267a0>,
     <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x77b3486c89a0>,
     <OpOverload(op='aten.where', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894887c0>, kernel=<OpOverload(op='aten.where', overload='Scalar')>),
     <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x77b3486cb880>,
     <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x77b3486c8e00>,
     <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x77b3486c9260>,
     <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x77b3486c9260>,
     <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x77b348cf77e0>,
     <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x77b3486cb880>,
     <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x77b3486cbce0>,
     <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x77b3486cbce0>,
     <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x77b348908720>,
     <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x77b3486e0a40>,
     <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x77b3486e0a40>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x77b348908720>,
     <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x77b348908220>,
     <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x77b3487f9bc0>,
     <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x77b3487f9bc0>,
     <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x77b3487fa200>,
     <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x77b3487fa200>,
     <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x77b3487fa700>,
     <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x77b3487fa700>,
     <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x77b34890b380>,
     <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x77b3487fac00>,
     <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x77b3487fac00>,
     <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x77b3487fb100>,
     <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x77b3487fb100>,
     <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x77b34866c400>,
     <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x77b34866c400>,
     <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x77b34866c900>,
     <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x77b34866c900>,
     <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x77b3486e3240>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x77b3487fb600>,
     <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x77b3487fb600>,
     <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x77b3487fbb00>,
     <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x77b3487fbb00>,
     <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x77b3487fbec0>,
     <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x77b3487fbec0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x77b3486e3240>,
     <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x77b34866ce00>,
     <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x77b34866ce00>,
     <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x77b348cf77e0>,
     <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x77b3487f9c60>,
     <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x77b348694180>,
     <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a76a0>, kernel=<OpOverload(op='aten.any', overload='dimname')>),
     <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x77b34866d580>,
     <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x77b34866d580>,
     <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x77b34866da80>,
     <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x77b34866da80>,
     <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x77b34866df80>,
     <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x77b34866df80>,
     <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x77b34866e480>,
     <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x77b34866e480>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x77b348737c40>,
     <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x77b348908180>,
     <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x77b34866efc0>,
     <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x77b34866efc0>,
     <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x77b3485aaac0>,
     <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x77b34875dda0>,
     <OpOverload(op='aten.isfinite', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b291042fc0>, kernel=<OpOverload(op='aten.isfinite', overload='default')>),
     <OpOverload(op='aten.log', overload='out')>: <function log at 0x77b348681bc0>,
     <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x77b3486816c0>,
     <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x77b3486816c0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b29100a160>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>),
     <OpOverload(op='aten.log', overload='default')>: <function log at 0x77b348681bc0>,
     <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x77b348682ac0>,
     <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x77b3486820c0>,
     <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x77b3486820c0>,
     <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x77b3486825c0>,
     <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x77b3486825c0>,
     <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x77b348682ac0>,
     <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x77b348683880>,
     <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x77b348683880>,
     <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x77b348683240>,
     <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x77b348683240>,
     <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x77b3487f9c60>,
     <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x77b3487d2e80>,
     <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x77b3487d2e80>,
     <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x77b3487f8b80>,
     <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x77b3487f8b80>,
     <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894619e0>, kernel=<OpOverload(op='aten.linear', overload='default')>),
     <OpOverload(op='aten.conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b86660>, kernel=<OpOverload(op='aten.conv2d', overload='default')>),
     <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x77b348795e40>,
     <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x77b34890a020>,
     <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x77b348909b20>,
     <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x77b34875cea0>,
     <OpOverload(op='aten.round', overload='out')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x77b3487a8f40>,
     <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x77b348524220>,
     <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x77b34857d800>,
     <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x77b3487d1ee0>,
     <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x77b348795620>,
     <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x77b3487019e0>,
     <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x77b348527060>,
     <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x77b34854d6c0>,
     <OpOverload(op='aten.slice', overload='Tensor')>: <function slice_forward at 0x77b348735120>,
     <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten.view', overload='default')>: <function view at 0x77b348527100>,
     <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x77b34854c540>,
     <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x77b34854d9e0>,
     <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x77b34854d9e0>,
     <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x77b34854c0e0>,
     <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x77b34854c0e0>,
     <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x77b348527ce0>,
     <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x77b348527ce0>,
     <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x77b348527f60>,
     <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x77b348527f60>,
     <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x77b34854c180>,
     <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x77b34854c180>,
     <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x77b348526980>,
     <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x77b348526980>,
     <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x77b348525940>,
     <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x77b348525940>,
     <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b85d00>, kernel=<OpOverload(op='aten.item', overload='default')>),
     <OpOverload(op='aten.nonzero_numpy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284b849a0>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>),
     <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x77b3487a89a0>,
     <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x77b348735760>,
     <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x77b348735760>,
     <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x77b348525da0>,
     <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1da0>,
     <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x77b3486c8e00>,
     <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x77b34854c540>,
     <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba27a0>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba22a0>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>),
     <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1580>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0a40>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba05e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>),
     <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1300>, kernel=<OpOverload(op='quantized.conv3d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1120>, kernel=<OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1260>, kernel=<OpOverload(op='quantized.conv3d_transpose', overload='default')>),
     <OpOverload(op='quantized.conv3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1b20>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>),
     <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1f80>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a44a0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>),
     <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0220>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba07c0>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1e40>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>),
     <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2840>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4540>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba16c0>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>),
     <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6c00>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4720>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>),
     <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5d00>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6980>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a53a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6840>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6b60>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>),
     <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba25c0>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2520>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>),
     <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1da0>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba1d00>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>),
     <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0900>, kernel=<OpOverload(op='quantized.linear_unpack_fp16', overload='default')>),
     <OpOverload(op='quantized.conv3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba09a0>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba0f40>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a6520>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba04a0>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a5bc0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2894a4860>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>),
     <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284ba2160>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>)}

experimental_experiment.torch_dynamo.get_decomposition_table_onnxscript()

Returns the decomposition table used by torch.onnx.export().

The value is:

<<<

import pprint
from experimental_experiment.torch_dynamo import get_decomposition_table_onnxscript

pprint.pprint(get_decomposition_table_onnxscript())

>>>

    {<OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223600>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>),
     <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83240>, kernel=<OpOverload(op='image.write_file', overload='default')>),
     <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6dda0>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>),
     <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d1c0>, kernel=<OpOverload(op='aten.orgqr', overload='default')>),
     <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d260>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>),
     <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6cf40>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>),
     <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c900>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>),
     <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62f20>, kernel=<OpOverload(op='aten.arctanh', overload='default')>),
     <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f634c0>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>),
     <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63740>, kernel=<OpOverload(op='aten.align_as', overload='default')>),
     <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f837e0>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>),
     <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75440>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>),
     <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81440>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>),
     <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62020>, kernel=<OpOverload(op='aten.dstack', overload='default')>),
     <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f777e0>, kernel=<OpOverload(op='aten.max', overload='names_dim')>),
     <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f747c0>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>),
     <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81260>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>),
     <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f767a0>, kernel=<OpOverload(op='aten.special_erfinv', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f814e0>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>),
     <OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82ca0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>),
     <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62d40>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>),
     <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60f40>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62a20>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>),
     <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6dee0>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>),
     <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d300>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>),
     <OpOverload(op='aten.linalg_inv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80d60>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>),
     <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61c60>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>),
     <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f90180>, kernel=<OpOverload(op='image.decode_png', overload='default')>),
     <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f90360>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>),
     <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74360>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>),
     <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63240>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>),
     <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f637e0>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>),
     <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f772e0>, kernel=<OpOverload(op='aten.size', overload='Dimname')>),
     <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c680>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>),
     <OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74540>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f758a0>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77d80>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>),
     <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75f80>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>),
     <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61080>, kernel=<OpOverload(op='aten.inverse', overload='default')>),
     <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e0c0>, kernel=<OpOverload(op='aten.argsort', overload='stable')>),
     <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80540>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>),
     <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f834c0>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>),
     <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e480>, kernel=<OpOverload(op='aten.arctan2', overload='default')>),
     <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e020>, kernel=<OpOverload(op='aten.negative', overload='default')>),
     <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75620>, kernel=<OpOverload(op='aten.can_cast', overload='default')>),
     <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83ce0>, kernel=<OpOverload(op='image.read_file', overload='default')>),
     <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f744a0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>),
     <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74720>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>),
     <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76520>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>),
     <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83ba0>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>),
     <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77920>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>),
     <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75bc0>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>),
     <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77f60>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>),
     <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e2a0>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>),
     <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c360>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>),
     <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61120>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>),
     <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f805e0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>),
     <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f616c0>, kernel=<OpOverload(op='aten.concatenate', overload='names')>),
     <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75080>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>),
     <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61260>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>),
     <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75ee0>, kernel=<OpOverload(op='aten.smm', overload='default')>),
     <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62160>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>),
     <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61580>, kernel=<OpOverload(op='aten.combinations', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f9c0>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>),
     <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60b80>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>),
     <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77a60>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>),
     <OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e980>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>),
     <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76de0>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>),
     <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f7e0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>),
     <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77c40>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>),
     <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77ce0>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f800e0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>),
     <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80a40>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>),
     <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81080>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>),
     <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75c60>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>),
     <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76160>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>),
     <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6eb60>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>),
     <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74180>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>),
     <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f380>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>),
     <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e700>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>),
     <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d080>, kernel=<OpOverload(op='aten.trapz', overload='dx')>),
     <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60cc0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>),
     <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60e00>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>),
     <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f880>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61620>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>),
     <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61940>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>),
     <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c7c0>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>),
     <OpOverload(op='aten.affine_grid_generator_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f740>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>),
     <OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f754e0>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76700>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>),
     <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6db20>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>),
     <OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76c00>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d260>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>),
     <OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f90400>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>),
     <OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82340>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>),
     <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ef20>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>),
     <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5cea0>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>),
     <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d300>, kernel=<OpOverload(op='aten.special_logit', overload='default')>),
     <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d1c0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77560>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>),
     <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f4c0>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>),
     <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ccc0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>),
     <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6cfe0>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d9e0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>),
     <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d800>, kernel=<OpOverload(op='aten.is_neg', overload='default')>),
     <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c860>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>),
     <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63ce0>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>),
     <OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63c40>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>),
     <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61300>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>),
     <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76020>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>),
     <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81760>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>),
     <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f920>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>),
     <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75120>, kernel=<OpOverload(op='aten.column_stack', overload='default')>),
     <OpOverload(op='aten._debug_has_internal_overlap', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60ea0>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>),
     <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82f20>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>),
     <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80900>, kernel=<OpOverload(op='aten.flipud', overload='default')>),
     <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f816c0>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>),
     <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6cae0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>),
     <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75da0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f560>, kernel=<OpOverload(op='aten.diff', overload='default')>),
     <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f742c0>, kernel=<OpOverload(op='aten.argsort', overload='default')>),
     <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75b20>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6fd80>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>),
     <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82160>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>),
     <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d4e0>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>),
     <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6cc20>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>),
     <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f622a0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>),
     <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f771a0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>),
     <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f90040>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>),
     <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82c00>, kernel=<OpOverload(op='c10d_functional.broadcast', overload='default')>),
     <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60d60>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>),
     <OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6fe20>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>),
     <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c9a0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>),
     <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f604a0>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>),
     <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e3e0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>),
     <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d760>, kernel=<OpOverload(op='aten.fix', overload='default')>),
     <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60860>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>),
     <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74f40>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60680>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>),
     <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63920>, kernel=<OpOverload(op='aten.special_i0', overload='default')>),
     <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74a40>, kernel=<OpOverload(op='aten.pinverse', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d6c0>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>),
     <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83420>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>),
     <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d9e0>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>),
     <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2842231a0>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>),
     <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60900>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>),
     <OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f804a0>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>),
     <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60360>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>),
     <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60ae0>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>),
     <OpOverload(op='image._is_compiled_against_turbo', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81ee0>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>),
     <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c680>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>),
     <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284222e80>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f1a0>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>),
     <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61440>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>),
     <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f1a0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>),
     <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6de40>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75940>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>),
     <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6fec0>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62840>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f602c0>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>),
     <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c400>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>),
     <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81e40>, kernel=<OpOverload(op='_test.cat', overload='default')>),
     <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c7c0>, kernel=<OpOverload(op='aten.min', overload='names_dim')>),
     <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6dd00>, kernel=<OpOverload(op='aten.row_stack', overload='default')>),
     <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76fc0>, kernel=<OpOverload(op='aten.is_conj', overload='default')>),
     <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6da80>, kernel=<OpOverload(op='aten.is_inference', overload='default')>),
     <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d760>, kernel=<OpOverload(op='aten.argwhere', overload='default')>),
     <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f618a0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f240>, kernel=<OpOverload(op='aten.cov', overload='default')>),
     <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223380>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5da80>, kernel=<OpOverload(op='aten.to_dense', overload='default')>),
     <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60540>, kernel=<OpOverload(op='aten.cdist', overload='default')>),
     <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2842218a0>, kernel=<OpOverload(op='aten.diagflat', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223ec0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>),
     <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f100>, kernel=<OpOverload(op='aten.stride', overload='int')>),
     <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c180>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>),
     <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223ce0>, kernel=<OpOverload(op='aten.kron', overload='default')>),
     <OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62200>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>),
     <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c2c0>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>),
     <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81c60>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>),
     <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5dbc0>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>),
     <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f836a0>, kernel=<OpOverload(op='image.decode_image', overload='default')>),
     <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d440>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>),
     <OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76660>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>),
     <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74680>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f900e0>, kernel=<OpOverload(op='image.encode_png', overload='default')>),
     <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76ca0>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>),
     <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f776a0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>),
     <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74fe0>, kernel=<OpOverload(op='aten.output_nr', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81620>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>),
     <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74ea0>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>),
     <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c180>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>),
     <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76d40>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>),
     <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61da0>, kernel=<OpOverload(op='aten.sort', overload='dimname')>),
     <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75e40>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80b80>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>),
     <OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77e20>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>),
     <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ed40>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>),
     <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74400>, kernel=<OpOverload(op='aten.max', overload='other')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e200>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c720>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>),
     <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f740e0>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>),
     <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f762a0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>),
     <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6dbc0>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>),
     <OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75800>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ce00>, kernel=<OpOverload(op='aten.one_hot', overload='default')>),
     <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80680>, kernel=<OpOverload(op='aten.median', overload='names_dim')>),
     <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e520>, kernel=<OpOverload(op='aten.gather', overload='dimname')>),
     <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f768e0>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>),
     <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75300>, kernel=<OpOverload(op='aten.arcsin', overload='default')>),
     <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c4a0>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>),
     <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76480>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>),
     <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80400>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>),
     <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76ac0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>),
     <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75760>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63ba0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>),
     <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f611c0>, kernel=<OpOverload(op='aten.qr', overload='default')>),
     <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63420>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>),
     <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61e40>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>),
     <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76340>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>),
     <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f614e0>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>),
     <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75a80>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>),
     <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61f80>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76a20>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>),
     <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61760>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>),
     <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f818a0>, kernel=<OpOverload(op='aten.quantile', overload='default')>),
     <OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f813a0>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77240>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>),
     <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6cea0>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>),
     <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f620c0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>),
     <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e840>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>),
     <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e8e0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>),
     <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fd80>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>),
     <OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63e20>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>),
     <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fec0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>),
     <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77600>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>),
     <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f631a0>, kernel=<OpOverload(op='aten.min', overload='other')>),
     <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74ae0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f807c0>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f636a0>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>),
     <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2842232e0>, kernel=<OpOverload(op='aten.special_round', overload='default')>),
     <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fce0>, kernel=<OpOverload(op='aten.vstack', overload='default')>),
     <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60040>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>),
     <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ede0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>),
     <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62e80>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>),
     <OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f607c0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>),
     <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f627a0>, kernel=<OpOverload(op='aten.gradient', overload='array')>),
     <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60400>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>),
     <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83d80>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>),
     <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ff60>, kernel=<OpOverload(op='aten.slogdet', overload='default')>),
     <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fb00>, kernel=<OpOverload(op='aten.trapz', overload='x')>),
     <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d940>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>),
     <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76840>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>),
     <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5eca0>, kernel=<OpOverload(op='aten.nanmean', overload='default')>),
     <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62700>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284222ac0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>),
     <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74b80>, kernel=<OpOverload(op='aten.size', overload='int')>),
     <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f811c0>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>),
     <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d440>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>),
     <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d3a0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>),
     <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61d00>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>),
     <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60220>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d620>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>),
     <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5dc60>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>),
     <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d4e0>, kernel=<OpOverload(op='aten._backward', overload='default')>),
     <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2842239c0>, kernel=<OpOverload(op='aten.msort', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f060>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>),
     <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82020>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>),
     <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223f60>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>),
     <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284221800>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>),
     <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223a60>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>),
     <OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f600e0>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>),
     <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f880>, kernel=<OpOverload(op='aten.is_leaf', overload='default')>),
     <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77740>, kernel=<OpOverload(op='aten.ger', overload='default')>),
     <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82b60>, kernel=<OpOverload(op='image.decode_gif', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f779c0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>),
     <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5cd60>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>),
     <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f380>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>),
     <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e660>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>),
     <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80360>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>),
     <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80720>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>),
     <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ef20>, kernel=<OpOverload(op='aten.arccosh', overload='default')>),
     <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2842234c0>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>),
     <OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60180>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223880>, kernel=<OpOverload(op='aten.chalf', overload='default')>),
     <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f809a0>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>),
     <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76200>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ce00>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>),
     <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ff60>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>),
     <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77ec0>, kernel=<OpOverload(op='aten.data', overload='default')>),
     <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ea20>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>),
     <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c5e0>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>),
     <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e2a0>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>),
     <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81800>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>),
     <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f753a0>, kernel=<OpOverload(op='aten.arccos', overload='default')>),
     <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81bc0>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>),
     <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d620>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>),
     <OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c220>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>),
     <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81120>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>),
     <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f820c0>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>),
     <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60c20>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>),
     <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f740>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f060>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>),
     <OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62ca0>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>),
     <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f7e0>, kernel=<OpOverload(op='aten.rrelu', overload='default')>),
     <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6eca0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>),
     <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d3a0>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>),
     <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223060>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>),
     <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63d80>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>),
     <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77b00>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>),
     <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5cae0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>),
     <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223240>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>),
     <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c0e0>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>),
     <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5dee0>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>),
     <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5de40>, kernel=<OpOverload(op='aten.hstack', overload='default')>),
     <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e7a0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f420>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>),
     <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f100>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>),
     <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ca40>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>),
     <OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5efc0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f609a0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>),
     <OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77420>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60a40>, kernel=<OpOverload(op='aten.fliplr', overload='default')>),
     <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74040>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>),
     <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e020>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>),
     <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fc40>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>),
     <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83ec0>, kernel=<OpOverload(op='image.decode_webp', overload='default')>),
     <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77060>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>),
     <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ee80>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>),
     <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284222c00>, kernel=<OpOverload(op='aten.outer', overload='default')>),
     <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d120>, kernel=<OpOverload(op='aten.is_signed', overload='default')>),
     <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5cf40>, kernel=<OpOverload(op='aten.vander', overload='default')>),
     <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223b00>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>),
     <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82200>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>),
     <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75260>, kernel=<OpOverload(op='aten.concat', overload='names')>),
     <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77880>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223420>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>),
     <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c540>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>),
     <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5db20>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>),
     <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fe20>, kernel=<OpOverload(op='aten._version', overload='default')>),
     <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c860>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d8a0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>),
     <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c9a0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>),
     <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f759e0>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>),
     <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63b00>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='other_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f763e0>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80ea0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>),
     <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81f80>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223d80>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>),
     <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81300>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>),
     <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d580>, kernel=<OpOverload(op='aten.inner', overload='default')>),
     <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6cd60>, kernel=<OpOverload(op='aten.kl_div', overload='default')>),
     <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c360>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>),
     <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74d60>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>),
     <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f619e0>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>),
     <OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82d40>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>),
     <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62660>, kernel=<OpOverload(op='aten.less', overload='Scalar')>),
     <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83880>, kernel=<OpOverload(op='_test.get_first', overload='default')>),
     <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f745e0>, kernel=<OpOverload(op='aten._remove_batch_dim', overload='default')>),
     <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f6a0>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>),
     <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82de0>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6fba0>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>),
     <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f605e0>, kernel=<OpOverload(op='aten.isreal', overload='default')>),
     <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ee80>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>),
     <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60720>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>),
     <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f560>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>),
     <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d580>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>),
     <OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5eb60>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5ccc0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>),
     <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f9c0>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>),
     <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62480>, kernel=<OpOverload(op='aten.absolute', overload='default')>),
     <OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83600>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>),
     <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62980>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>),
     <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74860>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223920>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>),
     <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c900>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>),
     <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f832e0>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>),
     <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f628e0>, kernel=<OpOverload(op='aten.arctan', overload='default')>),
     <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e480>, kernel=<OpOverload(op='aten.special_psi', overload='default')>),
     <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5cfe0>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>),
     <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e980>, kernel=<OpOverload(op='aten.square', overload='default')>),
     <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e660>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>),
     <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e340>, kernel=<OpOverload(op='aten.clip', overload='default')>),
     <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5df80>, kernel=<OpOverload(op='aten.diag', overload='default')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e700>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e0c0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>),
     <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x77b3484187c0>,
     <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x77b348419bc0>,
     <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x77b348419e40>,
     <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x77b34841a520>,
     <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x77b34841afc0>,
     <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x77b34841afc0>,
     <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x77b34841ad40>,
     <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x77b34841b2e0>,
     <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x77b348448400>,
     <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x77b348448720>,
     <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x77b348448400>,
     <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x77b348448720>,
     <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63060>, kernel=<OpOverload(op='aten.margin_ranking_loss', overload='default')>),
     <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63560>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>),
     <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x77b348449760>,
     <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x77b348449b20>,
     <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x77b348449b20>,
     <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x77b34841b240>,
     <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x77b34841b240>,
     <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x77b348449da0>,
     <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x77b34844a660>,
     <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x77b34844bec0>,
     <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x77b34844b1a0>,
     <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f2e0>, kernel=<OpOverload(op='aten.pairwise_distance', overload='default')>),
     <OpOverload(op='aten.pdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6fc40>, kernel=<OpOverload(op='aten.pdist', overload='default')>),
     <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x77b34844ba60>,
     <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x77b34844bce0>,
     <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x77b34844a7a0>,
     <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x77b34844bb00>,
     <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x77b34844bd80>,
     <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x77b34844bf60>,
     <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x77b348468220>,
     <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x77b348468220>,
     <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x77b348468680>,
     <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x77b348468680>,
     <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x77b348468a40>,
     <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x77b348468a40>,
     <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x77b348468e00>,
     <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x77b34846a520>,
     <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x77b3484693a0>,
     <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x77b3484693a0>,
     <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x77b3484698a0>,
     <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x77b3484698a0>,
     <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x77b348469da0>,
     <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x77b348469da0>,
     <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x77b34846a160>,
     <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x77b34846a160>,
     <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x77b348469940>,
     <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x77b34846a660>,
     <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x77b34846a660>,
     <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f60fe0>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>),
     <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x77b34846aa20>,
     <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x77b34846ade0>,
     <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x77b34846ade0>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x77b34846b420>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x77b34846b420>,
     <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x77b34846b880>,
     <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x77b348524ea0>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5dda0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>),
     <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5dd00>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>),
     <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x77b3485254e0>,
     <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x77b348525300>,
     <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x77b348525da0>,
     <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x77b348524220>,
     <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x77b348526160>,
     <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x77b348526160>,
     <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x77b3485263e0>,
     <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x77b3485263e0>,
     <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x77b348526d40>,
     <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x77b348526d40>,
     <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f6a0>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>),
     <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x77b348526de0>,
     <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5fa60>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>),
     <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x77b348526e80>,
     <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x77b34854c7c0>,
     <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x77b34854cb80>,
     <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x77b34854ce00>,
     <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x77b34854d080>,
     <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x77b3485271a0>,
     <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c5e0>, kernel=<OpOverload(op='aten.meshgrid', overload='default')>),
     <OpOverload(op='aten.meshgrid', overload='indexing')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5c720>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>),
     <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x77b34854de40>,
     <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x77b34854de40>,
     <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d6c0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>),
     <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x77b34854e520>,
     <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x77b34854e7a0>,
     <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x77b34854e7a0>,
     <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x77b34854f2e0>,
     <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x77b34854f2e0>,
     <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x77b34854eca0>,
     <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x77b34854ef20>,
     <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x77b34854e700>,
     <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x77b34854e700>,
     <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x77b34854df80>,
     <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x77b34854f560>,
     <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x77b34854f560>,
     <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x77b34854f920>,
     <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x77b34854f920>,
     <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x77b34854fce0>,
     <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x77b34854fce0>,
     <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x77b34857c0e0>,
     <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x77b34857c0e0>,
     <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x77b34857c180>,
     <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x77b34857cae0>,
     <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x77b34857cfe0>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x77b34857d260>,
     <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x77b34857cb80>,
     <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x77b34857d580>,
     <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x77b34857d580>,
     <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x77b34857d800>,
     <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x77b34857ef20>,
     <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x77b34854f880>,
     <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x77b34857e5c0>,
     <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x77b34857d620>,
     <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x77b34857d8a0>,
     <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x77b34857e840>,
     <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x77b34857d9e0>,
     <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x77b34857d9e0>,
     <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x77b34857db20>,
     <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x77b34857df80>,
     <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x77b34857dc60>,
     <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x77b34857eac0>,
     <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x77b34857eac0>,
     <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x77b34857dda0>,
     <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x77b34857dee0>,
     <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x77b34857ed40>,
     <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x77b34857e020>,
     <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x77b34857e160>,
     <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x77b34857ed40>,
     <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x77b34857e2a0>,
     <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x77b34857e3e0>,
     <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x77b34857e3e0>,
     <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x77b34857efc0>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x77b34857e520>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x77b34857e520>,
     <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x77b3485a9da0>,
     <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x77b34857e660>,
     <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x77b34857e480>,
     <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x77b34857e7a0>,
     <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x77b34857e7a0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x77b34857e8e0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x77b34857e8e0>,
     <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x77b34857f240>,
     <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x77b34857ea20>,
     <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x77b34857ea20>,
     <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x77b34857eb60>,
     <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x77b34857eca0>,
     <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x77b34857eca0>,
     <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x77b34857f1a0>,
     <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x77b34857f060>,
     <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x77b34857ede0>,
     <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x77b34857ede0>,
     <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x77b34857f1a0>,
     <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x77b34857ef20>,
     <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x77b34857e980>,
     <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x77b34857da80>,
     <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x77b34857e340>,
     <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x77b34857e0c0>,
     <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x77b34857e0c0>,
     <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x77b34857de40>,
     <OpOverload(op='aten.log_', overload='default')>: <function log at 0x77b34857d080>,
     <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x77b34857dbc0>,
     <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x77b34857d940>,
     <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x77b34857f4c0>,
     <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x77b34857d4e0>,
     <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x77b34857f2e0>,
     <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x77b34857f100>,
     <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x77b34857f420>,
     <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x77b34857f740>,
     <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x77b34857f560>,
     <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x77b34857f560>,
     <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x77b34857f9c0>,
     <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x77b34857f6a0>,
     <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x77b34857f7e0>,
     <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x77b34857f7e0>,
     <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x77b34857f920>,
     <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x77b34857f920>,
     <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x77b34857fc40>,
     <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x77b34857fa60>,
     <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x77b34857fba0>,
     <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x77b34857fce0>,
     <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x77b34857fce0>,
     <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x77b34857fd80>,
     <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x77b34857fe20>,
     <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x77b34857fe20>,
     <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x77b34857e700>,
     <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x77b34857fd80>,
     <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x77b34857ff60>,
     <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x77b3485aade0>,
     <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x77b3485a80e0>,
     <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x77b3485a8220>,
     <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x77b3485a8360>,
     <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x77b34857e200>,
     <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x77b3485a84a0>,
     <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x77b34857e200>,
     <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x77b34857fec0>,
     <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x77b34857fec0>,
     <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x77b3485a85e0>,
     <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x77b34857ee80>,
     <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x77b34857ee80>,
     <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x77b3485a8720>,
     <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x77b3485a8720>,
     <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x77b3485a8900>,
     <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x77b3485a8680>,
     <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x77b3485a8400>,
     <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x77b3485a8400>,
     <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x77b3485a9800>,
     <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x77b3485a8180>,
     <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x77b3485a89a0>,
     <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x77b3485a8ae0>,
     <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x77b3485a8ae0>,
     <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x77b3485a9a80>,
     <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x77b3485a8c20>,
     <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x77b3485a9d00>,
     <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x77b3485a8d60>,
     <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x77b3485a8ea0>,
     <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x77b3485a8ea0>,
     <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x77b3485a8fe0>,
     <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x77b3485a9120>,
     <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x77b3485a9f80>,
     <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x77b3485a9260>,
     <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x77b3485a93a0>,
     <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x77b34857fb00>,
     <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x77b3485a94e0>,
     <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x77b3485a9620>,
     <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x77b3485a9760>,
     <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x77b3485a98a0>,
     <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x77b34857f600>,
     <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x77b34857f600>,
     <OpOverload(op='aten.square_', overload='default')>: <function square at 0x77b3485a99e0>,
     <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x77b3485a9b20>,
     <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x77b3485a9b20>,
     <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x77b34857d760>,
     <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x77b3485a9c60>,
     <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x77b3485a9300>,
     <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x77b3485a9300>,
     <OpOverload(op='aten.as_strided_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x77b3485a8cc0>,
     <OpOverload(op='aten.as_strided_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x77b3485a8cc0>,
     <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x77b3485a8540>,
     <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x77b3485aa160>,
     <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x77b3485aa160>,
     <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x77b3485aa480>,
     <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x77b3485aa480>,
     <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x77b3485aa7a0>,
     <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x77b3485aaac0>,
     <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x77b3485aade0>,
     <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x77b3485ab100>,
     <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x77b3485ab100>,
     <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x77b3485aa840>,
     <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x77b3485aa840>,
     <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x77b3485a91c0>,
     <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x77b3485e04a0>,
     <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x77b3485e0720>,
     <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x77b3485a9ee0>,
     <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62520>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>),
     <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x77b3485e11c0>,
     <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62de0>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>),
     <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63600>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>),
     <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x77b3485e0680>,
     <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63a60>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>),
     <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x77b3485e14e0>,
     <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c220>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>),
     <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x77b3485e1760>,
     <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6c4a0>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>),
     <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x77b3485e19e0>,
     <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ca40>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>),
     <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x77b3485e22a0>,
     <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d120>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>),
     <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x77b3485e2520>,
     <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d940>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>),
     <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x77b3485e27a0>,
     <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x77b3485e2980>,
     <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e200>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>),
     <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x77b3485e2a20>,
     <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6efc0>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>),
     <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e160>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>),
     <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x77b3485e3100>,
     <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6ede0>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>),
     <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x77b3485e3380>,
     <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6f600>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>),
     <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x77b3485e1940>,
     <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6fb00>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>),
     <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x77b3485e3420>,
     <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74220>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>),
     <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x77b3485e36a0>,
     <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f74900>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>),
     <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x77b3485e3920>,
     <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f751c0>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>),
     <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x77b3485e3ba0>,
     <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f756c0>, kernel=<OpOverload(op='aten.fft_fftshift', overload='default')>),
     <OpOverload(op='aten.fft_ifftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f75d00>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>),
     <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x77b348418400>,
     <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5cb80>, kernel=<OpOverload(op='aten.is_complex', overload='default')>),
     <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x77b34866d080>,
     <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x77b34866d080>,
     <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x77b34857f380>,
     <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x77b34866eac0>,
     <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x77b34866eac0>,
     <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x77b34866f4c0>,
     <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x77b34866fec0>,
     <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x77b34866db20>,
     <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x77b3486802c0>,
     <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x77b3486807c0>,
     <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x77b3486811c0>,
     <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x77b3486811c0>,
     <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x77b348683060>,
     <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e520>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>),
     <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x77b348682b60>,
     <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x77b348682b60>,
     <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x77b348694680>,
     <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x77b348694b80>,
     <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x77b348694b80>,
     <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x77b348695f80>,
     <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x77b34841bce0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x77b3486a4860>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x77b3486a5120>,
     <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x77b3486a5300>,
     <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x77b3486ca980>,
     <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x77b3486c8540>,
     <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x77b3486c9800>,
     <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x77b3486c9800>,
     <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x77b3486ca0c0>,
     <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x77b3486c98a0>,
     <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x77b3486cab60>,
     <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x77b3486cafc0>,
     <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x77b3486e1580>,
     <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x77b3486e1d00>,
     <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x77b3486e20c0>,
     <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x77b3486e2480>,
     <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x77b3486e2700>,
     <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x77b3486e2980>,
     <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.all', overload='out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x77b3486e3a60>,
     <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223100>, kernel=<OpOverload(op='aten.std_mean', overload='default')>),
     <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284222fc0>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>),
     <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223e20>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>),
     <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x77b3486e39c0>,
     <OpOverload(op='aten.any', overload='out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223740>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>),
     <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f76f20>, kernel=<OpOverload(op='aten.std', overload='default')>),
     <OpOverload(op='aten.std', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80180>, kernel=<OpOverload(op='aten.std', overload='dim')>),
     <OpOverload(op='aten.std', overload='out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80c20>, kernel=<OpOverload(op='aten.std', overload='names_dim')>),
     <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f80e00>, kernel=<OpOverload(op='aten.std', overload='correction_names')>),
     <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x77b348700860>,
     <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81580>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>),
     <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x77b348700680>,
     <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x77b348701b20>,
     <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223ba0>, kernel=<OpOverload(op='aten.var_mean', overload='default')>),
     <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284222f20>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>),
     <OpOverload(op='aten.var_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b284223560>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>),
     <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b2842237e0>, kernel=<OpOverload(op='aten.var_mean', overload='correction_names')>),
     <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x77b348701440>,
     <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x77b348702520>,
     <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x77b348702d40>,
     <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x77b348703740>,
     <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x77b348703880>,
     <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x77b3485244a0>,
     <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x77b348701080>,
     <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x77b348701080>,
     <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x77b348703c40>,
     <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x77b348524540>,
     <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x77b348524540>,
     <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x77b348525b20>,
     <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5e5c0>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>),
     <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x77b348524860>,
     <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x77b34890a020>,
     <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x77b34890a3e0>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x77b34890a480>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x77b34890a480>,
     <OpOverload(op='aten.ldexp', overload='out')>: <function ldexp at 0x77b348908860>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x77b34890b4c0>,
     <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x77b34890aca0>,
     <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x77b34890aca0>,
     <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x77b34890afc0>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x77b34890b6a0>,
     <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x77b34890b420>,
     <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x77b34890b880>,
     <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x77b34890b380>,
     <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x77b34890ba60>,
     <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x77b34890bce0>,
     <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x77b34890bce0>,
     <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x77b3487340e0>,
     <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x77b3487340e0>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x77b348734400>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x77b348734400>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x77b34890b100>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x77b34890b100>,
     <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x77b348736840>,
     <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x77b3487347c0>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x77b3487363e0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x77b348734860>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x77b348734860>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x77b3487363e0>,
     <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x77b348734e00>,
     <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x77b348734e00>,
     <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x77b348735080>,
     <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x77b348735080>,
     <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x77b348735300>,
     <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x77b348735300>,
     <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x77b348735bc0>,
     <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x77b3487359e0>,
     <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x77b3487359e0>,
     <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x77b348735c60>,
     <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x77b348735da0>,
     <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x77b348735da0>,
     <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x77b348736980>,
     <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x77b348734d60>,
     <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x77b348736520>,
     <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x77b348736520>,
     <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x77b3487367a0>,
     <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x77b3487367a0>,
     <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x77b348737240>,
     <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x77b348737420>,
     <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x77b348737100>,
     <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x77b3487374c0>,
     <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x77b348737740>,
     <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x77b3487379c0>,
     <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x77b348737f60>,
     <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x77b348737f60>,
     <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x77b348737c40>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x77b348737ba0>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x77b348737ba0>,
     <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x77b34875c2c0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x77b34875c9a0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x77b34875ca40>,
     <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x77b34875c5e0>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x77b34875ccc0>,
     <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x77b34875c900>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x77b34875cd60>,
     <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x77b34875cfe0>,
     <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x77b34875d800>,
     <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x77b34875e160>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x77b34875ede0>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x77b34875ede0>,
     <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x77b34875cc20>,
     <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x77b34875cc20>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x77b34875eca0>,
     <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x77b34875ee80>,
     <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x77b34875e3e0>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x77b34875ef20>,
     <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x77b34875f060>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x77b34875eca0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x77b34875f7e0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x77b34875f7e0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x77b34875fec0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x77b34875fec0>,
     <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x77b3487376a0>,
     <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x77b3487376a0>,
     <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x77b348794720>,
     <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x77b348794220>,
     <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x77b348794220>,
     <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x77b348794040>,
     <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x77b3487945e0>,
     <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6d8a0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>),
     <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x77b348796a20>,
     <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x77b348794680>,
     <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x77b348794680>,
     <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f6e5c0>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>),
     <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x77b348794c20>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x77b348796700>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x77b348795300>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x77b348795300>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x77b348795620>,
     <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x77b3487954e0>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x77b348794540>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x77b348794540>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x77b34875efc0>,
     <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x77b3487963e0>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x77b348796700>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x77b348796d40>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x77b348796d40>,
     <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x77b348797560>,
     <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x77b348797740>,
     <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x77b348797920>,
     <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x77b348797b00>,
     <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x77b3487a8c20>,
     <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x77b3487a8180>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x77b3487a8400>,
     <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x77b3487a84a0>,
     <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x77b3487a8860>,
     <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x77b3487a8a40>,
     <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x77b3487a9120>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x77b348796980>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x77b3487a9440>,
     <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x77b3487a8f40>,
     <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x77b348795c60>,
     <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x77b3487a85e0>,
     <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x77b3487a80e0>,
     <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x77b3487a9b20>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x77b3487aa200>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x77b3487aa200>,
     <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x77b3487aaca0>,
     <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x77b3487aaca0>,
     <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x77b3487ab060>,
     <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x77b3487ab380>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x77b3487ab600>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x77b3487ab600>,
     <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x77b3487abb00>,
     <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x77b3487d00e0>,
     <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x77b3487abf60>,
     <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2a20>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x77b3487a82c0>,
     <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2a20>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x77b3487d04a0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d04a0>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x77b3487d0680>,
     <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x77b3487ab920>,
     <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x77b3487d0cc0>,
     <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x77b3487d0cc0>,
     <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x77b3487d1080>,
     <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x77b3487d1080>,
     <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x77b3487d00e0>,
     <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487abc40>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x77b3487a82c0>,
     <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x77b3487d20c0>,
     <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x77b3487d23e0>,
     <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x77b3487d23e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x77b3487d2840>,
     <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d27a0>,
     <OpOverload(op='aten.take', overload='default')>: <function take at 0x77b3487d2d40>,
     <OpOverload(op='aten.take', overload='out')>: <function take at 0x77b3487d2d40>,
     <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x77b3487d2b60>,
     <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x77b3487d2b60>,
     <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2ca0>,
     <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2de0>,
     <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3240>,
     <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2f20>,
     <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3060>,
     <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2fc0>,
     <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2480>,
     <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2480>,
     <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2200>,
     <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d2200>,
     <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1940>,
     <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1120>,
     <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3380>,
     <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3380>,
     <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d34c0>,
     <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d34c0>,
     <OpOverload(op='aten.ldexp_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3600>,
     <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3740>,
     <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3880>,
     <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d39c0>,
     <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3b00>,
     <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3c40>,
     <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3c40>,
     <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3d80>,
     <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d3ec0>,
     <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x77b3487f8040>,
     <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487f8180>,
     <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x77b3487013a0>,
     <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x77b3487019e0>,
     <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x77b3486e1120>,
     <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5eac0>, kernel=<OpOverload(op='aten.cat', overload='names')>),
     <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x77b34890ac00>,
     <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5f600>, kernel=<OpOverload(op='aten.where', overload='default')>),
     <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x77b3486e2fc0>,
     <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x77b3486e3b00>,
     <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61800>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>),
     <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x77b3487d2160>,
     <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f62c00>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>),
     <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x77b3486e3ec0>,
     <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f632e0>, kernel=<OpOverload(op='aten.var', overload='default')>),
     <OpOverload(op='aten.var', overload='correction')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f639c0>, kernel=<OpOverload(op='aten.var', overload='names_dim')>),
     <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63f60>, kernel=<OpOverload(op='aten.var', overload='dim')>),
     <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x77b348735c60>,
     <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f63ec0>, kernel=<OpOverload(op='aten.var', overload='correction_names')>),
     <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x77b3487005e0>,
     <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x77b3487000e0>,
     <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x77b3486e3f60>,
     <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x77b348701f80>,
     <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x77b34841a980>,
     <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x77b3487347c0>,
     <OpOverload(op='aten.full', overload='out')>: <function full at 0x77b34854cfe0>,
     <OpOverload(op='aten.ldexp', overload='Tensor')>: <function ldexp at 0x77b348908860>,
     <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x77b3488ebb00>,
     <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x77b34857f380>,
     <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77100>, kernel=<OpOverload(op='aten.svd', overload='default')>),
     <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1120>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x77b3487956c0>,
     <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x77b34857c4a0>,
     <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x77b3487956c0>,
     <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x77b3486a77e0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x77b34854e020>,
     <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x77b3486e3ce0>,
     <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x77b3486a77e0>,
     <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x77b3488eb740>,
     <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x77b3488eb740>,
     <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x77b3488ebb00>,
     <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x77b3488eb6a0>,
     <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x77b3488eb6a0>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x77b3488eb920>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x77b3488eb920>,
     <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x77b3489082c0>,
     <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x77b348908cc0>,
     <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x77b348908fe0>,
     <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x77b348908fe0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x77b34890a8e0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x77b34890a8e0>,
     <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x77b348909080>,
     <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x77b348909080>,
     <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x77b3489093a0>,
     <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x77b3489093a0>,
     <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x77b348909760>,
     <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x77b348909760>,
     <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x77b348909f80>,
     <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x77b348695080>,
     <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x77b3485242c0>,
     <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x77b348695580>,
     <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x77b348695a80>,
     <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x77b348683920>,
     <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x77b34854d6c0>,
     <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x77b348694720>,
     <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x77b348696c00>,
     <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x77b348697100>,
     <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x77b348697600>,
     <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x77b3486ca520>,
     <OpOverload(op='aten.add', overload='out')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x77b348697b00>,
     <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x77b34875c5e0>,
     <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x77b348697f60>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x77b3486a4400>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x77b3486c89a0>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x77b3486a4cc0>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x77b3486a5080>,
     <OpOverload(op='aten.div', overload='out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x77b3486a5580>,
     <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x77b3486a59e0>,
     <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x77b3486a6980>,
     <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x77b3486a6980>,
     <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x77b3486a6de0>,
     <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x77b3486a6de0>,
     <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x77b34890b420>,
     <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x77b3486a6a20>,
     <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x77b3486a6a20>,
     <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x77b3486a7c40>,
     <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x77b3485267a0>,
     <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x77b3486c80e0>,
     <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x77b3486c89a0>,
     <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x77b3486c8e00>,
     <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x77b3486c9260>,
     <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x77b3486c9260>,
     <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x77b348cf77e0>,
     <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x77b3486c9c60>,
     <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x77b3486cb420>,
     <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x77b3486cb880>,
     <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x77b3486cbce0>,
     <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x77b3486e05e0>,
     <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x77b348908720>,
     <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x77b3486e0a40>,
     <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x77b3486e0a40>,
     <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x77b3486a5e40>,
     <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x77b348700fe0>,
     <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x77b3486e0ea0>,
     <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x77b3486cb060>,
     <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x77b3487f9bc0>,
     <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x77b3487fa200>,
     <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x77b3487fa700>,
     <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x77b34890b380>,
     <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x77b3487fac00>,
     <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x77b3487fb100>,
     <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x77b34866c400>,
     <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x77b34866c900>,
     <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x77b3486e3240>,
     <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x77b3487d1e40>,
     <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x77b3487fb600>,
     <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x77b3487fbb00>,
     <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x77b3487fbec0>,
     <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x77b3487fbec0>,
     <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x77b34866ce00>,
     <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x77b34866ce00>,
     <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x77b348cf77e0>,
     <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x77b348694180>,
     <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f77380>, kernel=<OpOverload(op='aten.any', overload='dimname')>),
     <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x77b34866d580>,
     <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x77b34866da80>,
     <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x77b348702020>,
     <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x77b34866df80>,
     <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x77b34866e480>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x77b348737c40>,
     <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x77b34866efc0>,
     <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x77b3485aaac0>,
     <OpOverload(op='aten.log', overload='out')>: <function log at 0x77b348681bc0>,
     <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x77b3486816c0>,
     <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x77b3486816c0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f5d800>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>),
     <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x77b3486820c0>,
     <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x77b3486825c0>,
     <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x77b348682ac0>,
     <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x77b348683880>,
     <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x77b348683240>,
     <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x77b3487f9c60>,
     <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x77b3487d2e80>,
     <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x77b3487f8b80>,
     <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x77b3486e0180>,
     <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x77b34890a020>,
     <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x77b348909b20>,
     <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x77b34875cea0>,
     <OpOverload(op='aten.round', overload='out')>: <function round at 0x77b348683c40>,
     <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x77b3486a7240>,
     <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x77b3487d1ee0>,
     <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x77b3487019e0>,
     <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x77b34875e660>,
     <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x77b34854d9e0>,
     <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x77b34854c0e0>,
     <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x77b348527ce0>,
     <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x77b348527f60>,
     <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x77b34854c180>,
     <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x77b348526980>,
     <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x77b348525940>,
     <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61bc0>, kernel=<OpOverload(op='aten.item', overload='default')>),
     <OpOverload(op='aten.nonzero_numpy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f61ee0>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>),
     <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x77b348735760>,
     <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x77b348525da0>,
     <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x77b3487d1da0>,
     <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x77b3486c8e00>,
     <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x77b34854c540>,
     <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f904a0>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83f60>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>),
     <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83560>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f828e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83100>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>),
     <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82ac0>, kernel=<OpOverload(op='quantized.conv3d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82a20>, kernel=<OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82980>, kernel=<OpOverload(op='quantized.conv3d_transpose', overload='default')>),
     <OpOverload(op='quantized.conv3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f839c0>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>),
     <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83c40>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82480>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>),
     <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83380>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83060>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83b00>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>),
     <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f90540>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f827a0>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83740>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>),
     <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81a80>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f825c0>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>),
     <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82660>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81da0>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82700>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81d00>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f81b20>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>),
     <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f902c0>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f90220>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>),
     <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83a60>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83920>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>),
     <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82fc0>, kernel=<OpOverload(op='quantized.linear_unpack_fp16', overload='default')>),
     <OpOverload(op='quantized.conv3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82840>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82e80>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f823e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f831a0>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f822a0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f82520>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>),
     <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x77b277f83e20>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>)}

experimental_experiment.torch_dynamo.pprint_storage(storage: Any, indent: int = 0, as_list: bool = False) → List[str] | str

Pretty print for the storage.

Parameters:

• storage – any object

• indent – indentation

• as_list – return list or string

Returns:

list or string