.torch_dynamo

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: 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[source]

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 301: Compares LLAMA exporters for onnxrt backend or 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.00039450299982490833 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.0011331509999763512 s
    [to_onnx] start conversion to onnx (before optimization) mask_outputs=None
    [GraphBuilder-DEE.optimize] start with 13 nodes
    [GraphBuilder-DEE.optimize] #patterns=48
    [GraphBuilder-DEE.optimize] start with subgraphs
    [GraphBuilder-DEE.optimize] done with subgraphs
    [GraphBuilderPatternOptimization-DEE.optimize] start with 7 nodes, 0 initializers, 48 patterns, priorities=[0, 1, 3]
    [GraphBuilderPatternOptimization-DEE.optimize] iteration 0: 7 nodes, priority=0
    [GraphBuilderPatternOptimization-DEE.optimize] increase priority to 1
    [GraphBuilderPatternOptimization-DEE.optimize] iteration 1: 7 nodes, priority=1
    [GraphBuilderPatternOptimization-DEE.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.000 | max_time=TransposeMatMulPattern:0.000
    [GraphBuilderPatternOptimization-DEE.optimize] iteration 2: 6 nodes, priority=1
    [GraphBuilderPatternOptimization-DEE.optimize] increase priority to 3
    [GraphBuilderPatternOptimization-DEE.optimize] iteration 3: 6 nodes, priority=3
    [GraphBuilderPatternOptimization-DEE.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.000 | max_time=MatMulAddPattern:0.000
    [GraphBuilderPatternOptimization-DEE.optimize] iteration 4: 4 nodes, priority=3
    [GraphBuilderPatternOptimization-DEE.optimize] stops current_priority_index=3, priorities=[0, 1, 3]
    [GraphBuilderPatternOptimization-DEE.optimize] done after 5 iterations with 4 nodes in 0.004
    [GraphBuilder-DEE.optimize] done with 4 nodes in 0.005
    [GraphBuilder-DEE.to_onnx] make_model 1 inits 0 params
    [GraphBuilder-DEE.time_evaluation_constants_] 0
    [GraphBuilder-DEE._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024
    [GraphBuilder-DEE._build_initializers] switch low/high order
    [GraphBuilder-DEE._build_initializers] done in 9.890000001178123e-07s with 1 initializers, 0 large initializers
    [GraphBuilder-DEE._add_shape_information] dynamic shapes replacements={}
    [to_onnx] to_onnx done in 0.00583350200008681s and 4 nodes, 1 initializers, 5 inputs, 1 outputs
    [onnx_custom_backend] to_onnx done in 0.007672169999977996 with 4 nodes and 0 local functions.
    [onnx_custom_backend] starts creating InferenceSession
    [onnx_custom_backend] InferenceSession done in 0.007816831999889473
    discrepancies: -1.4901161193847656e-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: 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[source]

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

See 301: Compares LLAMA exporters for onnxrt backend for an example. 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.000349352000057479 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.0011803459999555344 s
    [to_onnx] start conversion to onnx (before optimization) mask_outputs=None
    [GraphBuilder-JCQ.optimize] start with 13 nodes
    [GraphBuilder-JCQ.optimize] #patterns=48
    [GraphBuilder-JCQ.optimize] start with subgraphs
    [GraphBuilder-JCQ.optimize] done with subgraphs
    [GraphBuilderPatternOptimization-JCQ.optimize] start with 7 nodes, 0 initializers, 48 patterns, priorities=[0, 1, 3]
    [GraphBuilderPatternOptimization-JCQ.optimize] iteration 0: 7 nodes, priority=0
    [GraphBuilderPatternOptimization-JCQ.optimize] increase priority to 1
    [GraphBuilderPatternOptimization-JCQ.optimize] iteration 1: 7 nodes, priority=1
    [GraphBuilderPatternOptimization-JCQ.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=Reshape2Of3Pattern:0.000
    [GraphBuilderPatternOptimization-JCQ.optimize] iteration 2: 6 nodes, priority=1
    [GraphBuilderPatternOptimization-JCQ.optimize] increase priority to 3
    [GraphBuilderPatternOptimization-JCQ.optimize] iteration 3: 6 nodes, priority=3
    [GraphBuilderPatternOptimization-JCQ.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.000 | max_time=MatMulAddPattern:0.000
    [GraphBuilderPatternOptimization-JCQ.optimize] iteration 4: 4 nodes, priority=3
    [GraphBuilderPatternOptimization-JCQ.optimize] stops current_priority_index=3, priorities=[0, 1, 3]
    [GraphBuilderPatternOptimization-JCQ.optimize] done after 5 iterations with 4 nodes in 0.004
    [GraphBuilder-JCQ.optimize] done with 4 nodes in 0.005
    [GraphBuilder-JCQ.to_onnx] make_model 1 inits 0 params
    [GraphBuilder-JCQ.time_evaluation_constants_] 0
    [GraphBuilder-JCQ._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024
    [GraphBuilder-JCQ._build_initializers] switch low/high order
    [GraphBuilder-JCQ._build_initializers] done in 7.680000635446049e-07s with 1 initializers, 0 large initializers
    [GraphBuilder-JCQ._add_shape_information] dynamic shapes replacements={}
    [to_onnx] to_onnx done in 0.005745961999991778s and 4 nodes, 1 initializers, 5 inputs, 1 outputs
     +C init7_s2_-1_1: int64:(2,):[-1, 1]
     +I input0: float32:(32, 10):0.21417555212974548,-0.1473262757062912,0.2478228658437729,-0.26707154512405396,0.16643796861171722...
     +I input1: float32:(32,):0.31574228405952454,0.2734023332595825,-0.20839324593544006,0.1352638602256775,-0.0894305557012558...
     +I input2: float32:(3, 10):0.3235389292240143,-0.9476402401924133,0.8603475689888,0.49417442083358765,0.830785870552063...
     +I input3: float32:(1, 32):0.01611590012907982,-0.01301494613289833,0.16923104226589203,0.14445868134498596,-0.13226519525051117...
     +I input4: float32:(1,):[-0.15789936482906342]
    Gemm(input2, input0, input1) -> input_1
     + input_1: float32:(3, 32):0.40392887592315674,-0.0670713484287262,0.2317640483379364,0.691973865032196,0.030237600207328796...
    Sigmoid(input_1) -> input_2
     + input_2: float32:(3, 32):0.5996312499046326,0.48323845863342285,0.5576830506324768,0.6664058566093445,0.5075588226318359...
    Reshape(input3, init7_s2_-1_1) -> _onx_transpose_l_self_modules_layers_modules_2_parameters_weight_0
     + _onx_transpose_l_self_modules_layers_modules_2_parameters_weight_0: float32:(32, 1):0.01611590012907982,-0.01301494613289833,0.16923104226589203,0.14445868134498596,-0.13226519525051117...
    Gemm(input_2, _onx_transpose_l_self_modules_layers_modules_2_parameters_weight_0, input4) -> output_0
     + output_0: float32:(3, 1):[-0.32225972414016724, -0.37874895334243774, -0.30708634853363037]
    discrepancies: 0.0

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[source]

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.8595420122146606,-0.41951894760131836,-0.034631773829460144,0.4959401488304138,-0.8967417478561401...
    <built-in method sigmoid of type object at 0x7febd8cf5320>((input_1,)) -> input_2
      + input_2: torch.float32:torch.Size([3, 32]):0.7025649547576904,0.3966318368911743,0.4913429617881775,0.6215047836303711,0.2897205352783203...
    <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.20882782340049744,0.12201979011297226,0.1486150026321411
    [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[source]

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 0x7feadab8e6c0>: <function out_dtype_decomp at 0x7feada451b20>,
     <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7feada2c4860>,
     <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x7feada2c4fe0>,
     <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x7feada2c5300>,
     <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x7feada2c58a0>,
     <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7feada2c58a0>,
     <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7feada2c5c60>,
     <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7feada2c5c60>,
     <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7feada2c5f80>,
     <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7feada2c5f80>,
     <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7feada2c5bc0>,
     <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7feada2c5bc0>,
     <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7feada2c67a0>,
     <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7feada2c67a0>,
     <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x7feada2c6a20>,
     <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x7feada2c6ca0>,
     <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7feada2c72e0>,
     <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7feada2c72e0>,
     <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x7feada2c7560>,
     <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x7feada2c7920>,
     <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7feada2c7ba0>,
     <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7feada2c7ba0>,
     <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7feada2c74c0>,
     <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7feada2c7e20>,
     <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7feada2c7e20>,
     <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7feada2c68e0>,
     <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7feada2c7c40>,
     <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7feada2c7ec0>,
     <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7feada0e4040>,
     <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7feada0e4180>,
     <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7feada2c74c0>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7feada0e4220>,
     <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7feada0e4220>,
     <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7feada0e4c20>,
     <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7feada0e4c20>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7feada0e4fe0>,
     <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7feada0e4fe0>,
     <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7feada0e5580>,
     <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7feada0e5580>,
     <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7feada0e5a80>,
     <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7feada0e5a80>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7feada0e5f80>,
     <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7feada0e5f80>,
     <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7feada0e6340>,
     <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7feada0e6340>,
     <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7feada0e6700>,
     <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7feada0e6700>,
     <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7feada0e6020>,
     <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7feada0e6020>,
     <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x7feada0e6c00>,
     <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7feada0e6c00>,
     <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7feada0e6fc0>,
     <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7feada0e6fc0>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7feada0e7600>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7feada0e7600>,
     <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x7feada3c6020>,
     <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x7feada3c6020>,
     <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7feada3c6f20>,
     <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7feada3c6a20>,
     <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7feada3c6f20>,
     <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7feada3c6c00>,
     <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7feada3c6c00>,
     <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7feada3c7560>,
     <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7feada3c7560>,
     <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7feada3c7a60>,
     <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7feada3c7a60>,
     <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7feada3c7ce0>,
     <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7feada3c7ce0>,
     <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7feada1f00e0>,
     <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7feada1f00e0>,
     <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7feada1f18a0>,
     <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7feada1f02c0>,
     <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7feada1f02c0>,
     <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7feada1f18a0>,
     <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7feada1f0680>,
     <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7feada1f0680>,
     <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7feada1f0a40>,
     <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7feada1f0a40>,
     <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7feada1f0e00>,
     <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7feada1f0e00>,
     <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7feada1f0ea0>,
     <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7feada1f1800>,
     <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7feada1f1800>,
     <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7feada1f1d00>,
     <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7feada1f1d00>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7feada1f2340>,
     <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7feada1f2340>,
     <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7feada1f2520>,
     <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7feada1f28e0>,
     <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7feada1f2020>,
     <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7feada1f22a0>,
     <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7feada1f25c0>,
     <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7feada1f2700>,
     <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7feada1f2700>,
     <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7feada1f3ba0>,
     <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7feada1f2840>,
     <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7feada1f3ba0>,
     <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7feada1f2980>,
     <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7feada1f2ac0>,
     <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7feada1f0220>,
     <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7feada1f2c00>,
     <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7feada1f3e20>,
     <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7feada1f2d40>,
     <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7feada1f2e80>,
     <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7feada1f2fc0>,
     <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7feada1f3100>,
     <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7feada1f3100>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7feada1f3240>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7feada1f3240>,
     <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7feada1f3f60>,
     <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7feada1f3380>,
     <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7feada1f34c0>,
     <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7feada1f34c0>,
     <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7feada1f28e0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7feada1f3600>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7feada1f3600>,
     <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7feada1f3740>,
     <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7feada1f3740>,
     <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7feada1f3ec0>,
     <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7feada1f3880>,
     <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7feada1f3ec0>,
     <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7feada1f39c0>,
     <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7feada1f39c0>,
     <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7feada1f3b00>,
     <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7feada1f3b00>,
     <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7feada1f3920>,
     <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7feada1f3c40>,
     <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7feada1f3c40>,
     <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7feada1f3d80>,
     <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7feada1f3920>,
     <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7feada1f36a0>,
     <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7feada1f3420>,
     <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7feada2214e0>,
     <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7feada1f2f20>,
     <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7feada1f2f20>,
     <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7feada2214e0>,
     <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7feada1f2ca0>,
     <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7feada1f2a20>,
     <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7feada1f27a0>,
     <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7feada1f2480>,
     <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7feada1f1da0>,
     <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7feada2200e0>,
     <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7feada220220>,
     <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7feada220220>,
     <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7feada1f3a60>,
     <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7feada220360>,
     <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7feada2204a0>,
     <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7feada2204a0>,
     <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7feada2205e0>,
     <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7feada2205e0>,
     <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7feada220720>,
     <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7feada1f3560>,
     <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7feada220860>,
     <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7feada2209a0>,
     <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7feada2209a0>,
     <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7feada220ae0>,
     <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7feada220ae0>,
     <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7feada1f3060>,
     <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7feada220c20>,
     <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7feada220d60>,
     <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7feada220ea0>,
     <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7feada1f2b60>,
     <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7feada220fe0>,
     <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7feada221120>,
     <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7feada1f2660>,
     <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7feada221260>,
     <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7feada1f32e0>,
     <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7feada2213a0>,
     <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7feada2213a0>,
     <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7feada221620>,
     <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7feada2216c0>,
     <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7feada221440>,
     <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7feada2211c0>,
     <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7feada2211c0>,
     <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7feada220f40>,
     <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7feada220f40>,
     <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7feada1f37e0>,
     <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7feada1f37e0>,
     <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7feada220cc0>,
     <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7feada220a40>,
     <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7feada2207c0>,
     <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7feada2207c0>,
     <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7feada1f3ce0>,
     <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7feada220540>,
     <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7feada2202c0>,
     <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7feada220040>,
     <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7feada220040>,
     <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7feada2218a0>,
     <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7feada2219e0>,
     <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7feada221b20>,
     <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7feada221b20>,
     <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7feada221c60>,
     <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7feada221da0>,
     <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7feada221ee0>,
     <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7feada222b60>,
     <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7feada222020>,
     <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7feada222160>,
     <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7feada2222a0>,
     <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7feada222c00>,
     <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7feada2223e0>,
     <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7feada222520>,
     <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7feada222660>,
     <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7feada2227a0>,
     <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7feada2227a0>,
     <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7feada222e80>,
     <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7feada2228e0>,
     <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7feada222a20>,
     <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7feada222de0>,
     <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7feada222980>,
     <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function PyCapsule.transpose at 0x7feada2237e0>,
     <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7feada222700>,
     <OpOverload(op='aten.transpose_copy', overload='int')>: <function PyCapsule.transpose at 0x7feada2237e0>,
     <OpOverload(op='aten.alias_copy', overload='default')>: <function PyCapsule.alias at 0x7feada222200>,
     <OpOverload(op='aten.alias_copy', overload='out')>: <function PyCapsule.alias at 0x7feada222200>,
     <OpOverload(op='aten.as_strided_copy', overload='default')>: <function PyCapsule.as_strided at 0x7feada221d00>,
     <OpOverload(op='aten.as_strided_copy', overload='out')>: <function PyCapsule.as_strided at 0x7feada221d00>,
     <OpOverload(op='aten.diagonal_copy', overload='default')>: <function PyCapsule.diagonal at 0x7feada221800>,
     <OpOverload(op='aten.diagonal_copy', overload='out')>: <function PyCapsule.diagonal at 0x7feada221800>,
     <OpOverload(op='aten.expand_copy', overload='default')>: <function PyCapsule.expand at 0x7feada220900>,
     <OpOverload(op='aten.expand_copy', overload='out')>: <function PyCapsule.expand at 0x7feada220900>,
     <OpOverload(op='aten.narrow_copy', overload='default')>: <function PyCapsule.narrow at 0x7feada221300>,
     <OpOverload(op='aten.narrow_copy', overload='out')>: <function PyCapsule.narrow at 0x7feada221300>,
     <OpOverload(op='aten.squeeze_copy', overload='default')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.permute_copy', overload='default')>: <function PyCapsule.permute at 0x7feada2232e0>,
     <OpOverload(op='aten.permute_copy', overload='out')>: <function PyCapsule.permute at 0x7feada2232e0>,
     <OpOverload(op='aten.t_copy', overload='default')>: <function PyCapsule.t at 0x7feada223560>,
     <OpOverload(op='aten.t_copy', overload='out')>: <function PyCapsule.t at 0x7feada223560>,
     <OpOverload(op='aten.unbind_copy', overload='int')>: <function PyCapsule.unbind at 0x7feada223880>,
     <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function PyCapsule.unbind at 0x7feada223880>,
     <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function PyCapsule.unsqueeze at 0x7feada221580>,
     <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function PyCapsule.unsqueeze at 0x7feada221580>,
     <OpOverload(op='aten.view_copy', overload='default')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.view_copy', overload='dtype')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.view_copy', overload='out')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7feada268a40>,
     <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7feada268a40>,
     <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7feada268cc0>,
     <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7feada268cc0>,
     <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x7feada269760>,
     <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7feada269760>,
     <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x7feada2699e0>,
     <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7feada2699e0>,
     <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x7feada269800>,
     <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7feada269800>,
     <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x7feada2689a0>,
     <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7feada2689a0>,
     <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x7feada269c60>,
     <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7feada269c60>,
     <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x7feada269ee0>,
     <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7feada269ee0>,
     <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x7feada26a7a0>,
     <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7feada26a7a0>,
     <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x7feada26aa20>,
     <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7feada26aa20>,
     <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7feada26bb00>,
     <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x7feada26aca0>,
     <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7feada26aca0>,
     <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x7feada26af20>,
     <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7feada26af20>,
     <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x7feada26bb00>,
     <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x7feada26b600>,
     <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7feada26b600>,
     <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x7feada26b880>,
     <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7feada26b880>,
     <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x7feada26a980>,
     <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7feada26a980>,
     <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x7feada268c20>,
     <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7feada268c20>,
     <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x7feada26bc40>,
     <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7feada26bc40>,
     <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x7feada26bec0>,
     <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7feada26bec0>,
     <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x7feada298180>,
     <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7feada298180>,
     <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x7feada298220>,
     <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x7feada2982c0>,
     <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7feada298a40>,
     <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7feada298a40>,
     <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7feada298e00>,
     <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7feada298e00>,
     <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7feada299da0>,
     <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7feada29a020>,
     <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7feada29a020>,
     <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7feada29a700>,
     <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7feada29a700>,
     <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7feada29ab60>,
     <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7feada29ab60>,
     <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7feada29b1a0>,
     <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7feada29b1a0>,
     <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7feada29b380>,
     <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7feada29b380>,
     <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7feada29b100>,
     <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7feada29b100>,
     <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x7feada29b880>,
     <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7feada2c4540>,
     <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7feada29be20>,
     <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7feada29be20>,
     <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7feada2c4860>,
     <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7feada2c4540>,
     <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x7feada2e85e0>,
     <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7feada2e9f80>,
     <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7feada2e9f80>,
     <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7feada2f8e00>,
     <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7feada2eb9c0>,
     <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7feada2eb9c0>,
     <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7feada2f8400>,
     <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7feada2f8400>,
     <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7feada2f8e00>,
     <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7feada2e99e0>,
     <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7feada2e99e0>,
     <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7feada2f9120>,
     <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7feada2f9120>,
     <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7feada2f9620>,
     <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7feada2f9620>,
     <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7feada2fb4c0>,
     <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7feada2fa020>,
     <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7feada2fa020>,
     <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7feada2fb4c0>,
     <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x7feada2fb4c0>,
     <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7feada2fb4c0>,
     <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7feada2fa0c0>,
     <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7feada2fa0c0>,
     <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7feada311580>,
     <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7feada311580>,
     <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7feada311a80>,
     <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7feada311a80>,
     <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7feada312e80>,
     <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7feada312e80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7feada3413a0>,
     <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7feada3413a0>,
     <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7feada3437e0>,
     <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7feada3437e0>,
     <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7feada342660>,
     <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7feada342660>,
     <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7feada342f20>,
     <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7feada342f20>,
     <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7feada343380>,
     <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7feada343380>,
     <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7feada342160>,
     <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7feada342160>,
     <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7feada3439c0>,
     <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7feada3439c0>,
     <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7feada343e20>,
     <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7feada343e20>,
     <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7feada362b60>,
     <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7feada362b60>,
     <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7feada362f20>,
     <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7feada362f20>,
     <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7feada3632e0>,
     <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7feada3632e0>,
     <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7feada3632e0>,
     <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7feada3632e0>,
     <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7feada363560>,
     <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7feada363560>,
     <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7feada363560>,
     <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7feada363560>,
     <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7feada3637e0>,
     <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7feada3637e0>,
     <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7feada3637e0>,
     <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7feada3637e0>,
     <OpOverload(op='aten.all', overload='default')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.any', overload='default')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std', overload='default')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='dim')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7feada37a200>,
     <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7feada37a200>,
     <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7feada37b2e0>,
     <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7feada37b2e0>,
     <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7feada37b4c0>,
     <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7feada37be20>,
     <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7feada37bb00>,
     <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7feada37bb00>,
     <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7feada39c720>,
     <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7feada39c540>,
     <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7feada39c540>,
     <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7feada37ad40>,
     <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7feada37ad40>,
     <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7feada39c9a0>,
     <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7feada39c9a0>,
     <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7feada39cf40>,
     <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7feada39cf40>,
     <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7feada39d1c0>,
     <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7feada39d1c0>,
     <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7feada39d4e0>,
     <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7feada39d4e0>,
     <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7feada39db20>,
     <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7feada39db20>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7feada39e840>,
     <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7feada39cea0>,
     <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7feada39e7a0>,
     <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7feada39e7a0>,
     <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x7feada39e7a0>,
     <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7feada39e7a0>,
     <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7feada39eac0>,
     <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7feada39cae0>,
     <OpOverload(op='aten.t', overload='default')>: <function t at 0x7feada39f2e0>,
     <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7feada39ede0>,
     <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7feada39ede0>,
     <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7feada39f1a0>,
     <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7feada39f1a0>,
     <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7feada39f4c0>,
     <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7feada39f6a0>,
     <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7feada39fb00>,
     <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7feada39fb00>,
     <OpOverload(op='aten.view', overload='default')>: <function view at 0x7feada39fec0>,
     <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7feada39fba0>,
     <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x7feada39fba0>,
     <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7feada39fba0>,
     <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7feada39fba0>,
     <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7feada39fd80>,
     <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7feada39ea20>,
     <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7feada3c5580>,
     <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7feada3c5580>,
     <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7feada5b7ba0>,
     <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7feada5b7880>,
     <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7feada5b7880>,
     <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7feada5b7b00>,
     <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7feada5b7b00>,
     <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7feada5b7e20>,
     <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7feada5b6e80>,
     <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7feada5b6e80>,
     <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7feada5b5a80>,
     <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7feada5b5a80>,
     <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7feada3e04a0>,
     <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7feada3e04a0>,
     <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7feada3e0ae0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7feada3e0ae0>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7feada3e0b80>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7feada3e0b80>,
     <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7feada3e0cc0>,
     <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7feada3e0e00>,
     <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7feada3e0f40>,
     <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7feada3e18a0>,
     <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7feada3e18a0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7feada3e1c60>,
     <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7feada3e1bc0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7feada3e1d00>,
     <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7feada3e1bc0>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7feada3e1f80>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7feada3e2020>,
     <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7feada3e0ea0>,
     <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7feada3e0ea0>,
     <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7feada3e2200>,
     <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7feada3e22a0>,
     <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7feada3e23e0>,
     <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7feada3e2520>,
     <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7feada3e27a0>,
     <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7feada3e2840>,
     <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7feada3e2980>,
     <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7feada3e3240>,
     <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7feada3e3240>,
     <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7feada3e3600>,
     <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7feada3e3600>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7feada3e3ec0>,
     <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7feada40c0e0>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7feada3e3ec0>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7feada40c180>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7feada3e37e0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7feada3e2b60>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7feada3e2b60>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7feada40c7c0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7feada40c7c0>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7feada40cc20>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7feada40cc20>,
     <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7feada40cf40>,
     <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7feada40cf40>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7feada40db20>,
     <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7feada40d1c0>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7feada40db20>,
     <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7feada40d1c0>,
     <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7feada40cfe0>,
     <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7feada40d6c0>,
     <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7feada40d620>,
     <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7feada40d620>,
     <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7feada40e700>,
     <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7feada429760>,
     <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7feada40d800>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7feada40e520>,
     <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7feada40e700>,
     <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7feada40ef20>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7feada40ef20>,
     <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7feada40f1a0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7feada40f240>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7feada40f240>,
     <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7feada40f560>,
     <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7feada40f560>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7feada40f880>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7feada40f880>,
     <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7feada40fba0>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7feada428a40>,
     <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7feada40fba0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7feada40fec0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7feada40fec0>,
     <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7feada428720>,
     <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7feada428900>,
     <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7feada428ae0>,
     <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7feada428cc0>,
     <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7feada429120>,
     <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7feada429300>,
     <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7feada429620>,
     <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7feada429800>,
     <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7feada429760>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7feada40d940>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7feada428a40>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7feada4299e0>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7feada4299e0>,
     <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7feada429da0>,
     <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7feada40e840>,
     <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7feada42a160>,
     <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7feada42a160>,
     <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7feada42a160>,
     <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7feada40e840>,
     <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7feada429f80>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7feada42a340>,
     <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7feada42a480>,
     <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7feada42aca0>,
     <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7feada42aca0>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7feada42b380>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7feada42b380>,
     <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7feada42a520>,
     <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7feada42a520>,
     <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7feada429940>,
     <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7feada429940>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7feada42bd80>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7feada42bd80>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7feada4505e0>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7feada4505e0>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7feada450a40>,
     <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4514e0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4514e0>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7feada40eac0>,
     <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7feada40eac0>,
     <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7feada451a80>,
     <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7feada450540>,
     <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7feada450540>,
     <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7feada4518a0>,
     <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7feada4520c0>,
     <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7feada4520c0>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7feada4527a0>,
     <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7feada451c60>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7feada4527a0>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7feada453380>,
     <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7feada453380>,
     <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7feada452980>,
     <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7feada453f60>,
     <OpOverload(op='aten.take', overload='default')>: <function take at 0x7feada452d40>,
     <OpOverload(op='aten.take', overload='out')>: <function take at 0x7feada452d40>,
     <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453f60>,
     <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7feada4532e0>,
     <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7feada4532e0>,
     <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada452a20>,
     <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada452020>,
     <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada451bc0>,
     <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada450680>,
     <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada453880>,
     <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada453880>,
     <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4539c0>,
     <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ba0>,
     <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453b00>,
     <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453c40>,
     <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453d80>,
     <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ec0>,
     <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ec0>,
     <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478040>,
     <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478040>,
     <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4782c0>,
     <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478400>,
     <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478400>,
     <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478540>,
     <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478540>,
     <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478680>,
     <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478680>,
     <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4787c0>,
     <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478900>,
     <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478a40>,
     <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478b80>,
     <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7feada453920>,
     <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4519e0>,
     <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7feada378860>,
     <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7feada453100>,
     <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7feada378860>,
     <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7feada453100>,
     <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.var', overload='default')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='dim')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7feada378e00>,
     <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7feada378e00>,
     <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7feada378cc0>,
     <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7feada378cc0>,
     <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7feada3c6480>,
     <OpOverload(op='aten.full', overload='out')>: <function full at 0x7feada3c6160>,
     <OpOverload(op='aten.full', overload='default')>: <function full at 0x7feada3c6160>,
     <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7feada40d120>,
     <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7feada40d120>,
     <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7feada340680>,
     <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7feada340680>,
     <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7feada56e3e0>,
     <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7feada56c720>,
     <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7feada56c720>,
     <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7feada56cb80>,
     <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7feada56cb80>,
     <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7feada56c680>,
     <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7feada56c680>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7feada56f600>,
     <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7feada56d760>,
     <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7feada56d760>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7feada56d800>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7feada56d800>,
     <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7feada56dbc0>,
     <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7feada56dbc0>,
     <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7feada56e160>,
     <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7feada56e160>,
     <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7feada56e480>,
     <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7feada56e480>,
     <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7feada56e520>,
     <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7feada56e520>,
     <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7feada56e3e0>,
     <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7feada56d300>,
     <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7feada56d300>,
     <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7feada56e8e0>,
     <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7feada56ed40>,
     <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7feada56ed40>,
     <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7feada56ede0>,
     <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7feada56ede0>,
     <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7feada56f100>,
     <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7feada56f1a0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7feada56f6a0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7feada56f6a0>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7feada56f740>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7feada56f740>,
     <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7feada56c9a0>,
     <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7feada56fec0>,
     <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7feada56fec0>,
     <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7feada56efc0>,
     <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7feada56ff60>,
     <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7feada56ff60>,
     <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7feada56ea20>,
     <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7feada5b42c0>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7feada5b47c0>,
     <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7feada5b4720>,
     <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7feada5b4720>,
     <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7feada5b4180>,
     <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7feada5b4180>,
     <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7feada5b4b80>,
     <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7feada5b4b80>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7feada5b4ea0>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7feada5b4ea0>,
     <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7feada5b5440>,
     <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7feada5b5440>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7feada5b54e0>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7feada5b54e0>,
     <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7feada5b5b20>,
     <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7feada5b5b20>,
     <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7feada5b7ce0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7feada5b5bc0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7feada5b5bc0>,
     <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7feada5b7ce0>,
     <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7feada5b6160>,
     <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7feada5b6160>,
     <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7feada5b63e0>,
     <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7feada5b63e0>,
     <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7feada5b49a0>,
     <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7feada5b67a0>,
     <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7feada5b67a0>,
     <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7feada5b68e0>,
     <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7feada5b68e0>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7feada5b6f20>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7feada5b6f20>,
     <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7feada5b7240>,
     <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7feada5b7240>,
     <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7feada5b7600>,
     <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7feada5b7600>,
     <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7feada313f60>,
     <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7feada313f60>,
     <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7feada3244a0>,
     <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7feada3244a0>,
     <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7feada3249a0>,
     <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7feada3249a0>,
     <OpOverload(op='aten.add', overload='out')>: <function add at 0x7feada3249a0>,
     <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7feada3249a0>,
     <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7feada326840>,
     <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7feada324e00>,
     <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7feada324e00>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='out')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7feada326840>,
     <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7feada326840>,
     <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7feada326840>,
     <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7feada3277e0>,
     <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7feada3277e0>,
     <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7feada327c40>,
     <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7feada327c40>,
     <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7feada937ba0>,
     <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7feada3400e0>,
     <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7feada3400e0>,
     <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7feada3400e0>,
     <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7feada3400e0>,
     <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7feada326200>,
     <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7feada326200>,
     <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7feada340ae0>,
     <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7feada340ae0>,
     <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7feada340ae0>,
     <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7feada340ae0>,
     <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7feada340f40>,
     <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7feada340f40>,
     <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7feada340f40>,
     <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7feada340f40>,
     <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7feada341800>,
     <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7feada341800>,
     <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7feada341c60>,
     <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7feada341c60>,
     <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7feada3420c0>,
     <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7feada3420c0>,
     <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7feada342ac0>,
     <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7feada342ac0>,
     <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7feada342ac0>,
     <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7feada342ac0>,
     <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7feada3602c0>,
     <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7feada3602c0>,
     <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7feada3602c0>,
     <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7feada3602c0>,
     <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7feada360720>,
     <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7feada360720>,
     <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7feada360b80>,
     <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7feada360b80>,
     <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7feada37ade0>,
     <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7feada361440>,
     <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7feada361440>,
     <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7feada361440>,
     <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7feada361440>,
     <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7feada3618a0>,
     <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7feada3618a0>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7feada341d00>,
     <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7feada341d00>,
     <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7feada341d00>,
     <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7feada341d00>,
     <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7feada453240>,
     <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7feada453240>,
     <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7feada39e5c0>,
     <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7feada39f560>,
     <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7feada39f560>,
     <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7feada37a7a0>,
     <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x7feada37ade0>,
     <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7feada37ade0>,
     <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7feada37ade0>,
     <OpOverload(op='aten.where', overload='self')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.where', overload='default')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7feada47ba60>,
     <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7feada47bf60>,
     <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7feada47bf60>,
     <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7feada2e9260>,
     <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7feada2e9260>,
     <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7feada2e9760>,
     <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7feada2e9760>,
     <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7feada2e84a0>,
     <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7feada2e84a0>,
     <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7feada2e89a0>,
     <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7feada2e89a0>,
     <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7feada2e8d60>,
     <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7feada2e8d60>,
     <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7feada363e20>,
     <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7feada47a700>,
     <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7feada47a700>,
     <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7feada363e20>,
     <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7feada2e9a80>,
     <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7feada2e9a80>,
     <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7feada2ea480>,
     <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7feada2ea480>,
     <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7feada937ba0>,
     <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7feada2ea980>,
     <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7feada2ea980>,
     <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7feada2eae80>,
     <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7feada2eae80>,
     <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7feada2eb380>,
     <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7feada2eb380>,
     <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7feada56d1c0>,
     <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7feada2ebec0>,
     <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7feada56d260>,
     <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7feada2ebec0>,
     <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7feada2fa520>,
     <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7feada2fa520>,
     <OpOverload(op='aten.log', overload='default')>: <function log at 0x7feada2faa20>,
     <OpOverload(op='aten.log', overload='out')>: <function log at 0x7feada2faa20>,
     <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7feada2faf20>,
     <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7feada2faf20>,
     <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7feada2fb420>,
     <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7feada2fb420>,
     <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7feada2fb920>,
     <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7feada2fb920>,
     <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7feada3107c0>,
     <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7feada3107c0>,
     <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7feada310180>,
     <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7feada310180>,
     <OpOverload(op='aten.round', overload='default')>: <function round at 0x7feada310b80>,
     <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7feada310b80>,
     <OpOverload(op='aten.round', overload='out')>: <function round at 0x7feada310b80>,
     <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7feada311080>,
     <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7feada311080>,
     <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7feada310b80>,
     <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7feada311f80>,
     <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7feada311f80>,
     <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7feada312480>,
     <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7feada312480>,
     <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7feada451a80>,
     <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7feada312980>,
     <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7feada312980>,
     <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7feada312520>,
     <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7feada312520>,
     <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7feada313100>,
     <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7feada313100>,
     <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7feada313a60>,
     <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7feada313a60>,
     <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7feada47a160>,
     <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7feada47a160>,
     <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7feada47a660>,
     <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7feada47a660>,
     <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7feada47ab60>,
     <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7feada47ab60>,
     <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7feada47b060>,
     <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7feada47b060>,
     <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7feada47b560>,
     <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7feada47b560>,
     <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7feada47ba60>,
     <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7feada40d940>,
     <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7feada40e700>,
     <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7feada39eac0>,
     <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7feada39cae0>,
     <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7feada1f2520>,
     <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7feada452e80>,
     <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7feada3c6480>,
     <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7feada37a7a0>,
     <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7feada39ff60>,
     <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7feada3c5300>,
     <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7feada3c5300>,
     <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7feada3c67a0>,
     <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7feada3c67a0>,
     <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7feada3c5260>,
     <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7feada3c5260>,
     <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7feada3c4ae0>,
     <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7feada3c4ae0>,
     <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7feada3c4d60>,
     <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7feada3c4d60>,
     <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7feada3c4fe0>,
     <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7feada3c4fe0>,
     <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7feada39f420>,
     <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7feada39f420>,
     <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7feada3c4e00>,
     <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7feada3c4e00>,
     <OpOverload(op='aten.item', overload='default')>: <function item at 0x7feada363ec0>,
     <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7feada42a200>,
     <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7feada5b56c0>,
     <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7feada5b56c0>,
     <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478180>}
experimental_experiment.torch_dynamo.get_decomposition_table()[source]

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.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7feada3e1f80>}
experimental_experiment.torch_dynamo.get_decomposition_table_by_name(name: str)[source]

Returns a predefined decomposition table.

Parameters:

name – name see below

Returns:

decomposition table

experimental_experiment.torch_dynamo.get_decomposition_table_dynamo(onnx_registry=None)[source]

Returns the decomposition table needed for the dynamo exporter.

Parameters:

onnx_registry – instance of class torch.onnx._internal.exporter.OnnxRegistry

The value is:

<<<

import pprint
from experimental_experiment.torch_dynamo import get_decomposition_table_dynamo

pprint.pprint(get_decomposition_table_dynamo())

>>>

    /home/xadupre/github/experimental-experiment/experimental_experiment/torch_dynamo/__init__.py:140: DeprecationWarning: torch.onnx.dynamo_export is deprecated since 2.7.0. Please use torch.onnx.export(..., dynamo=True) instead.
      return create_onnx_friendly_decomposition_table(onnx_registry or OnnxRegistry())
    /home/xadupre/vv/this312/lib/python3.12/site-packages/torch/onnx/_internal/_exporter_legacy.py:91: UserWarning: torch.onnx.dynamo_export only implements opset version 18 for now. If you need to use a different opset version, please register them with register_custom_op.
      warnings.warn(
    {<torch._higher_order_ops.out_dtype.OutDtypeOperator object at 0x7feadab8e6c0>: <function out_dtype_decomp at 0x7feada451b20>,
     <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899760>, kernel=<OpOverload(op='image.decode_png', overload='default')>),
     <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898180>, kernel=<OpOverload(op='image.read_file', overload='default')>),
     <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8993a0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>),
     <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e0860>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>),
     <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3ba0>, kernel=<OpOverload(op='torchvision._cuda_version', 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 0x7fe9be8982c0>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>),
     <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898900>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>),
     <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898ae0>, kernel=<OpOverload(op='image.encode_png', overload='default')>),
     <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898360>, kernel=<OpOverload(op='image.decode_gif', overload='default')>),
     <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899300>, kernel=<OpOverload(op='image.decode_webp', overload='default')>),
     <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e23e0>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>),
     <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e31a0>, kernel=<OpOverload(op='image.write_file', overload='default')>),
     <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89ae80>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>),
     <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e02c0>, kernel=<OpOverload(op='image.decode_image', overload='default')>),
     <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708a40>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>),
     <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c98a0>, kernel=<OpOverload(op='aten.chain_matmul', 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 0x7fe9be71a700>, 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 0x7fe9be718f40>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>),
     <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 0x7fe9be702660>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be730360>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>),
     <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf2384a0>, kernel=<OpOverload(op='aten.argsort', overload='stable')>),
     <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898fe0>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be6faac0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>),
     <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12bec0>, kernel=<OpOverload(op='aten.arctan2', overload='default')>),
     <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf129440>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>),
     <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9120>, kernel=<OpOverload(op='aten.can_cast', overload='default')>),
     <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9c60>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b6520>, kernel=<OpOverload(op='aten.linalg_matrix_power', 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 0x7fe9be7236a0>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>),
     <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3240>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>),
     <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dbc0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>),
     <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b560>, 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 0x7fe9be8c98a0>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>),
     <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a0c0>, kernel=<OpOverload(op='aten.special_gammaincc', 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 0x7fea23f8e980>, 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 0x7fea23f8dee0>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>),
     <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb9a80>, 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 0x7fe9be7080e0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>),
     <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708680>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>),
     <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708e00>, 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 0x7fe9be8aee80>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>),
     <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a8e0>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>),
     <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b6200>, kernel=<OpOverload(op='aten.combinations', overload='default')>),
     <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709620>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b240>, kernel=<OpOverload(op='aten.inverse', overload='default')>),
     <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708fe0>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722700>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7fe9be721da0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>),
     <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720f40>, kernel=<OpOverload(op='aten.multilabel_margin_loss', 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 0x7fe9be7214e0>, 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 0x7fe9be722b60>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>),
     <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720540>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>),
     <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7202c0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>),
     <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7232e0>, kernel=<OpOverload(op='aten.negative', overload='default')>),
     <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7227a0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d6c0>, kernel=<OpOverload(op='aten.histogramdd', 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 0x7fe9be8e3880>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>),
     <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f4c0>, kernel=<OpOverload(op='aten.kl_div', overload='default')>),
     <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f7e0>, kernel=<OpOverload(op='aten.arcsin', overload='default')>),
     <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ef20>, 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 0x7fe9be745080>, kernel=<OpOverload(op='aten.matrix_exp', 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 0x7fea23f8f1a0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>),
     <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dd00>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f560>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>),
     <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e5c0>, kernel=<OpOverload(op='aten.trapz', overload='dx')>),
     <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b9c0>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>),
     <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bce0>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>),
     <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7471a0>, 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 0x7fe9be747ba0>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>),
     <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7476a0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744860>, 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 0x7fe9be747240>, 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 0x7fe9be746160>, 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 0x7fe9be744cc0>, 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 0x7fe9be744b80>, 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 0x7fe9be745620>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>),
     <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745120>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', 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 0x7fe9be745bc0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>),
     <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722e80>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>),
     <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718e00>, kernel=<OpOverload(op='aten.orgqr', overload='default')>),
     <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719620>, kernel=<OpOverload(op='aten.stride', overload='int')>),
     <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 0x7fe9be71a660>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71ac00>, kernel=<OpOverload(op='aten.quantile', overload='default')>),
     <OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899d00>, 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 0x7fe9be89a160>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb560>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>),
     <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b920>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be731260>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>),
     <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8af740>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>),
     <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701d00>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>),
     <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c8f40>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12a0c0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>),
     <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb420>, kernel=<OpOverload(op='aten.special_logit', overload='default')>),
     <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7234c0>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>),
     <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b060>, kernel=<OpOverload(op='_test.get_first', overload='default')>),
     <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be703060>, kernel=<OpOverload(op='aten.absolute', overload='default')>),
     <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70af20>, kernel=<OpOverload(op='aten.take_along_dim', 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 0x7fe9be89afc0>, 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 0x7fe9be8c9f80>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>),
     <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709580>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>),
     <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8a0180>, kernel=<OpOverload(op='aten.inner', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fa60>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>),
     <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a840>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>),
     <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8a0400>, kernel=<OpOverload(op='aten.median', overload='names_dim')>),
     <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71bb00>, kernel=<OpOverload(op='aten._lu_with_info', 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 0x7fe9be89a020>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>),
     <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12ade0>, kernel=<OpOverload(op='aten._remove_batch_dim', overload='default')>),
     <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71be20>, kernel=<OpOverload(op='aten.less', overload='Scalar')>),
     <OpOverload(op='aten.special_xlogy', overload='other_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7093a0>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>),
     <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c80e0>, kernel=<OpOverload(op='aten.vander', overload='default')>),
     <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71bc40>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>),
     <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720d60>, kernel=<OpOverload(op='aten.thnn_conv2d', 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 0x7fe9be7187c0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8aeca0>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', 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 0x7fe9be700680>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>),
     <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7305e0>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>),
     <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7327a0>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>),
     <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c8ea0>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>),
     <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca5c0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>),
     <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca020>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>),
     <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8caf20>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>),
     <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9b20>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', 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 0x7fe9be8ca980>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b7a60>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>),
     <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c99e0>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>),
     <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 0x7fea23f8fe20>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e660>, kernel=<OpOverload(op='aten._cast_Long', 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 0x7fea23f8f100>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d760>, kernel=<OpOverload(op='aten.bilinear', 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 0x7fe9be7082c0>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>),
     <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709e40>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>),
     <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 0x7fe9be70a340>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e020>, 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 0x7fe9be8ac540>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>),
     <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70ae80>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>),
     <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b380>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>),
     <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708b80>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>),
     <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c99e0>, kernel=<OpOverload(op='aten.arccos', overload='default')>),
     <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c9e40>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>),
     <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709800>, kernel=<OpOverload(op='aten.ger', overload='default')>),
     <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a7a0>, kernel=<OpOverload(op='aten.data', overload='default')>),
     <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf23b420>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>),
     <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a3e0>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>),
     <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b4c0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>),
     <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702b60>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb7e0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>),
     <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7004a0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>),
     <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf23ab60>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>),
     <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700720>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>),
     <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700860>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be6fa2a0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>),
     <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca840>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cbec0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>),
     <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb100>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>),
     <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9300>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>),
     <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca200>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>),
     <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c040>, kernel=<OpOverload(op='aten._backward', overload='default')>),
     <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d080>, kernel=<OpOverload(op='aten._reshape_from_tensor', 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 0x7fea23f8ce00>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fc40>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c9a0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>),
     <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf476520>, kernel=<OpOverload(op='aten.is_leaf', overload='default')>),
     <OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e520>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cae0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>),
     <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709f80>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>),
     <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89ade0>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>),
     <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a200>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>),
     <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718fe0>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>),
     <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b060>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>),
     <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709b20>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>),
     <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b600>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>),
     <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b7920>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>),
     <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9e40>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>),
     <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708ea0>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>),
     <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7098a0>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7207c0>, kernel=<OpOverload(op='aten.arctan', overload='default')>),
     <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723ba0>, kernel=<OpOverload(op='aten.gradient', overload='array')>),
     <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723920>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>),
     <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7220c0>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>),
     <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a5c0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>),
     <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d940>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>),
     <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722de0>, kernel=<OpOverload(op='aten.trapz', overload='x')>),
     <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723c40>, kernel=<OpOverload(op='aten.nanmean', overload='default')>),
     <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fd80>, kernel=<OpOverload(op='aten.tensordot', overload='default')>),
     <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d300>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8eac0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>),
     <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3c40>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c900>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>),
     <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8999e0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>),
     <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744900>, kernel=<OpOverload(op='aten.fix', 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 0x7fe9be898e00>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>),
     <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747380>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>),
     <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746d40>, kernel=<OpOverload(op='aten.vstack', overload='default')>),
     <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b920>, kernel=<OpOverload(op='aten.special_round', overload='default')>),
     <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89aca0>, kernel=<OpOverload(op='aten.msort', overload='default')>),
     <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747ec0>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>),
     <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e16c0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>),
     <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7479c0>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>),
     <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720860>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>),
     <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744680>, 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 0x7fe9be8e2f20>, kernel=<OpOverload(op='_test.cat', overload='default')>),
     <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7454e0>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>),
     <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747740>, kernel=<OpOverload(op='aten.kron', overload='default')>),
     <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746980>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745440>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>),
     <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722d40>, kernel=<OpOverload(op='aten.mode', overload='dimname')>),
     <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723f60>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>),
     <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746fc0>, kernel=<OpOverload(op='aten.diagflat', 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 0x7fe9be723240>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>),
     <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722340>, kernel=<OpOverload(op='aten.to_dense', overload='default')>),
     <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7237e0>, 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 0x7fe9be723a60>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7300e0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>),
     <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f600>, kernel=<OpOverload(op='aten.argwhere', overload='default')>),
     <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721b20>, kernel=<OpOverload(op='aten.is_inference', overload='default')>),
     <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ad940>, kernel=<OpOverload(op='aten.is_conj', overload='default')>),
     <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720fe0>, kernel=<OpOverload(op='aten.row_stack', 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 0x7fe9be720400>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7205e0>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be731760>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>),
     <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c180>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>),
     <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722a20>, kernel=<OpOverload(op='aten.fliplr', overload='default')>),
     <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cc20>, kernel=<OpOverload(op='aten.one_hot', 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 0x7fea23f8ec00>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bd80>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>),
     <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899800>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>),
     <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bc40>, kernel=<OpOverload(op='aten.cdist', overload='default')>),
     <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746ca0>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>),
     <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747ce0>, kernel=<OpOverload(op='aten.chalf', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746520>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>),
     <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7468e0>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>),
     <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746020>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>),
     <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744360>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>),
     <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744f40>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>),
     <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f740>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>),
     <OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7449a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', 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 0x7fe9be722c00>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>),
     <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746de0>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>),
     <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 0x7fe9be723600>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>),
     <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718c20>, kernel=<OpOverload(op='aten.masked_select_backward', 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 0x7fead83c9bc0>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>),
     <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a3e0>, kernel=<OpOverload(op='c10d_functional.broadcast', 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 0x7fe9be7194e0>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', 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 0x7fe9be719260>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>),
     <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7027a0>, kernel=<OpOverload(op='aten.pinverse', overload='default')>),
     <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be733e20>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>),
     <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf128900>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>),
     <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be703380>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>),
     <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709d00>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>),
     <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb880>, kernel=<OpOverload(op='aten.qr', overload='default')>),
     <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c8e00>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>),
     <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca2a0>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', 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 0x7fea23f8cb80>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>),
     <OpOverload(op='profiler._record_function_exit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bec0>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>),
     <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ea20>, kernel=<OpOverload(op='aten.linalg_eigvalsh', 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 0x7fea23f8d1c0>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', 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 0x7fe9bf474900>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f920>, kernel=<OpOverload(op='aten.linalg_matrix_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 0x7fea23f8e840>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>),
     <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70bce0>, kernel=<OpOverload(op='aten.special_i0', overload='default')>),
     <OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745d00>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>),
     <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70be20>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>),
     <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70aa20>, kernel=<OpOverload(op='aten.size', overload='int')>),
     <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7193a0>, kernel=<OpOverload(op='aten.l1_loss', 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 0x7fe9be8afec0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>),
     <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70ad40>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>),
     <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9ee0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746200>, kernel=<OpOverload(op='aten.flipud', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719c60>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>),
     <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f420>, kernel=<OpOverload(op='aten._nnpack_available', 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 0x7fe9be71ba60>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>),
     <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899940>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>),
     <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7020c0>, 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 0x7fe9be7309a0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>),
     <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ae2a0>, kernel=<OpOverload(op='aten.is_neg', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be6fb9c0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>),
     <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12a980>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>),
     <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca160>, kernel=<OpOverload(op='aten.diff', overload='default')>),
     <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca7a0>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>),
     <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ac680>, kernel=<OpOverload(op='aten.special_expit', 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 0x7fe9be89bf60>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>),
     <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ede0>, kernel=<OpOverload(op='aten.argsort', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fec0>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>),
     <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d4e0>, kernel=<OpOverload(op='aten.stft', 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 0x7fea23f8f880>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', 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 0x7fea23f8e2a0>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>),
     <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709300>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>),
     <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8acfe0>, kernel=<OpOverload(op='aten.output_nr', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ae5c0>, 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 0x7fe9be7099e0>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>),
     <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c4a0>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>),
     <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708900>, kernel=<OpOverload(op='aten.arccosh', overload='default')>),
     <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83ca0c0>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>),
     <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a340>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>),
     <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7319e0>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>),
     <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701da0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>),
     <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7198a0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>),
     <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf129120>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>),
     <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12bb00>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>),
     <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf128ea0>, 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 0x7fe9be700ea0>, 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 0x7fe9be702de0>, 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 0x7fe9be8cb4c0>, kernel=<OpOverload(op='aten.align_as', 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 0x7fe9be8c84a0>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', 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 0x7fe9be722020>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8caca0>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>),
     <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8996c0>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>),
     <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8eca0>, 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 0x7fea23f8e3e0>, kernel=<OpOverload(op='aten.dstack', overload='default')>),
     <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e480>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>),
     <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cd60>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>),
     <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c860>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>),
     <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf477a60>, 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 0x7fe9be7231a0>, 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 0x7fea23f8cf40>, 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 0x7fe9be899580>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf477b00>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>),
     <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b7e0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>),
     <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b920>, 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 0x7fe9be70bc40>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>),
     <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9080>, 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 0x7fe9be8ae980>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', 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 0x7fe9be8b44a0>, 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 0x7fe9be8a28e0>, 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 0x7fe9be709760>, kernel=<OpOverload(op='aten._cast_Short', 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 0x7fe9be89b380>, 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 0x7fe9be8b7ec0>, 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 0x7fead83c9f80>, 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 0x7fe9be708720>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>),
     <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702c00>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>),
     <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb380>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>),
     <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7313a0>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>),
     <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709260>, kernel=<OpOverload(op='aten.size', overload='Dimname')>),
     <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b79c0>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>),
     <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be733240>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>),
     <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ccc0>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>),
     <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b6ac0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718360>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>),
     <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b4400>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>),
     <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719b20>, kernel=<OpOverload(op='aten.hstack', overload='default')>),
     <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fb00>, kernel=<OpOverload(op='aten.is_signed', overload='default')>),
     <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c96c0>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>),
     <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720cc0>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e0c0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>),
     <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 0x7fea23fb8360>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e700>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>),
     <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702f20>, kernel=<OpOverload(op='aten.outer', overload='default')>),
     <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70bba0>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>),
     <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8afc40>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>),
     <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d3a0>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>),
     <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719760>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>),
     <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745f80>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>),
     <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b2e0>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>),
     <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8efc0>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>),
     <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8caa20>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>),
     <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747420>, kernel=<OpOverload(op='aten.isreal', overload='default')>),
     <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723420>, kernel=<OpOverload(op='aten.rrelu', overload='default')>),
     <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b6d40>, 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 0x7fe9be8accc0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>),
     <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8998a0>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>),
     <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f240>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>),
     <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb8a40>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>),
     <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719080>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>),
     <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701080>, kernel=<OpOverload(op='aten.special_psi', overload='default')>),
     <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 0x7fe9be701f80>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718540>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>),
     <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720c20>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>),
     <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf1b3d80>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718cc0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>),
     <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb8ea0>, kernel=<OpOverload(op='aten._version', overload='default')>),
     <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71af20>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>),
     <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b5da0>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>),
     <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb9c0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>),
     <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e0900>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>),
     <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e0220>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>),
     <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cfe0>, kernel=<OpOverload(op='aten.diag', overload='default')>),
     <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8af380>, kernel=<OpOverload(op='aten.clip', overload='default')>),
     <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746ac0>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747d80>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746660>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>),
     <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ad120>, kernel=<OpOverload(op='aten.square', overload='default')>),
     <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7feada2c4860>,
     <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7199e0>, 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 0x7fe9be718220>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>),
     <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7feada2c5c60>,
     <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7feada2c5c60>,
     <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7feada2c5f80>,
     <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7feada2c5f80>,
     <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709940>, 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 0x7fe9be70b6a0>, kernel=<OpOverload(op='aten.pdist', overload='default')>),
     <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7feada2c68e0>,
     <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7feada2c7c40>,
     <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7feada2c7ec0>,
     <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7feada0e4180>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7feada0e4c20>,
     <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7feada0e4c20>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7feada0e6340>,
     <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7feada0e6340>,
     <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7feada0e6020>,
     <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7feada0e6020>,
     <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8da80>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>),
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718ea0>, 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 0x7fe9be719e40>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>),
     <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7feada3c6c00>,
     <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7feada3c6c00>,
     <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c7c0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>),
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf474540>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>),
     <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7feada3c7560>,
     <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7feada3c7560>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7feada1f2340>,
     <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7feada1f2340>,
     <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7feada1f28e0>,
     <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7feada1f2840>,
     <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7feada1f2980>,
     <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7feada1f28e0>,
     <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7feada1f36a0>,
     <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7feada2214e0>,
     <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7feada2214e0>,
     <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7feada220720>,
     <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7feada220c20>,
     <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7feada220cc0>,
     <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7feada220a40>,
     <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7feada1f3ce0>,
     <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7feada2218a0>,
     <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7feada221da0>,
     <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7feada222b60>,
     <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7feada2222a0>,
     <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function PyCapsule.transpose at 0x7feada2237e0>,
     <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7feada222700>,
     <OpOverload(op='aten.transpose_copy', overload='int')>: <function PyCapsule.transpose at 0x7feada2237e0>,
     <OpOverload(op='aten.alias_copy', overload='default')>: <function PyCapsule.alias at 0x7feada222200>,
     <OpOverload(op='aten.alias_copy', overload='out')>: <function PyCapsule.alias at 0x7feada222200>,
     <OpOverload(op='aten.as_strided_copy', overload='default')>: <function PyCapsule.as_strided at 0x7feada221d00>,
     <OpOverload(op='aten.as_strided_copy', overload='out')>: <function PyCapsule.as_strided at 0x7feada221d00>,
     <OpOverload(op='aten.expand_copy', overload='default')>: <function PyCapsule.expand at 0x7feada220900>,
     <OpOverload(op='aten.expand_copy', overload='out')>: <function PyCapsule.expand at 0x7feada220900>,
     <OpOverload(op='aten.narrow_copy', overload='default')>: <function PyCapsule.narrow at 0x7feada221300>,
     <OpOverload(op='aten.narrow_copy', overload='out')>: <function PyCapsule.narrow at 0x7feada221300>,
     <OpOverload(op='aten.squeeze_copy', overload='default')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.permute_copy', overload='default')>: <function PyCapsule.permute at 0x7feada2232e0>,
     <OpOverload(op='aten.permute_copy', overload='out')>: <function PyCapsule.permute at 0x7feada2232e0>,
     <OpOverload(op='aten.t_copy', overload='default')>: <function PyCapsule.t at 0x7feada223560>,
     <OpOverload(op='aten.t_copy', overload='out')>: <function PyCapsule.t at 0x7feada223560>,
     <OpOverload(op='aten.unbind_copy', overload='int')>: <function PyCapsule.unbind at 0x7feada223880>,
     <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function PyCapsule.unbind at 0x7feada223880>,
     <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function PyCapsule.unsqueeze at 0x7feada221580>,
     <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function PyCapsule.unsqueeze at 0x7feada221580>,
     <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf6eee80>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>),
     <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70aca0>, 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 0x7fe9be7196c0>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>),
     <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723ce0>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>),
     <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721c60>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>),
     <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721300>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>),
     <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ca40>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>),
     <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d580>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>),
     <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dda0>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>),
     <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a840>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>),
     <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7459e0>, 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 0x7fe9be7467a0>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>),
     <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745b20>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>),
     <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745260>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>),
     <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744ae0>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>),
     <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722ac0>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>),
     <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721760>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>),
     <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7211c0>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>),
     <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7213a0>, 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 0x7fe9be7216c0>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>),
     <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7feada29b1a0>,
     <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7feada29b1a0>,
     <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7feada2c4540>,
     <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7feada2c4860>,
     <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7feada2c4540>,
     <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d9e0>, kernel=<OpOverload(op='aten.is_complex', overload='default')>),
     <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7feada2eb9c0>,
     <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7feada2eb9c0>,
     <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7feada2fa0c0>,
     <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7feada2fa0c0>,
     <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7feada311a80>,
     <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7feada311a80>,
     <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7477e0>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>),
     <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7462a0>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>),
     <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7463e0>, 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 0x7fe9be745e40>, kernel=<OpOverload(op='aten.std', overload='dim')>),
     <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744d60>, kernel=<OpOverload(op='aten.std', overload='names_dim')>),
     <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7453a0>, kernel=<OpOverload(op='aten.std', overload='correction_names')>),
     <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71aa20>, kernel=<OpOverload(op='aten.std_mean', overload='default')>),
     <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a980>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>),
     <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b100>, 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 0x7fe9be703ec0>, 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 0x7fe9be702520>, 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 0x7fe9be702ca0>, kernel=<OpOverload(op='aten.var_mean', overload='correction_names')>),
     <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7feada37a8e0>,
     <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7feada37ba60>,
     <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7feada37ad40>,
     <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7feada37ad40>,
     <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7feada37a160>,
     <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7feada39d1c0>,
     <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7feada39d1c0>,
     <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7feada39db20>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c89a0>, 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 0x7fea23f8d120>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>),
     <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7feada39ede0>,
     <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7feada39ede0>,
     <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7feada39f1a0>,
     <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7feada39f1a0>,
     <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7feada39fb00>,
     <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7feada39fb00>,
     <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be732ca0>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>),
     <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7feada5b7ba0>,
     <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7feada5b7880>,
     <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7feada5b7880>,
     <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7feada5b7b00>,
     <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7feada5b7b00>,
     <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7feada3e0ae0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7feada3e0ae0>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7feada3e0b80>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7feada3e0b80>,
     <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7feada3e0e00>,
     <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7feada3e18a0>,
     <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7feada3e18a0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7feada3e1c60>,
     <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7feada3e1d00>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7feada3e1f80>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7feada3e2020>,
     <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7feada3e27a0>,
     <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7feada3e2840>,
     <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7feada3e2980>,
     <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7feada3e3240>,
     <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7feada3e3240>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7feada3e3ec0>,
     <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7feada40c0e0>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7feada3e3ec0>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7feada40c180>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7feada3e37e0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7feada3e2b60>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7feada3e2b60>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7feada40c7c0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7feada40c7c0>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7feada40cc20>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7feada40cc20>,
     <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7feada40cf40>,
     <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7feada40cf40>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7feada40db20>,
     <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7feada40d1c0>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7feada40db20>,
     <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7feada40d1c0>,
     <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7feada40cfe0>,
     <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7feada40d6c0>,
     <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7016c0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>),
     <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7feada40d620>,
     <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7feada40d620>,
     <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fba0>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>),
     <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7feada40d800>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7feada40e520>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7feada40f1a0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7feada40f240>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7feada40f240>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7feada40f880>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7feada40f880>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7feada40fec0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7feada40fec0>,
     <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7feada428720>,
     <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7feada428900>,
     <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7feada428ae0>,
     <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7feada428cc0>,
     <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7feada429120>,
     <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7feada429300>,
     <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7feada429620>,
     <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7feada429800>,
     <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7feada429da0>,
     <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7feada40e840>,
     <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7feada42a160>,
     <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7feada40e840>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7feada42a340>,
     <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7feada42a480>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7feada42b380>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7feada42b380>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7feada42bd80>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7feada42bd80>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4514e0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4514e0>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7feada40eac0>,
     <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7feada40eac0>,
     <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7feada450540>,
     <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7feada450540>,
     <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7feada4520c0>,
     <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7feada4520c0>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7feada4527a0>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7feada4527a0>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7feada453380>,
     <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7feada453380>,
     <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7feada453f60>,
     <OpOverload(op='aten.take', overload='default')>: <function take at 0x7feada452d40>,
     <OpOverload(op='aten.take', overload='out')>: <function take at 0x7feada452d40>,
     <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453f60>,
     <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7feada4532e0>,
     <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7feada4532e0>,
     <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada452a20>,
     <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada452020>,
     <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada451bc0>,
     <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada450680>,
     <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada453880>,
     <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada453880>,
     <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4539c0>,
     <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ba0>,
     <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453b00>,
     <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453c40>,
     <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453d80>,
     <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ec0>,
     <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ec0>,
     <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478040>,
     <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478040>,
     <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4782c0>,
     <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478400>,
     <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478400>,
     <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478540>,
     <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478540>,
     <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478680>,
     <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478680>,
     <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4787c0>,
     <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478900>,
     <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478a40>,
     <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478b80>,
     <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7feada453920>,
     <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4519e0>,
     <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e340>, kernel=<OpOverload(op='aten.var', overload='default')>),
     <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747920>, kernel=<OpOverload(op='aten.var', overload='dim')>),
     <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70bf60>, kernel=<OpOverload(op='aten.var', overload='correction_names')>),
     <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ade40>, kernel=<OpOverload(op='aten.var', overload='names_dim')>),
     <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7feada40d120>,
     <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7feada40d120>,
     <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7feada56e3e0>,
     <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7feada56c720>,
     <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7feada56c720>,
     <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7feada56cb80>,
     <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7feada56cb80>,
     <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7feada56c680>,
     <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7feada56c680>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7feada56f600>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7feada56d800>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7feada56d800>,
     <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7feada56e480>,
     <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7feada56e480>,
     <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7feada56e520>,
     <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7feada56e520>,
     <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7feada56e3e0>,
     <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7feada56d300>,
     <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7feada56d300>,
     <OpOverload(op='aten.mish_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708540>, kernel=<OpOverload(op='aten.mish_backward', overload='default')>),
     <OpOverload(op='aten.silu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a840>, kernel=<OpOverload(op='aten.silu_backward', overload='default')>),
     <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7feada56ede0>,
     <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7feada56ea20>,
     <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7feada56f1a0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7feada56f6a0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7feada56f6a0>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7feada56f740>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7feada56f740>,
     <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7feada56c9a0>,
     <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7feada56efc0>,
     <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7feada56ff60>,
     <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7feada56ff60>,
     <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7feada56ea20>,
     <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7feada5b42c0>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7feada5b47c0>,
     <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7feada5b4720>,
     <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7feada5b4720>,
     <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7feada5b4180>,
     <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7feada5b4180>,
     <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7feada5b4b80>,
     <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7feada5b4b80>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7feada5b4ea0>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7feada5b4ea0>,
     <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7feada5b5440>,
     <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7feada5b5440>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7feada5b54e0>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7feada5b54e0>,
     <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7feada5b5b20>,
     <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7feada5b5b20>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7feada5b5bc0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7feada5b5bc0>,
     <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7feada5b6160>,
     <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7feada5b6160>,
     <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7feada5b63e0>,
     <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7feada5b63e0>,
     <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7feada5b49a0>,
     <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7feada5b49a0>,
     <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7feada5b67a0>,
     <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7feada5b67a0>,
     <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7feada5b68e0>,
     <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7feada5b68e0>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7feada5b6f20>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7feada5b6f20>,
     <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7feada937ba0>,
     <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709440>, kernel=<OpOverload(op='aten.where', overload='default')>),
     <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7feada937ba0>,
     <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb1a0>, kernel=<OpOverload(op='aten.svd', overload='default')>),
     <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7feada452e80>,
     <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d260>, 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 0x7fea23f8ee80>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>),
     <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478180>,
     <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898400>, 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 0x7fe9be89b600>, 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 0x7fe9be8e2660>, 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 0x7fe9be8e13a0>, 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 0x7fe9be89bba0>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899da0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898a40>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899260>, 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 0x7fe9be89ba60>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898540>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>),
     <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e2520>, kernel=<OpOverload(op='quantized.conv1d_unpack', 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 0x7fe9be8e00e0>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e1d00>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e2de0>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e22a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a2a0>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e1300>, 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 0x7fe9be8e11c0>, kernel=<OpOverload(op='quantized.conv2d_groups', 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 0x7fe9be89a7a0>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>),
     <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e1440>, 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 0x7fe9be8e2e80>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b7e0>, 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 0x7fe9be89b420>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>),
     <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899ee0>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8987c0>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>),
     <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899a80>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>),
     <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b100>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>),
     <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898ea0>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a520>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898680>, 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 0x7fe9be899f80>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b4c0>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>),
     <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899b20>, kernel=<OpOverload(op='sparse.qlinear_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 0x7fe9be8991c0>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b240>, kernel=<OpOverload(op='quantized.conv2d_output_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 0x7fe9be89a660>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>),
     <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e19e0>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}
experimental_experiment.torch_dynamo.get_decomposition_table_onnxscript()[source]

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='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8eb60>, kernel=<OpOverload(op='image.decode_png', overload='default')>),
     <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dee0>, kernel=<OpOverload(op='image.read_file', overload='default')>),
     <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ede0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>),
     <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e2f20>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>),
     <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e19e0>, kernel=<OpOverload(op='torchvision._cuda_version', 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 0x7fea23f8d1c0>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>),
     <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e0c0>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>),
     <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f600>, kernel=<OpOverload(op='image.encode_png', overload='default')>),
     <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ea20>, kernel=<OpOverload(op='image.decode_gif', overload='default')>),
     <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e340>, kernel=<OpOverload(op='image.decode_webp', overload='default')>),
     <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e11c0>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>),
     <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e0900>, kernel=<OpOverload(op='image.write_file', overload='default')>),
     <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dc60>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>),
     <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e1300>, kernel=<OpOverload(op='image.decode_image', overload='default')>),
     <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7468e0>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>),
     <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745d00>, kernel=<OpOverload(op='aten.chain_matmul', 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 0x7fe9be898400>, 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 0x7fe9be899f80>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>),
     <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 0x7fe9be89b600>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899d00>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>),
     <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8acfe0>, kernel=<OpOverload(op='aten.argsort', overload='stable')>),
     <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fc40>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7218a0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>),
     <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf474900>, kernel=<OpOverload(op='aten.arctan2', overload='default')>),
     <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf474540>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>),
     <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723ba0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>),
     <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721da0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720fe0>, kernel=<OpOverload(op='aten.linalg_matrix_power', 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 0x7fe9be8c9b20>, 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 0x7fe9be720400>, 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 0x7fe9be8e0220>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>),
     <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7313a0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>),
     <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bd80>, 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 0x7fe9be745260>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>),
     <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898900>, kernel=<OpOverload(op='aten.special_gammaincc', 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 0x7fe9be718f40>, 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 0x7fe9be71a0c0>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>),
     <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a660>, 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 0x7fe9be708cc0>, 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 0x7fead83c9f80>, 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 0x7fe9be708ea0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>),
     <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7476a0>, 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 0x7fe9be708b80>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>),
     <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746020>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>),
     <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7080e0>, kernel=<OpOverload(op='aten.combinations', overload='default')>),
     <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747ec0>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746d40>, kernel=<OpOverload(op='aten.inverse', overload='default')>),
     <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746ac0>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb420>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>),
     <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca5c0>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>),
     <OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c84a0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>),
     <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be6f89a0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', 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 0x7fe9be8c96c0>, 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 0x7fe9bf12ade0>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>),
     <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf1298a0>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>),
     <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf128680>, kernel=<OpOverload(op='aten.promote_types', overload='default')>),
     <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf128ea0>, kernel=<OpOverload(op='aten.negative', overload='default')>),
     <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf129120>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b100>, kernel=<OpOverload(op='aten.histogramdd', 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 0x7fe9be8e2520>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>),
     <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700c20>, kernel=<OpOverload(op='aten.kl_div', overload='default')>),
     <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf2384a0>, kernel=<OpOverload(op='aten.arcsin', overload='default')>),
     <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700720>, 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 0x7fe9be8b7920>, kernel=<OpOverload(op='aten.matrix_exp', 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 0x7fe9be700860>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>),
     <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701f80>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>),
     <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7020c0>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>),
     <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701d00>, kernel=<OpOverload(op='aten.trapz', overload='dx')>),
     <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d120>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>),
     <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fb00>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>),
     <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e660>, 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 0x7fea23f8d940>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>),
     <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f1a0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ca40>, 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 0x7fea23f8dd00>, 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 0x7fea23f8d260>, 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 0x7fe9be8b44a0>, 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 0x7fe9be8b77e0>, 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 0x7fe9be8b4400>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>),
     <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b7ec0>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', 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 0x7fe9be8cbe20>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>),
     <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c89a0>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>),
     <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb880>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>),
     <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746980>, kernel=<OpOverload(op='aten.orgqr', overload='default')>),
     <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899800>, kernel=<OpOverload(op='aten.stride', overload='int')>),
     <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 0x7fe9be7449a0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>),
     <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb9a80>, kernel=<OpOverload(op='aten.quantile', overload='default')>),
     <OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c7c0>, 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 0x7fea23f8cae0>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a200>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>),
     <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8996c0>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a520>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>),
     <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70af20>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>),
     <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf6ed760>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>),
     <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722480>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf476ca0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>),
     <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7202c0>, kernel=<OpOverload(op='aten.special_logit', overload='default')>),
     <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12bb00>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>),
     <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e13a0>, kernel=<OpOverload(op='_test.get_first', overload='default')>),
     <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898ea0>, kernel=<OpOverload(op='aten.absolute', overload='default')>),
     <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744360>, kernel=<OpOverload(op='aten.take_along_dim', 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 0x7fea23f8f2e0>, 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 0x7fe9be720720>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>),
     <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745b20>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>),
     <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b600>, kernel=<OpOverload(op='aten.inner', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7194e0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>),
     <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709620>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>),
     <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7098a0>, kernel=<OpOverload(op='aten.median', overload='names_dim')>),
     <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744d60>, kernel=<OpOverload(op='aten._lu_with_info', 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 0x7fea23f8e520>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>),
     <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ae5c0>, kernel=<OpOverload(op='aten._remove_batch_dim', overload='default')>),
     <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8a28e0>, kernel=<OpOverload(op='aten.less', overload='Scalar')>),
     <OpOverload(op='aten.special_xlogy', overload='other_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746520>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>),
     <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746de0>, kernel=<OpOverload(op='aten.vander', overload='default')>),
     <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb8ea0>, kernel=<OpOverload(op='aten.concat', overload='names')>),
     <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8a0540>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>),
     <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf12a980>, kernel=<OpOverload(op='aten.thnn_conv2d', 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 0x7fe9be898180>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709580>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', 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 0x7fe9be899580>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>),
     <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899940>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>),
     <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a020>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>),
     <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720c20>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>),
     <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723920>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>),
     <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7205e0>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>),
     <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723ce0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>),
     <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722d40>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', 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 0x7fe9be7227a0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7211c0>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>),
     <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720860>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>),
     <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 0x7fe9be733240>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>),
     <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be730860>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>),
     <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719b20>, kernel=<OpOverload(op='aten.gather', overload='dimname')>),
     <OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719260>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718cc0>, kernel=<OpOverload(op='aten.bilinear', 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 0x7fe9be70ba60>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>),
     <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70bd80>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>),
     <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 0x7fe9be708540>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b6a0>, 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 0x7fe9be708680>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>),
     <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747240>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>),
     <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747420>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>),
     <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a5c0>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>),
     <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745760>, kernel=<OpOverload(op='aten.arccos', overload='default')>),
     <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7472e0>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>),
     <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70afc0>, kernel=<OpOverload(op='aten.ger', overload='default')>),
     <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf1b3d80>, kernel=<OpOverload(op='aten.data', overload='default')>),
     <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ac540>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>),
     <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a840>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>),
     <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898360>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>),
     <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b420>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721c60>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>),
     <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b240>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>),
     <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ae980>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>),
     <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89afc0>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>),
     <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8acf40>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>),
     <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720900>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>),
     <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7216c0>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722340>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>),
     <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722ac0>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>),
     <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722700>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>),
     <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7234c0>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>),
     <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be731760>, kernel=<OpOverload(op='aten._backward', overload='default')>),
     <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7305e0>, kernel=<OpOverload(op='aten._reshape_from_tensor', 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 0x7fe9be7319e0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b920>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719620>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>),
     <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c8220>, kernel=<OpOverload(op='aten.is_leaf', overload='default')>),
     <OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71aa20>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71ac00>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>),
     <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708180>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>),
     <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ed40>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>),
     <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708400>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>),
     <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898e00>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>),
     <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70ab60>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>),
     <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708900>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>),
     <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7462a0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>),
     <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70be20>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>),
     <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709e40>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>),
     <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747ce0>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>),
     <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7463e0>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cbec0>, kernel=<OpOverload(op='aten.arctan', overload='default')>),
     <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c99e0>, kernel=<OpOverload(op='aten.gradient', overload='array')>),
     <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca160>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>),
     <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb7e0>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>),
     <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7479c0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>),
     <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700680>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>),
     <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf23ae80>, kernel=<OpOverload(op='aten.trapz', overload='x')>),
     <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf128900>, kernel=<OpOverload(op='aten.nanmean', overload='default')>),
     <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7037e0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>),
     <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702520>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>),
     <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702c00>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>),
     <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e1d00>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>),
     <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be703ec0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>),
     <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8eca0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>),
     <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b5080>, kernel=<OpOverload(op='aten.fix', 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 0x7fea23f8d080>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>),
     <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ec00>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>),
     <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f560>, kernel=<OpOverload(op='aten.vstack', overload='default')>),
     <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dbc0>, kernel=<OpOverload(op='aten.special_round', overload='default')>),
     <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d3a0>, kernel=<OpOverload(op='aten.msort', overload='default')>),
     <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8dda0>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>),
     <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e2e80>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>),
     <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e5c0>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>),
     <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9300>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>),
     <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fd80>, 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 0x7fe9be8e0860>, kernel=<OpOverload(op='_test.cat', overload='default')>),
     <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b6200>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>),
     <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9620>, kernel=<OpOverload(op='aten.kron', overload='default')>),
     <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9760>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b5260>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>),
     <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca2a0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>),
     <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9f80>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>),
     <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb600>, kernel=<OpOverload(op='aten.diagflat', 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 0x7fe9be8c8180>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>),
     <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8caf20>, kernel=<OpOverload(op='aten.to_dense', overload='default')>),
     <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb100>, 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 0x7fe9be8c9120>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a2a0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>),
     <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718ae0>, kernel=<OpOverload(op='aten.argwhere', overload='default')>),
     <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb4c0>, kernel=<OpOverload(op='aten.is_inference', overload='default')>),
     <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a340>, kernel=<OpOverload(op='aten.is_conj', overload='default')>),
     <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be6faac0>, kernel=<OpOverload(op='aten.row_stack', overload='default')>),
     <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8caa20>, kernel=<OpOverload(op='aten.min', overload='names_dim')>),
     <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 0x7fe9bf1294e0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>),
     <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf128540>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>),
     <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899ee0>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>),
     <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701940>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>),
     <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf2387c0>, kernel=<OpOverload(op='aten.fliplr', overload='default')>),
     <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702de0>, kernel=<OpOverload(op='aten.one_hot', 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 0x7fe9be701080>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>),
     <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e3e0>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>),
     <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c040>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>),
     <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fe20>, kernel=<OpOverload(op='aten.cdist', overload='default')>),
     <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ce00>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>),
     <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d300>, kernel=<OpOverload(op='aten.chalf', overload='default')>),
     <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cfe0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>),
     <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ccc0>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>),
     <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ee80>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>),
     <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f7e0>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>),
     <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b6d40>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>),
     <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702f20>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>),
     <OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b49a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', 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 0x7fe9be8ca200>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>),
     <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c87c0>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>),
     <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 0x7fe9be8cb9c0>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>),
     <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745bc0>, kernel=<OpOverload(op='aten.masked_select_backward', 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 0x7fe9be747f60>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>),
     <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cf40>, kernel=<OpOverload(op='c10d_functional.broadcast', 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 0x7fe9be89bc40>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', 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 0x7fe9be899260>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>),
     <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b100>, kernel=<OpOverload(op='aten.pinverse', overload='default')>),
     <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8998a0>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>),
     <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8af740>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>),
     <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ac680>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>),
     <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746ca0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>),
     <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7207c0>, kernel=<OpOverload(op='aten.qr', overload='default')>),
     <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7232e0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>),
     <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723100>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', 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 0x7fe9be71ad40>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>),
     <OpOverload(op='profiler._record_function_exit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c860>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>),
     <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be730e00>, kernel=<OpOverload(op='aten.sort', overload='dimname')>),
     <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a980>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>),
     <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b240>, kernel=<OpOverload(op='aten.linalg_eigvalsh', 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 0x7fe9be719080>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', 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 0x7fead83c80e0>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719c60>, kernel=<OpOverload(op='aten.linalg_matrix_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 0x7fe9be71bc40>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>),
     <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a700>, kernel=<OpOverload(op='aten.special_i0', overload='default')>),
     <OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f240>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>),
     <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708040>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>),
     <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a3e0>, kernel=<OpOverload(op='aten.size', overload='int')>),
     <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b9c0>, kernel=<OpOverload(op='aten.l1_loss', 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 0x7fe9be70a8e0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>),
     <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746160>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>),
     <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b7e0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>),
     <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb920>, kernel=<OpOverload(op='aten.flipud', overload='default')>),
     <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b920>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>),
     <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be701da0>, kernel=<OpOverload(op='aten._nnpack_available', 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 0x7fe9be744b80>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>),
     <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fa60>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>),
     <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89ba60>, 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 0x7fe9be899da0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>),
     <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709b20>, kernel=<OpOverload(op='aten.is_neg', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be721620>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>),
     <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722b60>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>),
     <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf476520>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>),
     <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722c00>, kernel=<OpOverload(op='aten.diff', overload='default')>),
     <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7231a0>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>),
     <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720540>, kernel=<OpOverload(op='aten.special_expit', 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 0x7fea23f8fba0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>),
     <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a3e0>, kernel=<OpOverload(op='aten.argsort', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71a700>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>),
     <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b560>, kernel=<OpOverload(op='aten.stft', 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 0x7fe9be71ba60>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', 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 0x7fe9be71a340>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>),
     <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b560>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>),
     <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70a840>, kernel=<OpOverload(op='aten.output_nr', overload='default')>),
     <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70ad40>, 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 0x7fe9be745620>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>),
     <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718360>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>),
     <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7471a0>, kernel=<OpOverload(op='aten.arccosh', overload='default')>),
     <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744400>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>),
     <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8a0400>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>),
     <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a7a0>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>),
     <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bec0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>),
     <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8987c0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>),
     <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8af380>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>),
     <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8aeca0>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>),
     <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8afec0>, 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 0x7fe9be89ae80>, 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 0x7fe9bf6eee80>, 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 0x7fe9be723e20>, kernel=<OpOverload(op='aten.align_as', 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 0x7fe9be721760>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', 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 0x7fe9be8c9c60>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>),
     <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722020>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>),
     <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e020>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>),
     <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7327a0>, 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 0x7fe9be733e20>, 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 0x7fe9be718220>, 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 0x7fe9be7187c0>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>),
     <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719760>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>),
     <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719e40>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>),
     <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b380>, 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 0x7fe9be8ca980>, 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 0x7fe9be718540>, 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 0x7fea23f8fce0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>),
     <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c9e40>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>),
     <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c98a0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>),
     <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70b2e0>, 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 0x7fe9be70a0c0>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>),
     <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723f60>, 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 0x7fe9be70b920>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', 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 0x7fe9be70b060>, 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 0x7fe9be7093a0>, 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 0x7fe9be744860>, kernel=<OpOverload(op='aten._cast_Short', 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 0x7fea23f8de40>, 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 0x7fe9be709f80>, 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 0x7fe9be7458a0>, 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 0x7fe9be747380>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>),
     <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ade40>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>),
     <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722de0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>),
     <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a3e0>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>),
     <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7477e0>, kernel=<OpOverload(op='aten.size', overload='Dimname')>),
     <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70bce0>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>),
     <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8993a0>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>),
     <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be703380>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>),
     <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be720cc0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>),
     <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898fe0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>),
     <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7082c0>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>),
     <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bce0>, kernel=<OpOverload(op='aten.hstack', overload='default')>),
     <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7300e0>, kernel=<OpOverload(op='aten.is_signed', overload='default')>),
     <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745120>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>),
     <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be6fa2a0>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>),
     <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71b4c0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>),
     <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 0x7fe9be71af20>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718ea0>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>),
     <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8accc0>, kernel=<OpOverload(op='aten.outer', overload='default')>),
     <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708a40>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>),
     <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709940>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>),
     <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be731260>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>),
     <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898540>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>),
     <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c8360>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>),
     <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747920>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>),
     <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be71afc0>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>),
     <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745080>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>),
     <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca480>, kernel=<OpOverload(op='aten.isreal', overload='default')>),
     <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8ca7a0>, kernel=<OpOverload(op='aten.rrelu', overload='default')>),
     <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70aa20>, 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 0x7fe9be709800>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>),
     <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d4e0>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>),
     <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be732ca0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>),
     <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7196c0>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>),
     <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be898680>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>),
     <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89bba0>, kernel=<OpOverload(op='aten.special_psi', overload='default')>),
     <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 0x7fe9be89b380>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>),
     <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb8360>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>),
     <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c8e00>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>),
     <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf129440>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>),
     <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83ca0c0>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>),
     <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23fb8a40>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>),
     <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718c20>, kernel=<OpOverload(op='aten._version', overload='default')>),
     <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be746200>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>),
     <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709300>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>),
     <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be722a20>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>),
     <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e31a0>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>),
     <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3240>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>),
     <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7016c0>, kernel=<OpOverload(op='aten.diag', overload='default')>),
     <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be70bba0>, kernel=<OpOverload(op='aten.clip', overload='default')>),
     <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8da80>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d580>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>),
     <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f740>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>),
     <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709760>, kernel=<OpOverload(op='aten.square', overload='default')>),
     <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7feada2c4860>,
     <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744cc0>, 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 0x7fe9be898a40>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>),
     <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7feada2c58a0>,
     <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7feada2c5c60>,
     <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7feada2c5c60>,
     <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7feada2c5f80>,
     <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7feada2c5f80>,
     <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7feada2c5bc0>,
     <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7feada2c67a0>,
     <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7feada2c72e0>,
     <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745e40>, 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 0x7fe9be747ba0>, kernel=<OpOverload(op='aten.pdist', overload='default')>),
     <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7feada2c7ba0>,
     <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7feada2c74c0>,
     <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7feada2c7e20>,
     <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7feada2c68e0>,
     <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7feada2c7c40>,
     <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7feada2c7ec0>,
     <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7feada0e4040>,
     <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7feada0e4180>,
     <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7feada2c74c0>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7feada0e4220>,
     <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7feada0e4220>,
     <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7feada0e4c20>,
     <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7feada0e4c20>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7feada0e4fe0>,
     <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7feada0e5580>,
     <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7feada0e5580>,
     <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7feada0e5a80>,
     <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7feada0e5a80>,
     <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7feada0e5f80>,
     <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7feada0e5f80>,
     <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7feada0e6340>,
     <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7feada0e6340>,
     <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7feada0e6700>,
     <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7feada0e6ac0>,
     <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7feada0e6020>,
     <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7feada0e6020>,
     <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be702b60>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>),
     <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7feada0e6c00>,
     <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7feada0e6fc0>,
     <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7feada0e6fc0>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7feada0e7600>,
     <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7feada0e7600>,
     <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7feada0e7a60>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be744f40>, 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 0x7fe9be744900>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>),
     <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7feada3c6f20>,
     <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7feada3c6f20>,
     <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7feada3c6de0>,
     <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7feada3c6c00>,
     <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7feada3c6c00>,
     <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7193a0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>),
     <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be718e00>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>),
     <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7feada3c72e0>,
     <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7feada3c7560>,
     <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7feada3c7560>,
     <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7feada3c7a60>,
     <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7feada3c7ce0>,
     <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7feada1f00e0>,
     <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7feada1f00e0>,
     <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7feada3c54e0>,
     <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7feada1f18a0>,
     <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7feada1f02c0>,
     <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7feada1f02c0>,
     <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7feada1f0680>,
     <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7feada1f0680>,
     <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7feada1f0a40>,
     <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7feada1f0a40>,
     <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7feada1f0e00>,
     <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7feada1f0e00>,
     <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7feada1f0ea0>,
     <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7feada1f1800>,
     <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7feada1f1d00>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7feada1f1f80>,
     <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7feada1f2340>,
     <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7feada1f2340>,
     <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7feada1f2520>,
     <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7feada1f28e0>,
     <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7feada1f2020>,
     <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7feada1f22a0>,
     <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7feada1f25c0>,
     <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7feada1f2700>,
     <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7feada1f2700>,
     <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7feada1f3ba0>,
     <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7feada1f2840>,
     <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7feada1f3ba0>,
     <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7feada1f2980>,
     <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7feada1f2ac0>,
     <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7feada1f0220>,
     <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7feada1f2c00>,
     <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7feada1f3e20>,
     <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7feada1f2d40>,
     <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7feada1f2e80>,
     <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7feada1f2fc0>,
     <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7feada1f3100>,
     <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7feada1f3100>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7feada1f3240>,
     <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7feada1f3240>,
     <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7feada1f3f60>,
     <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7feada1f3380>,
     <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7feada1f34c0>,
     <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7feada1f34c0>,
     <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7feada1f28e0>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7feada1f3600>,
     <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7feada1f3600>,
     <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7feada1f3740>,
     <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7feada1f3740>,
     <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7feada1f3ec0>,
     <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7feada1f3880>,
     <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7feada1f3ec0>,
     <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7feada1f39c0>,
     <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7feada1f39c0>,
     <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7feada1f3b00>,
     <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7feada1f3b00>,
     <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7feada1f3920>,
     <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7feada1f3c40>,
     <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7feada1f3c40>,
     <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7feada1f3d80>,
     <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7feada1f3920>,
     <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7feada1f36a0>,
     <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7feada1f3420>,
     <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7feada2214e0>,
     <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7feada1f31a0>,
     <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7feada1f2f20>,
     <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7feada1f2f20>,
     <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7feada2214e0>,
     <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7feada1f2ca0>,
     <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7feada1f2a20>,
     <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7feada1f27a0>,
     <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7feada1f2480>,
     <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7feada1f1da0>,
     <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7feada2200e0>,
     <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7feada220220>,
     <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7feada220220>,
     <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7feada1f3a60>,
     <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7feada220360>,
     <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7feada2204a0>,
     <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7feada2204a0>,
     <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7feada2205e0>,
     <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7feada2205e0>,
     <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7feada220720>,
     <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7feada1f3560>,
     <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7feada220860>,
     <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7feada2209a0>,
     <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7feada2209a0>,
     <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7feada220ae0>,
     <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7feada220ae0>,
     <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7feada1f3060>,
     <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7feada220c20>,
     <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7feada220d60>,
     <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7feada220ea0>,
     <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7feada1f2b60>,
     <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7feada220fe0>,
     <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7feada221120>,
     <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7feada1f2660>,
     <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7feada221260>,
     <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7feada1f32e0>,
     <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7feada2213a0>,
     <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7feada2213a0>,
     <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7feada221620>,
     <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7feada2216c0>,
     <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7feada221440>,
     <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7feada2211c0>,
     <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7feada2211c0>,
     <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7feada220f40>,
     <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7feada220f40>,
     <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7feada1f37e0>,
     <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7feada1f37e0>,
     <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7feada220cc0>,
     <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7feada220a40>,
     <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7feada2207c0>,
     <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7feada2207c0>,
     <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7feada1f3ce0>,
     <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7feada220540>,
     <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7feada2202c0>,
     <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7feada220040>,
     <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7feada220040>,
     <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7feada2218a0>,
     <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7feada2219e0>,
     <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7feada221b20>,
     <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7feada221b20>,
     <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7feada221c60>,
     <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7feada221da0>,
     <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7feada221ee0>,
     <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7feada222b60>,
     <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7feada222020>,
     <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7feada222160>,
     <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7feada2222a0>,
     <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7feada222c00>,
     <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7feada2223e0>,
     <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7feada222520>,
     <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7feada222660>,
     <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7feada2227a0>,
     <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7feada2227a0>,
     <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7feada222e80>,
     <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7feada2228e0>,
     <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7feada222a20>,
     <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7feada222de0>,
     <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7feada222980>,
     <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function PyCapsule.transpose at 0x7feada2237e0>,
     <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7feada222700>,
     <OpOverload(op='aten.transpose_copy', overload='int')>: <function PyCapsule.transpose at 0x7feada2237e0>,
     <OpOverload(op='aten.alias_copy', overload='default')>: <function PyCapsule.alias at 0x7feada222200>,
     <OpOverload(op='aten.alias_copy', overload='out')>: <function PyCapsule.alias at 0x7feada222200>,
     <OpOverload(op='aten.as_strided_copy', overload='default')>: <function PyCapsule.as_strided at 0x7feada221d00>,
     <OpOverload(op='aten.as_strided_copy', overload='out')>: <function PyCapsule.as_strided at 0x7feada221d00>,
     <OpOverload(op='aten.diagonal_copy', overload='out')>: <function PyCapsule.diagonal at 0x7feada221800>,
     <OpOverload(op='aten.expand_copy', overload='default')>: <function PyCapsule.expand at 0x7feada220900>,
     <OpOverload(op='aten.expand_copy', overload='out')>: <function PyCapsule.expand at 0x7feada220900>,
     <OpOverload(op='aten.narrow_copy', overload='default')>: <function PyCapsule.narrow at 0x7feada221300>,
     <OpOverload(op='aten.narrow_copy', overload='out')>: <function PyCapsule.narrow at 0x7feada221300>,
     <OpOverload(op='aten.squeeze_copy', overload='default')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function PyCapsule.squeeze at 0x7feada223060>,
     <OpOverload(op='aten.permute_copy', overload='default')>: <function PyCapsule.permute at 0x7feada2232e0>,
     <OpOverload(op='aten.permute_copy', overload='out')>: <function PyCapsule.permute at 0x7feada2232e0>,
     <OpOverload(op='aten.t_copy', overload='default')>: <function PyCapsule.t at 0x7feada223560>,
     <OpOverload(op='aten.t_copy', overload='out')>: <function PyCapsule.t at 0x7feada223560>,
     <OpOverload(op='aten.unbind_copy', overload='int')>: <function PyCapsule.unbind at 0x7feada223880>,
     <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function PyCapsule.unbind at 0x7feada223880>,
     <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function PyCapsule.unsqueeze at 0x7feada221580>,
     <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function PyCapsule.unsqueeze at 0x7feada221580>,
     <OpOverload(op='aten.view_copy', overload='dtype')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.view_copy', overload='out')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function PyCapsule.view at 0x7feada2220c0>,
     <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7feada268a40>,
     <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7feada268cc0>,
     <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7453a0>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>),
     <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7feada269760>,
     <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be747d80>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>),
     <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7feada2699e0>,
     <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745440>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>),
     <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7feada269800>,
     <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb1a0>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>),
     <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7feada2689a0>,
     <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c8ea0>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>),
     <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7feada269c60>,
     <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9ee0>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>),
     <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7feada269ee0>,
     <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700040>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>),
     <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7feada26a7a0>,
     <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7027a0>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>),
     <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7feada26aa20>,
     <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7feada26bb00>,
     <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be703060>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>),
     <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7feada26aca0>,
     <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8efc0>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>),
     <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7feada26af20>,
     <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b7a60>, 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 0x7fea23f8eac0>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>),
     <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7feada26b600>,
     <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b4720>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>),
     <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7feada26b880>,
     <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b79c0>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>),
     <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7feada26a980>,
     <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b4e00>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>),
     <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7feada268c20>,
     <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb380>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>),
     <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7feada26bc40>,
     <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c98a0>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>),
     <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7feada26bec0>,
     <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8c9e40>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>),
     <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7feada298180>,
     <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8cb560>, 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 0x7fe9bf12bec0>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>),
     <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7feada298a40>,
     <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7feada298e00>,
     <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7feada299da0>,
     <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7feada29a020>,
     <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7feada29a700>,
     <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7feada29ab60>,
     <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7feada29b1a0>,
     <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7feada29b1a0>,
     <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7feada29b380>,
     <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7feada29b100>,
     <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7feada2c4540>,
     <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7feada29be20>,
     <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7feada2c4860>,
     <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7feada2c4540>,
     <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9bf23b420>, kernel=<OpOverload(op='aten.is_complex', overload='default')>),
     <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7feada2e9f80>,
     <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7feada2e9f80>,
     <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7feada2f8e00>,
     <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7feada2eb9c0>,
     <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7feada2eb9c0>,
     <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7feada2f8400>,
     <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7feada2e99e0>,
     <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7feada2f9120>,
     <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7feada2f9620>,
     <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7feada2fb4c0>,
     <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7feada2fa020>,
     <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7feada2fa020>,
     <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7feada2fb4c0>,
     <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723420>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>),
     <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7feada2fa0c0>,
     <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7feada2fa0c0>,
     <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7feada311580>,
     <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7feada311a80>,
     <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7feada311a80>,
     <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7feada312e80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7feada325f80>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7feada3256c0>,
     <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7feada326160>,
     <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7feada3413a0>,
     <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7feada3437e0>,
     <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7feada342660>,
     <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7feada342660>,
     <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7feada342f20>,
     <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7feada343380>,
     <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7feada342160>,
     <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7feada3439c0>,
     <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7feada343e20>,
     <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7feada361f80>,
     <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7feada3623e0>,
     <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7feada362b60>,
     <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7feada362f20>,
     <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7feada3632e0>,
     <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7feada3632e0>,
     <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7feada363560>,
     <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7feada363560>,
     <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7feada3637e0>,
     <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7feada3637e0>,
     <OpOverload(op='aten.all', overload='out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.all', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be700ea0>, kernel=<OpOverload(op='aten.all', overload='dimname')>),
     <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7feada3787c0>,
     <OpOverload(op='aten.any', overload='out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f420>, kernel=<OpOverload(op='aten.any', overload='dimname')>),
     <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7feada378a40>,
     <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cc20>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>),
     <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7feada379c60>,
     <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c180>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>),
     <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f4c0>, 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 0x7fe9be8b7ce0>, kernel=<OpOverload(op='aten.std', overload='dim')>),
     <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b5e40>, kernel=<OpOverload(op='aten.std', overload='names_dim')>),
     <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8b5da0>, kernel=<OpOverload(op='aten.std', overload='correction_names')>),
     <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7feada379580>,
     <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7099e0>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>),
     <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7feada3793a0>,
     <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89aca0>, kernel=<OpOverload(op='aten.std_mean', overload='default')>),
     <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8999e0>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>),
     <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899300>, 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 0x7fe9be89b7e0>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>),
     <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.var_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89ade0>, 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 0x7fe9be89b4c0>, kernel=<OpOverload(op='aten.var_mean', overload='correction_names')>),
     <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7feada3789a0>,
     <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7feada37a200>,
     <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7feada37a8e0>,
     <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7feada37ba60>,
     <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7feada37b2e0>,
     <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7feada37bb00>,
     <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7feada39c540>,
     <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7feada37ad40>,
     <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7feada37ad40>,
     <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7feada37a160>,
     <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7feada39c9a0>,
     <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7feada39cf40>,
     <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7feada39d1c0>,
     <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7feada39d1c0>,
     <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7feada39d4e0>,
     <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7feada39db20>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7219e0>, 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 0x7fe9be730360>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>),
     <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7feada39e160>,
     <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7feada39df80>,
     <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7feada39cea0>,
     <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7feada39e7a0>,
     <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7087c0>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>),
     <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7feada39e7a0>,
     <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7feada39eac0>,
     <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7feada39cae0>,
     <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7feada39ede0>,
     <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7feada39ede0>,
     <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7feada39f1a0>,
     <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7feada39f1a0>,
     <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7feada39fb00>,
     <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7feada39fb00>,
     <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c900>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>),
     <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7feada39fba0>,
     <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7feada39fba0>,
     <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89a660>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>),
     <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7feada39fc40>,
     <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7feada3c5580>,
     <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7feada3c6700>,
     <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7feada3c5940>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7feada3c5bc0>,
     <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7feada3c5e40>,
     <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7feada5b7ba0>,
     <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7feada5b7880>,
     <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7feada5b7880>,
     <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7feada5b7b00>,
     <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7feada5b7b00>,
     <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7feada5b6e80>,
     <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7feada5b5a80>,
     <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7feada3e04a0>,
     <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7feada3e0ae0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7feada3e07c0>,
     <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7feada3e0ae0>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7feada3e0b80>,
     <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7feada3e0b80>,
     <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7feada3e0e00>,
     <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7feada3e1580>,
     <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7feada3e18a0>,
     <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7feada3e18a0>,
     <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7feada3e1c60>,
     <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7feada3e1d00>,
     <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7feada3e1bc0>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7feada3e1f80>,
     <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7feada3e2020>,
     <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7feada3e0ea0>,
     <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7feada3e27a0>,
     <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7feada3e2840>,
     <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7feada3e2980>,
     <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7feada3e3240>,
     <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7feada3e3240>,
     <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7feada3e3600>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7feada3e3ec0>,
     <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7feada40c0e0>,
     <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7feada3e3ec0>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7feada40c180>,
     <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7feada3e37e0>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7feada3e2b60>,
     <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7feada3e2b60>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7feada40c7c0>,
     <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7feada40c7c0>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7feada40cc20>,
     <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7feada40cc20>,
     <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7feada40cf40>,
     <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7feada40cf40>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7feada40db20>,
     <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7feada40d1c0>,
     <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7feada40db20>,
     <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7feada40d1c0>,
     <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7feada40cfe0>,
     <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7feada40d6c0>,
     <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7feada40d580>,
     <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be89b060>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>),
     <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7feada40d620>,
     <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7feada40d620>,
     <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7004a0>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>),
     <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7feada40dbc0>,
     <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7feada40d800>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7feada40e520>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7feada40ef20>,
     <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7feada40f1a0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7feada40eb60>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7feada40f240>,
     <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7feada40f240>,
     <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7feada40f560>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7feada40f880>,
     <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7feada40f880>,
     <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7feada40fba0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7feada40fec0>,
     <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7feada40fec0>,
     <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7feada428720>,
     <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7feada428900>,
     <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7feada428ae0>,
     <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7feada428cc0>,
     <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7feada40d940>,
     <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7feada429120>,
     <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7feada429300>,
     <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7feada429620>,
     <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7feada429800>,
     <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7feada429760>,
     <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7feada428a40>,
     <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7feada4299e0>,
     <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7feada429da0>,
     <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7feada40e840>,
     <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7feada42a160>,
     <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7feada42a160>,
     <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7feada40e840>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7feada42a340>,
     <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7feada42a480>,
     <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7feada42aca0>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7feada42b380>,
     <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7feada42b380>,
     <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7feada42a520>,
     <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7feada429940>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7feada42bd80>,
     <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7feada42bd80>,
     <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7feada4505e0>,
     <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7feada450b80>,
     <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7feada450fe0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4514e0>,
     <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7feada4514e0>,
     <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7feada4516c0>,
     <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7feada40eac0>,
     <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7feada40eac0>,
     <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7feada451a80>,
     <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7feada450540>,
     <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7feada450540>,
     <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7feada4520c0>,
     <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7feada4520c0>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7feada4527a0>,
     <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7feada4527a0>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7feada452de0>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7feada453060>,
     <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7feada4537e0>,
     <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7feada453380>,
     <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7feada453380>,
     <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7feada453f60>,
     <OpOverload(op='aten.take', overload='default')>: <function take at 0x7feada452d40>,
     <OpOverload(op='aten.take', overload='out')>: <function take at 0x7feada452d40>,
     <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453f60>,
     <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7feada4532e0>,
     <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7feada4532e0>,
     <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada452a20>,
     <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada452020>,
     <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada451bc0>,
     <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada450680>,
     <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada453880>,
     <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada453880>,
     <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4539c0>,
     <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ba0>,
     <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453b00>,
     <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453c40>,
     <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada453d80>,
     <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ec0>,
     <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada453ec0>,
     <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478040>,
     <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478040>,
     <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4782c0>,
     <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478400>,
     <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478400>,
     <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478540>,
     <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478540>,
     <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7feada478680>,
     <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7feada478680>,
     <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4787c0>,
     <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478900>,
     <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7feada453e20>,
     <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478a40>,
     <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478b80>,
     <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7feada453920>,
     <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada4519e0>,
     <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7feada378860>,
     <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7feada453100>,
     <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be899a80>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>),
     <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be708e00>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>),
     <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7feada378c20>,
     <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be719120>, kernel=<OpOverload(op='aten.var', overload='default')>),
     <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ef20>, kernel=<OpOverload(op='aten.var', overload='dim')>),
     <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fead83c99e0>, kernel=<OpOverload(op='aten.var', overload='correction_names')>),
     <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7feada379300>,
     <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709440>, kernel=<OpOverload(op='aten.var', overload='names_dim')>),
     <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7feada378e00>,
     <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7feada378cc0>,
     <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7feada3c6480>,
     <OpOverload(op='aten.full', overload='out')>: <function full at 0x7feada3c6160>,
     <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7feada1f11c0>,
     <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7feada40d120>,
     <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7feada40d120>,
     <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7feada340680>,
     <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7feada340680>,
     <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7feada56e3e0>,
     <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7feada56c720>,
     <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7feada56c720>,
     <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7feada56cb80>,
     <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7feada56cb80>,
     <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7feada56c680>,
     <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7feada56c680>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7feada56f600>,
     <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7feada56d760>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7feada56d800>,
     <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7feada56d800>,
     <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7feada56dbc0>,
     <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7feada56e160>,
     <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7feada56e480>,
     <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7feada56e480>,
     <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7feada56e520>,
     <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7feada56e520>,
     <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7feada56e3e0>,
     <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7feada56d300>,
     <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7feada56d300>,
     <OpOverload(op='aten.mish_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7467a0>, kernel=<OpOverload(op='aten.mish_backward', overload='default')>),
     <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7feada56ed40>,
     <OpOverload(op='aten.silu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8982c0>, kernel=<OpOverload(op='aten.silu_backward', overload='default')>),
     <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7feada56ede0>,
     <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7feada56ea20>,
     <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7feada56f1a0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7feada56f6a0>,
     <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7feada56f6a0>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7feada56f740>,
     <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7feada56f740>,
     <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7feada56c9a0>,
     <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7feada56fec0>,
     <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7feada56efc0>,
     <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7feada56ff60>,
     <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7feada56ff60>,
     <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7feada56ea20>,
     <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7feada5b42c0>,
     <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7feada5b47c0>,
     <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7feada5b4720>,
     <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7feada5b4720>,
     <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7feada5b4180>,
     <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7feada5b4180>,
     <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7feada5b4b80>,
     <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7feada5b4b80>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7feada5b4ea0>,
     <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7feada5b4ea0>,
     <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7feada5b5440>,
     <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7feada5b5440>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7feada5b54e0>,
     <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7feada5b54e0>,
     <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7feada5b5b20>,
     <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7feada5b5b20>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7feada5b5bc0>,
     <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7feada5b5bc0>,
     <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7feada5b7ce0>,
     <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7feada5b6160>,
     <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7feada5b6160>,
     <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7feada5b63e0>,
     <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7feada5b63e0>,
     <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7feada5b49a0>,
     <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7feada5b49a0>,
     <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7feada5b67a0>,
     <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7feada5b67a0>,
     <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7feada5b68e0>,
     <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7feada5b68e0>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7feada5b6f20>,
     <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7feada5b6f20>,
     <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7feada5b7240>,
     <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7feada5b7600>,
     <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7feada313f60>,
     <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7feada3244a0>,
     <OpOverload(op='aten.add', overload='out')>: <function add at 0x7feada3249a0>,
     <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7feada3249a0>,
     <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7feada326840>,
     <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7feada324e00>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7feada325260>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7feada325b20>,
     <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7feada325760>,
     <OpOverload(op='aten.div', overload='out')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7feada3263e0>,
     <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7feada326840>,
     <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7feada3277e0>,
     <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7feada3277e0>,
     <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7feada327c40>,
     <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7feada327c40>,
     <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7feada937ba0>,
     <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7feada3400e0>,
     <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7feada3400e0>,
     <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7feada326200>,
     <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7feada326200>,
     <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7feada340ae0>,
     <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7feada340ae0>,
     <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7feada340f40>,
     <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7feada340f40>,
     <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7feada341800>,
     <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7feada341800>,
     <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7feada341c60>,
     <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7feada341c60>,
     <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7feada3420c0>,
     <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7feada3420c0>,
     <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7feada342ac0>,
     <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7feada342ac0>,
     <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7feada3602c0>,
     <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7feada3602c0>,
     <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7feada360720>,
     <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7feada360b80>,
     <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7feada361440>,
     <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7feada361440>,
     <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7feada3618a0>,
     <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7feada3618a0>,
     <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7feada326ca0>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7feada361d00>,
     <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7feada341d00>,
     <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7feada341d00>,
     <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7feada39e5c0>,
     <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7feada39f560>,
     <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7feada37a7a0>,
     <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be745f80>, kernel=<OpOverload(op='aten.cat', overload='names')>),
     <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7feada37ade0>,
     <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7feada37ade0>,
     <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be709c60>, kernel=<OpOverload(op='aten.where', overload='default')>),
     <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7feada363ba0>,
     <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7feada47ba60>,
     <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7feada47bf60>,
     <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7feada2e9260>,
     <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7feada2e9760>,
     <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7feada2e84a0>,
     <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7feada2e89a0>,
     <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7feada2e8d60>,
     <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7feada2e8d60>,
     <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7feada47a700>,
     <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7feada47a700>,
     <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7feada363e20>,
     <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7feada2e9a80>,
     <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7feada2ea480>,
     <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7feada937ba0>,
     <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7feada2ea980>,
     <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7feada2eae80>,
     <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7feada2eb380>,
     <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7feada2ebec0>,
     <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7feada2fa520>,
     <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7feada2fa520>,
     <OpOverload(op='aten.log', overload='out')>: <function log at 0x7feada2faa20>,
     <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7feada2faf20>,
     <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7feada2fb420>,
     <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7feada2fb920>,
     <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7feada3107c0>,
     <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7feada310180>,
     <OpOverload(op='aten.round', overload='out')>: <function round at 0x7feada310b80>,
     <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7feada311080>,
     <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7feada310b80>,
     <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7feada311f80>,
     <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7feada312480>,
     <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7feada312480>,
     <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7feada312980>,
     <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7feada312520>,
     <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7feada313100>,
     <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7feada313a60>,
     <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7feada47a160>,
     <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7feada47a660>,
     <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7feada47ab60>,
     <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7feada47b060>,
     <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7feada47b560>,
     <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be723c40>, kernel=<OpOverload(op='aten.svd', overload='default')>),
     <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7feada360fe0>,
     <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7feada39eac0>,
     <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7feada452e80>,
     <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7feada37a7a0>,
     <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7feada3e3880>,
     <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7feada3c5300>,
     <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7feada3c67a0>,
     <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7feada3c5260>,
     <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7feada3c4ae0>,
     <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7feada3c4d60>,
     <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7feada3c4fe0>,
     <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7feada39f420>,
     <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7feada3c4e00>,
     <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be7032e0>, 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 0x7fe9bf23ab60>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>),
     <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7feada5b56c0>,
     <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7feada478180>,
     <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8cd60>, 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 0x7fea23f8f6a0>, 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 0x7fe9be8e22a0>, 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 0x7fe9be8e09a0>, 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 0x7fea23f8f880>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c9a0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d6c0>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c4a0>, 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 0x7fea23f8e7a0>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8fec0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>),
     <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3880>, kernel=<OpOverload(op='quantized.conv1d_unpack', 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 0x7fe9be8e1440>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3c40>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e2de0>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e2660>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d760>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e02c0>, 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 0x7fe9be8e23e0>, kernel=<OpOverload(op='quantized.conv2d_groups', 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 0x7fea23f8e700>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>),
     <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e00e0>, 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 0x7fe9be8e16c0>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>),
     <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d800>, 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 0x7fea23f8df80>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>),
     <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f920>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f100>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>),
     <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f9c0>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>),
     <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8ff60>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>),
     <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8c720>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e2a0>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8e480>, 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 0x7fea23f8cb80>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>),
     <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8f380>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>),
     <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8d620>, kernel=<OpOverload(op='sparse.qlinear_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 0x7fea23f8e980>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>),
     <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fea23f8db20>, kernel=<OpOverload(op='quantized.conv2d_output_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 0x7fea23f8e840>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>),
     <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7fe9be8e3ba0>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}
experimental_experiment.torch_dynamo.pprint_storage(storage: Any, indent: int = 0, as_list: bool = False) List[str] | str[source]

Pretty print for the storage.

Parameters:
  • storage – any object

  • indent – indentation

  • as_list – return list or string

Returns:

list or string