experimental_experiment.torch_dynamo¶
modules
- experimental_experiment.torch_dynamo._dynamo_exporter
- experimental_experiment.torch_dynamo.backend_helper
- experimental_experiment.torch_dynamo.debug_backend
- experimental_experiment.torch_dynamo.dynger_backend
- experimental_experiment.torch_dynamo.fast_backend
- experimental_experiment.torch_dynamo.partition
onnx_custom_backend¶
- experimental_experiment.torch_dynamo.onnx_custom_backend(graph_module: torch.fx.GraphModule, args: List[torch.Tensor], target_opset: int | None = None, backend: str = 'ort', verbose: int | Tuple[int, int] = 0, dump_prefix: 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.Dispatcherrename_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.0002028359958785586 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.004350235001766123 s [to_onnx] start conversion to onnx (before optimization) [GraphBuilder.optimize] start with 13 nodes [GraphBuilder.optimize] #patterns=41 [GraphBuilderPatternOptimization.optimize] start with 7 nodes, 0 initializers, 41 patterns, priorities=[0, 1] [GraphBuilderPatternOptimization.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization.optimize] increase priority to 1 [GraphBuilderPatternOptimization.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000 [GraphBuilderPatternOptimization.optimize] iteration 2: 5 nodes, priority=1 [GraphBuilderPatternOptimization.optimize] applies 2 matches, 2*TransposeMatMulPattern - time=0.001 | max_time=TransposeMatMulPattern:0.000 [GraphBuilderPatternOptimization.optimize] iteration 3: 3 nodes, priority=1 [GraphBuilderPatternOptimization.optimize] done after 4 iterations with 3 nodes in 0.006 [GraphBuilder.optimize] done with 3 nodes in 0.007 [GraphBuilder-MTC.to_onnx] make_model 0 inits 0 params [GraphBuilder-MTC.time_evaluation_constants_] 0 [GraphBuilder-MTC._build_initializers] start with 0 initializers, large_model=False, external_threshold=1024 [GraphBuilder-MTC._build_initializers] switch low/high order [GraphBuilder-MTC._build_initializers] done in 7.979979272931814e-07s with 0 initializers, 0 large initializers [to_onnx] to_onnx done in 0.0073771480019786395s and 3 nodes, 0 initializers, 5 inputs, 1 outputs [onnx_custom_backend] to_onnx done in 0.012133980999351479 with 3 nodes and 0 local functions. [onnx_custom_backend] starts creating InferenceSession [onnx_custom_backend] InferenceSession done in 0.0017043429979821667 discrepancies: 3.91155481338501e-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.Dispatcherrename_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.00019428999803494662 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.0017815510000218637 s [to_onnx] start conversion to onnx (before optimization) [GraphBuilder.optimize] start with 13 nodes [GraphBuilder.optimize] #patterns=41 [GraphBuilderPatternOptimization.optimize] start with 7 nodes, 0 initializers, 41 patterns, priorities=[0, 1] [GraphBuilderPatternOptimization.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization.optimize] increase priority to 1 [GraphBuilderPatternOptimization.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000 [GraphBuilderPatternOptimization.optimize] iteration 2: 5 nodes, priority=1 [GraphBuilderPatternOptimization.optimize] applies 2 matches, 2*TransposeMatMulPattern - time=0.000 | max_time=TransposeMatMulPattern:0.000 [GraphBuilderPatternOptimization.optimize] iteration 3: 3 nodes, priority=1 [GraphBuilderPatternOptimization.optimize] done after 4 iterations with 3 nodes in 0.004 [GraphBuilder.optimize] done with 3 nodes in 0.005 [GraphBuilder-RQU.to_onnx] make_model 0 inits 0 params [GraphBuilder-RQU.time_evaluation_constants_] 0 [GraphBuilder-RQU._build_initializers] start with 0 initializers, large_model=False, external_threshold=1024 [GraphBuilder-RQU._build_initializers] switch low/high order [GraphBuilder-RQU._build_initializers] done in 8.049973985180259e-07s with 0 initializers, 0 large initializers [to_onnx] to_onnx done in 0.005273333001241554s and 3 nodes, 0 initializers, 5 inputs, 1 outputs +I input0: float32:(32, 10):0.09399528056383133,-0.04702427610754967,0.04685678705573082,0.25868648290634155,-0.02750461921095848... +I input1: float32:(32,):-0.19360753893852234,-0.25477561354637146,-0.20759518444538116,-0.06176048144698143,-0.28520330786705017... +I input2: float32:(3, 10):-0.27054494619369507,0.3067127466201782,-0.46791166067123413,0.4930364787578583,0.0186257753521204... +I input3: float32:(1, 32):-0.06619135290384293,-0.09224040061235428,0.12404367327690125,0.14847543835639954,0.07026579231023788... +I input4: float32:(1,):[-0.13115167617797852] Gemm(input2, input0, input1) -> input_1 + input_1: float32:(3, 32):0.29188060760498047,-0.4027822017669678,0.386757493019104,-0.25353118777275085,-0.6474719047546387... Sigmoid(input_1) -> input_2 + input_2: float32:(3, 32):0.5724564790725708,0.40064406394958496,0.5955018997192383,0.4369545578956604,0.34355947375297546... Gemm(input_2, input3, input4) -> output_0 + output_0: float32:(3, 1):[0.0339796245098114, 0.046193256974220276, -0.08388452231884003] discrepancies: 2.9802322387695312e-08
dynger_backend¶
- experimental_experiment.torch_dynamo.dynger_backend(graph_module: GraphModule, args: List[Tensor | SymInt | SymFloat], dynamic_shapes: Dict[str, Any] | Tuple[Any] | None = None, optimize: bool = True, verbose: int | Tuple[int, int] = 0) Callable[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.9246276617050171,1.5083554983139038,0.9712582230567932,-0.6411529779434204,0.5923361778259277... <built-in method sigmoid of type object at 0x7ff3cfc5e360>((input_1,)) -> input_2 + input_2: torch.float32:torch.Size([3, 32]):0.2840159237384796,0.8188173770904541,0.7253702282905579,0.3449859619140625,0.643900990486145... <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.10947975516319275,-0.003511786460876465,-0.1746644526720047 [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 0x7ff2ef919930>: <function out_dtype_decomp at 0x7ff2ef8180d0>, <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x7ff2ef4800d0>, <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7ff2ef480280>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7ff2ef480280>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7ff2ef480430>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7ff2ef480430>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7ff2ef657640>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7ff2ef480310>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7ff2ef4804c0>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7ff2ef480550>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7ff2ef4805e0>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7ff2ef480c10>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7ff2ef480c10>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7ff2ef480f70>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7ff2ef480f70>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7ff2ef4811b0>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7ff2ef4811b0>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7ff2ef4813f0>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7ff2ef4813f0>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7ff2ef4817e0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7ff2ef4817e0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7ff2ef481b40>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7ff2ef481b40>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7ff2ef481ea0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7ff2ef481ea0>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7ff2ef4820e0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7ff2ef4820e0>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7ff2ef482320>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7ff2ef482320>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7ff2ef4827a0>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7ff2ef4827a0>, <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x7ff2ef4829e0>, <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7ff2ef4829e0>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7ff2ef482c20>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7ff2ef482c20>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7ff2ef4830a0>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7ff2ef4830a0>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7ff2ef74f010>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7ff2ef74eb00>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7ff2ef74e9e0>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7ff2ef74ed40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7ff2ef74ed40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7ff2ef74eef0>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7ff2ef74eef0>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7ff2ef74f130>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7ff2ef74f250>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7ff2ef74f6d0>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7ff2ef74f490>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7ff2ef74f490>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7ff2ef74f640>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7ff2ef574d30>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7ff2ef574670>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7ff2ef574d30>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x7ff2ef5751b0>, <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x7ff2ef5751b0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7ff2ef575ea0>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7ff2ef5760e0>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7ff2ef5760e0>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7ff2ef576830>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7ff2ef576830>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7ff2ef576c20>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7ff2ef576c20>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7ff2ef576dd0>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7ff2ef576dd0>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7ff2ef5770a0>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7ff2ef5770a0>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7ff2ef5772e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7ff2ef5772e0>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7ff2ef5776d0>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7ff2ef5776d0>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7ff2ef577910>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7ff2ef577910>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7ff2ef577b50>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7ff2ef577b50>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7ff2ef577d90>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7ff2ef577d90>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7ff2ef577e20>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7ff2ef5b4430>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7ff2ef5b4430>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7ff2ef5b4790>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7ff2ef5b4790>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7ff2ef5b4c10>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7ff2ef5b4c10>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7ff2ef5b4e50>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7ff2ef5b4e50>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7ff2ef5b5000>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7ff2ef5b4820>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7ff2ef5b4ee0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7ff2ef5b5090>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7ff2ef5b5120>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7ff2ef5b5120>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7ff2ef5b51b0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7ff2ef5b5240>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7ff2ef5b52d0>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7ff2ef5b5360>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7ff2ef5b53f0>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7ff2ef5b5480>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7ff2ef5b5510>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7ff2ef5b55a0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7ff2ef5b55a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7ff2ef5b5630>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7ff2ef5b5630>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7ff2ef5b56c0>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7ff2ef5b5750>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7ff2ef5b5750>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7ff2ef5b57e0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7ff2ef5b57e0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7ff2ef5b5870>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7ff2ef5b5870>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7ff2ef5b5900>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7ff2ef5b5990>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7ff2ef5b5990>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7ff2ef5b5a20>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7ff2ef5b5a20>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7ff2ef5b5ab0>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7ff2ef5b5ab0>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7ff2ef5b5b40>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7ff2ef5b5bd0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7ff2ef5b5bd0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7ff2ef5b5c60>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7ff2ef5b5cf0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7ff2ef5b5d80>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7ff2ef5b5d80>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7ff2ef5b5e10>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7ff2ef5b5e10>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7ff2ef5b5ea0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7ff2ef5b5f30>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7ff2ef5b6050>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7ff2ef5b6050>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7ff2ef5b60e0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7ff2ef5b6170>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7ff2ef5b6200>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7ff2ef5b6290>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7ff2ef5b6320>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7ff2ef5b63b0>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7ff2ef5b6440>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7ff2ef5b6440>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7ff2ef5b64d0>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7ff2ef5b6560>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7ff2ef5b6560>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7ff2ef5b65f0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7ff2ef5b65f0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7ff2ef5b6680>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7ff2ef5b6710>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7ff2ef5b67a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7ff2ef5b67a0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7ff2ef5b6830>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7ff2ef5b6830>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7ff2ef5b68c0>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7ff2ef5b6950>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7ff2ef5b69e0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7ff2ef5b6a70>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7ff2ef5b6b00>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7ff2ef5b6b90>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7ff2ef5b6c20>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7ff2ef5b6c20>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7ff2ef5b6cb0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7ff2ef5b6cb0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7ff2ef5b6d40>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7ff2ef5b6dd0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7ff2ef5b6e60>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7ff2ef5b6ef0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7ff2ef5b6f80>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7ff2ef5b7010>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7ff2ef5b70a0>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7ff2ef5b7130>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7ff2ef5b71c0>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7ff2ef5b7250>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7ff2ef5b7250>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7ff2ef5b72e0>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7ff2ef5b72e0>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7ff2ef5b7370>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7ff2ef5b7400>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7ff2ef5b7490>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7ff2ef5b7490>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7ff2ef5b7520>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7ff2ef5b75b0>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7ff2ef5b7640>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7ff2ef5b7640>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7ff2ef5b76d0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7ff2ef5b7760>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7ff2ef5b77f0>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7ff2ef5b77f0>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7ff2ef5b7880>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7ff2ef5b7910>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7ff2ef5b79a0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7ff2ef5b7a30>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7ff2ef5b7ac0>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7ff2ef5b7b50>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7ff2ef5b7be0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7ff2ef5b7c70>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7ff2ef5b7d00>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7ff2ef5b7d90>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7ff2ef5b7d90>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7ff2ef5b7e20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7ff2ef5b7eb0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7ff2ef5b7f40>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7ff2ef5f4040>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7ff2ef5f40d0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7ff2ef5f40d0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7ff2ef5f4160>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7ff2ef5f41f0>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7ff2ef5f41f0>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7ff2ef5f4280>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7ff2ef5f4310>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7ff2ef5f43a0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7ff2ef5f4430>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7ff2ef5f44c0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function PyCapsule.alias at 0x7ff2ef5f45e0>, <OpOverload(op='aten.alias_copy', overload='out')>: <function PyCapsule.alias at 0x7ff2ef5f45e0>, <OpOverload(op='aten.as_strided_copy', overload='default')>: <function PyCapsule.as_strided at 0x7ff2ef5f4790>, <OpOverload(op='aten.as_strided_copy', overload='out')>: <function PyCapsule.as_strided at 0x7ff2ef5f4790>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function PyCapsule.diagonal at 0x7ff2ef5f4940>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function PyCapsule.diagonal at 0x7ff2ef5f4940>, <OpOverload(op='aten.expand_copy', overload='default')>: <function PyCapsule.expand at 0x7ff2ef5f4af0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function PyCapsule.expand at 0x7ff2ef5f4af0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function PyCapsule.narrow at 0x7ff2ef5f4ca0>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function PyCapsule.narrow at 0x7ff2ef5f4ca0>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.permute_copy', overload='out')>: <function PyCapsule.permute at 0x7ff2ef5f5000>, <OpOverload(op='aten.permute_copy', overload='default')>: <function PyCapsule.permute at 0x7ff2ef5f5000>, <OpOverload(op='aten.t_copy', overload='out')>: <function PyCapsule.t at 0x7ff2ef5f51b0>, <OpOverload(op='aten.t_copy', overload='default')>: <function PyCapsule.t at 0x7ff2ef5f51b0>, <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function PyCapsule.transpose at 0x7ff2ef5f5360>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function PyCapsule.transpose at 0x7ff2ef5f5360>, <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function PyCapsule.unbind at 0x7ff2ef5f5510>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function PyCapsule.unbind at 0x7ff2ef5f5510>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function PyCapsule.unsqueeze at 0x7ff2ef5f56c0>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function PyCapsule.unsqueeze at 0x7ff2ef5f56c0>, <OpOverload(op='aten.view_copy', overload='default')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.view_copy', overload='out')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7ff2ef5f67a0>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7ff2ef5f67a0>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7ff2ef5f6950>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7ff2ef5f6950>, <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x7ff2ef5f7490>, <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7ff2ef5f7490>, <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x7ff2ef5f7640>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7ff2ef5f7640>, <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x7ff2ef5f77f0>, <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7ff2ef5f77f0>, <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x7ff2ef5f79a0>, <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7ff2ef5f79a0>, <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x7ff2ef5f7b50>, <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7ff2ef5f7b50>, <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x7ff2ef5f7d00>, <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7ff2ef5f7d00>, <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x7ff2ef624430>, <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7ff2ef624430>, <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x7ff2ef6245e0>, <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7ff2ef6245e0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x7ff2ef624790>, <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7ff2ef624790>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x7ff2ef624940>, <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7ff2ef624940>, <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7ff2ef624f70>, <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x7ff2ef624f70>, <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x7ff2ef625120>, <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7ff2ef625120>, <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x7ff2ef6252d0>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7ff2ef6252d0>, <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x7ff2ef625480>, <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7ff2ef625480>, <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x7ff2ef625630>, <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7ff2ef625630>, <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x7ff2ef6257e0>, <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7ff2ef6257e0>, <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x7ff2ef625990>, <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7ff2ef625990>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x7ff2ef625b40>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7ff2ef625b40>, <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x7ff2ef625bd0>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x7ff2ef625c60>, <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7ff2ef626290>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7ff2ef626290>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7ff2ef626560>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7ff2ef626560>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7ff2ef627eb0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7ff2ef627eb0>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7ff2ef654430>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7ff2ef654430>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7ff2ef654790>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7ff2ef654790>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7ff2ef654b80>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7ff2ef654b80>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7ff2ef654dc0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7ff2ef654dc0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7ff2ef655090>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7ff2ef655090>, <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x7ff2ef655360>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7ff2ef655bd0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7ff2ef655750>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7ff2ef655750>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7ff2ef655d80>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7ff2ef655bd0>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7ff2ef655d80>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x7ff2ef656290>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x7ff2ef656440>, <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x7ff2ef6567a0>, <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7ff2ef6567a0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7ff2ef6569e0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7ff2ef6569e0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7ff2ef656c20>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7ff2ef656c20>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7ff2ef657010>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7ff2ef657010>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7ff2ef6575b0>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7ff2ef6575b0>, <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x7ff2ef657760>, <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x7ff2ef657880>, <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7ff2ef657c70>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7ff2ef657c70>, <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x7ff2ef657e20>, <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x7ff2ef6825f0>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7ff2ef683b50>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7ff2ef683b50>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7ff2ef6a4dc0>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7ff2ef6a4dc0>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7ff2ef6a5480>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7ff2ef6a5480>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7ff2ef6a5bd0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7ff2ef6a5bd0>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7ff2ef6a5f30>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7ff2ef6a5f30>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7ff2ef6a6290>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7ff2ef6a6290>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7ff2ef6a65f0>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7ff2ef6a65f0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7ff2ef6a6d40>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7ff2ef6a6d40>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7ff2ef6c0430>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7ff2ef6c0430>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7ff2ef6c15a0>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7ff2ef6c15a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7ff2ef6c1900>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7ff2ef6c1900>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7ff2ef6c2680>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7ff2ef6c2680>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7ff2ef6d7910>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7ff2ef6d7910>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7ff2ef6f8700>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7ff2ef6f8700>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7ff2ef6f8ca0>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7ff2ef6f8ca0>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7ff2ef6f8f70>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7ff2ef6f8f70>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7ff2ef6f9240>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7ff2ef6f9240>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7ff2ef6f95a0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7ff2ef6f95a0>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7ff2ef6f9870>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7ff2ef6f9870>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7ff2ef6f9b40>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7ff2ef6f9b40>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7ff2ef6fbd00>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7ff2ef6fbd00>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7ff2ef6fbf40>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7ff2ef6fbf40>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7ff2ef7201f0>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7ff2ef7201f0>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7ff2ef7201f0>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7ff2ef7201f0>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7ff2ef7203a0>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7ff2ef7203a0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7ff2ef7203a0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7ff2ef7203a0>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7ff2ef720550>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7ff2ef720550>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7ff2ef720550>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7ff2ef720550>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.std', overload='default')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='dim')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7ff2ef722cb0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7ff2ef722cb0>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7ff2ef723910>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7ff2ef723910>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7ff2ef723b50>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7ff2ef74c280>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7ff2ef723f40>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7ff2ef723f40>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7ff2ef74c940>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7ff2ef74c3a0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7ff2ef74c3a0>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7ff2ef74cb80>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7ff2ef74cb80>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7ff2ef74cee0>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7ff2ef74cee0>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7ff2ef74d240>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7ff2ef74d240>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7ff2ef74d3f0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7ff2ef74d3f0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7ff2ef74d630>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7ff2ef74d630>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7ff2ef74dbd0>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7ff2ef74dbd0>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7ff2ef74e950>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7ff2ef74e560>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7ff2ef771ea0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7ff2ef771ea0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7ff2ef771f30>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7ff2ef771f30>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7ff2ef7723b0>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7ff2ef7723b0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7ff2ef772440>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7ff2ef772440>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7ff2ef772830>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7ff2ef772830>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7ff2ef7729e0>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7ff2ef7729e0>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7ff2ef772b90>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7ff2ef773010>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7ff2ef7731c0>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7ff2ef7731c0>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7ff2ef7732e0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7ff2ef7732e0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7ff2ef773760>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7ff2ef773760>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7ff2ef7739a0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7ff2ef7739a0>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7ff2ef798040>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7ff2ef773be0>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7ff2ef773be0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7ff2ef773d90>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7ff2ef773d90>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7ff2ef773f40>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7ff2ef773f40>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7ff2ef798940>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7ff2ef798160>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7ff2ef798160>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7ff2ef798940>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7ff2ef798af0>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7ff2ef798af0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7ff2ef799120>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7ff2ef798ca0>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7ff2ef798ca0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7ff2ef799120>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7ff2ef798e50>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7ff2ef798e50>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7ff2ef7991b0>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7ff2ef7991b0>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7ff2ef799240>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7ff2ef7992d0>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7ff2ef799360>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7ff2ef7996c0>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7ff2ef7996c0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7ff2ef799bd0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7ff2ef799900>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7ff2ef799900>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7ff2ef799c60>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7ff2ef799b40>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7ff2ef799b40>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7ff2ef799d80>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7ff2ef799e10>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7ff2ef79a710>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7ff2ef79aa70>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7ff2ef79a9e0>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7ff2ef79a8c0>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7ff2ef79a830>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7ff2ef79a7a0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7ff2ef799fc0>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7ff2ef79ab00>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7ff2ef79b400>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7ff2ef79bc70>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7ff2ef79ac20>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7ff2ef79ac20>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7ff2ef79b400>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7ff2ef79bd90>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7ff2ef79b6d0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7ff2ef79bd00>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7ff2ef79beb0>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7ff2ef79beb0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7ff2ef7c05e0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7ff2ef7c05e0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7ff2ef7c0dc0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7ff2ef7c0dc0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7ff2ef7c1000>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7ff2ef7c1000>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7ff2ef7c11b0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7ff2ef7c1510>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7ff2ef7c11b0>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7ff2ef7c1090>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7ff2ef7c1480>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7ff2ef7c1480>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7ff2ef7c2680>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7ff2ef7c1870>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7ff2ef7c1870>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7ff2ef7c1f30>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7ff2ef7c2680>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7ff2ef7c2560>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7ff2ef7c2680>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7ff2ef7c2560>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7ff2ef7c2830>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7ff2ef7c2830>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7ff2ef7c29e0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7ff2ef7c2b90>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7ff2ef7c29e0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7ff2ef7c2b90>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7ff2ef7c2dd0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7ff2ef7c2dd0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7ff2ef7c2f80>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7ff2ef7c2f80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7ff2ef7c3640>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7ff2ef7c36d0>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7ff2ef7c3760>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7ff2ef7c37f0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7ff2ef7c3ac0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7ff2ef7c3b50>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7ff2ef7c3d00>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7ff2ef7c3d90>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7ff2ef7c3e20>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7ff2ef7c3eb0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7ff2ef7e8280>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7ff2ef7e8040>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7ff2ef7e8280>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7ff2ef7e8040>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7ff2ef7e8310>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7ff2ef7e8310>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7ff2ef7e8430>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7ff2ef7e8430>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7ff2ef7e8790>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7ff2ef7e89d0>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7ff2ef7e89d0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7ff2ef7e89d0>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7ff2ef7e88b0>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7ff2ef7e8af0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7ff2ef7e8b80>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7ff2ef7e8d30>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7ff2ef7e8d30>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7ff2ef7ea170>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7ff2ef7e93f0>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7ff2ef7e93f0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7ff2ef7ea170>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7ff2ef7ea440>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7ff2ef7ea440>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7ff2ef7ea680>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7ff2ef7ea680>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7ff2ef7ea830>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7ff2ef7ea830>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7ff2ef7ead40>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7ff2ef7ead40>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7ff2ef7eb010>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb5b0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb5b0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7ff2ef7ebe20>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7ff2ef818040>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7ff2ef818040>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7ff2ef7eb490>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7ff2ef7eb490>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7ff2ef7ebe20>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7ff2ef7ebeb0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7ff2ef8183a0>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7ff2ef8183a0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7ff2ef818160>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7ff2ef818430>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7ff2ef818160>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7ff2ef818d30>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7ff2ef818d30>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7ff2ef819090>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7ff2ef819090>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7ff2ef8195a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7ff2ef8195a0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7ff2ef819480>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7ff2ef819480>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819630>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef8196c0>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819750>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef8197e0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819870>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819870>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819900>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819990>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819a20>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ab0>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819b40>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819b40>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819bd0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819bd0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819cf0>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819d80>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819d80>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819e10>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819e10>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ea0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ea0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819f30>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819fc0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a050>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a0e0>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a170>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a170>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a320>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a3b0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a440>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.where', overload='default')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7ff2ef818f70>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7ff2ef7215a0>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7ff2ef7215a0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7ff2ef818f70>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.var', overload='default')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='dim')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7ff2ef721900>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7ff2ef721900>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7ff2ef721870>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7ff2ef721870>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7ff2ef5755a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7ff2ef575900>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7ff2ef575900>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7ff2ef7c2050>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7ff2ef7c2050>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7ff2ef6d6830>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7ff2ef6d6830>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7ff2ef935750>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7ff2ef935750>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7ff2ef935a20>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7ff2ef935a20>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7ff2ef935cf0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7ff2ef935cf0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7ff2ef935fc0>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7ff2ef935fc0>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7ff2ef936560>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7ff2ef936560>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7ff2ef9365f0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7ff2ef9365f0>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7ff2ef9368c0>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7ff2ef9368c0>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7ff2ef936cb0>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7ff2ef936cb0>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7ff2ef936ef0>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7ff2ef936ef0>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7ff2ef936f80>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7ff2ef936f80>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7ff2ef9371c0>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7ff2ef9371c0>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7ff2ef937490>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7ff2ef937490>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7ff2ef937760>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7ff2ef937a30>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7ff2ef937a30>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7ff2ef937ac0>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7ff2ef937ac0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7ff2ef937d00>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7ff2ef937d90>, <OpOverload(op='aten.rrelu_with_noise', overload='default')>: <function rrelu_with_noise at 0x7ff2ef770160>, <OpOverload(op='aten.rrelu_with_noise', overload='out')>: <function rrelu_with_noise at 0x7ff2ef770160>, <OpOverload(op='aten.rrelu_with_noise_', overload='default')>: <function rrelu_with_noise_ at 0x7ff2ef7701f0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7ff2ef770430>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7ff2ef770430>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7ff2ef7704c0>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7ff2ef7704c0>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7ff2ef770a60>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7ff2ef770a60>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7ff2ef770af0>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7ff2ef770af0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7ff2ef770d30>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7ff2ef771090>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7ff2ef771000>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7ff2ef7711b0>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7ff2ef771000>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7ff2ef7712d0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7ff2ef7713f0>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7ff2ef7715a0>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7ff2ef7715a0>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7ff2ef771870>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7ff2ef771870>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7ff2ef771ab0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7ff2ef771ab0>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7ff2ef6c2320>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7ff2ef6c29e0>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7ff2ef6c29e0>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7ff2ef6c2d40>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7ff2ef6c2d40>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7ff2ef6c3400>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7ff2ef6c3400>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7ff2ef6c3760>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7ff2ef6c3760>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7ff2ef6c3ac0>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7ff2ef6c3e20>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7ff2ef6c3ac0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7ff2ef6c3e20>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7ff2ef6d4160>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7ff2ef6c3e20>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7ff2ef6d4160>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7ff2ef6c3e20>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7ff2ef6d56c0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7ff2ef6d56c0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7ff2ef6d56c0>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7ff2ef6d6200>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7ff2ef6d56c0>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7ff2ef6d6200>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7ff2ef6d64d0>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7ff2ef6d67a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7ff2ef6d64d0>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7ff2ef6d67a0>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7ff2ef6d67a0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7ff2ef6d70a0>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7ff2ef6d67a0>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7ff2ef6d70a0>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7ff2ef6d7370>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7ff2ef6d7370>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7ff2ef6d7370>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7ff2ef6d7370>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7ff2ef6d7640>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7ff2ef6d7640>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7ff2ef6d7640>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7ff2ef6d7640>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7ff2ef6d7be0>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7ff2ef6d7eb0>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7ff2ef6d7eb0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7ff2ef6f81f0>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7ff2ef6d7be0>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7ff2ef6f89d0>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7ff2ef6f89d0>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7ff2ef6f81f0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7ff2ef6f89d0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7ff2ef6f89d0>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7ff2ef6f9e10>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7ff2ef6f9e10>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7ff2ef6f9e10>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7ff2ef6f9e10>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7ff2ef6fa0e0>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7ff2ef6fa0e0>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7ff2ef6fa3b0>, <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7ff2ef6fa3b0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7ff2ef6fa950>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7ff2ef6fa950>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7ff2ef6fa950>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7ff2ef6fac20>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7ff2ef6fa950>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7ff2ef6fac20>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7ff2ef6fb2e0>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7ff2ef6fb2e0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7ff2ef6fb2e0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7ff2ef6fb2e0>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7ff2ef819000>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7ff2ef819000>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7ff2ef74f1c0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7ff2ef74f1c0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7ff2ef74e3b0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7ff2ef723130>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7ff2ef6813f0>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7ff2ef681750>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7ff2ef681750>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7ff2ef681ab0>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7ff2ef681ab0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7ff2ef681e10>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7ff2ef681e10>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7ff2ef682170>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7ff2ef682170>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7ff2ef682dd0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7ff2ef682dd0>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7ff2ef683130>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7ff2ef683130>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7ff2ef6824d0>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7ff2ef6824d0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7ff2ef682830>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7ff2ef682830>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7ff2ef682a70>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7ff2ef720940>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7ff2ef683490>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7ff2ef683490>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7ff2ef6837f0>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7ff2ef720940>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7ff2ef6837f0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7ff2ef683eb0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7ff2ef683eb0>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7ff2ef6a4280>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7ff2ef6a4280>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7ff2ef6a45e0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7ff2ef6a45e0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7ff2ef6a4940>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7ff2ef936290>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7ff2ef6a4940>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7ff2ef936200>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7ff2ef6a5120>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7ff2ef6a5120>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7ff2ef6a70a0>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7ff2ef6a7400>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7ff2ef6a70a0>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7ff2ef6a7400>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7ff2ef6a7760>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7ff2ef6a7760>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7ff2ef6a7ac0>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7ff2ef6c0ca0>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7ff2ef6a7ac0>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7ff2ef6a7e20>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7ff2ef6a7e20>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7ff2ef6c0820>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7ff2ef6c0ca0>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7ff2ef6c0ee0>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7ff2ef6c0ee0>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7ff2ef6c0820>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7ff2ef6c1240>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7ff2ef6c0ee0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7ff2ef6c1c60>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7ff2ef6c0ee0>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7ff2ef6c1240>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7ff2ef6c1fc0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7ff2ef6c1c60>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7ff2ef6c1fc0>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7ff2ef6c2320>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7ff2ef680d30>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7ff2ef680d30>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7ff2ef681090>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7ff2ef681090>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7ff2ef6813f0>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7ff2ef818dc0>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7ff2ef7981f0>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7ff2ef799fc0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7ff2ef79b6d0>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7ff2ef682a70>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7ff2ef627d00>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7ff2ef74eb00>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7ff2ef5b5000>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7ff2ef74e9e0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7ff2ef574b80>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7ff2ef574b80>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7ff2ef575cf0>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7ff2ef575cf0>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7ff2ef575b40>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7ff2ef575b40>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7ff2ef74ff40>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7ff2ef74ff40>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7ff2ef574160>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7ff2ef574160>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7ff2ef5749d0>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7ff2ef5749d0>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7ff2ef5744c0>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7ff2ef5744c0>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7ff2ef574820>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7ff2ef574820>, <OpOverload(op='aten.item', overload='default')>: <function item at 0x7ff2ef7209d0>, <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7ff2ef7e8a60>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7ff2ef772e60>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7ff2ef772e60>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819c60>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7ff2ef723130>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7ff2ef5755a0>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7ff2ef574310>}
- 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 0x7ff2ef798e50>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7ff2ef799d80>, <OpOverload(op='aten.rrelu_with_noise', overload='default')>: <function rrelu_with_noise at 0x7ff2ef770160>}
- 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
‘none’: do not decompose
‘default’:
get_decomposition_table()‘onnxscript’:
get_decomposition_table_onnxscript()‘dynamo’:
get_decomposition_table_dynamo()
- experimental_experiment.torch_dynamo.get_decomposition_table_dynamo(onnx_registry=None)[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/vv/this/lib/python3.10/site-packages/torch/onnx/_internal/_exporter_legacy.py:108: 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( /home/xadupre/github/onnxscript/onnxscript/converter.py:820: FutureWarning: 'onnxscript.values.Op.param_schemas' is deprecated in version 0.1 and will be removed in the future. Please use '.op_signature' instead. param_schemas = callee.param_schemas() /home/xadupre/github/onnxscript/onnxscript/converter.py:820: FutureWarning: 'onnxscript.values.OnnxFunction.param_schemas' is deprecated in version 0.1 and will be removed in the future. Please use '.op_signature' instead. param_schemas = callee.param_schemas() {<torch._higher_order_ops.out_dtype.OutDtypeOperator object at 0x7ff2ef919930>: <function out_dtype_decomp at 0x7ff2ef8180d0>, <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59b40>, 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 0x7ff279a588b0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5ab00>, 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 0x7ff279a5b5b0>, 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 0x7ff279a5b370>, 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 0x7ff279a59990>, 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 0x7ff279a5b010>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59870>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5a9e0>, kernel=<OpOverload(op='aten.pad', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989ce50>, 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 0x7ff27989dea0>, 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 0x7ff27989d870>, 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 0x7ff27989da20>, 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 0x7ff27989feb0>, 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 0x7ff27989e5f0>, 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 0x7ff27989eb90>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a591b0>, 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 0x7ff2797d79a0>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d6f80>, 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 0x7ff2797d70a0>, 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 0x7ff2797d76d0>, 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 0x7ff27978eb90>, 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 0x7ff27978d6c0>, 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 0x7ff27978fd90>, 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 0x7ff27978ea70>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989dfc0>, 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 0x7ff27989d5a0>, 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 0x7ff27989edd0>, 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 0x7ff27978c9d0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f250>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f520>, 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 0x7ff279658310>, 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 0x7ff279634b80>, 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 0x7ff279637eb0>, 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 0x7ff279637e20>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f2e0>, 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 0x7ff27978fac0>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f400>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978c160>, 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 0x7ff27978c550>, 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 0x7ff27978f5b0>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d240>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d480>, 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 0x7ff27912d2d0>, 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 0x7ff27912cca0>, kernel=<OpOverload(op='aten.column_stack', 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 0x7ff27912cb80>, 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 0x7ff27912cee0>, 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 0x7ff27912c4c0>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d7c70>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978d000>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978e050>, 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 0x7ff27978cdc0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989e050>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989e560>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989d360>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>), <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978e7a0>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978c310>, kernel=<OpOverload(op='aten._remove_batch_dim', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='other_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978e830>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279658ca0>, 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 0x7ff27965beb0>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279637d00>, kernel=<OpOverload(op='aten.nll_loss2d', 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 0x7ff27978f0a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f490>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978cf70>, 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 0x7ff27978cca0>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978c790>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d510>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d750>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>), <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d1b0>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d360>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>), <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912ce50>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912c5e0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e680>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912eb00>, kernel=<OpOverload(op='aten.matrix_exp_backward', 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 0x7ff27912e8c0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e560>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912de10>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e0e0>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912da20>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912dab0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff010>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff130>, 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 0x7ff27a1fecb0>, 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 0x7ff27a1feef0>, 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 0x7ff27a1feb00>, 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 0x7ff27a1fe560>, 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 0x7ff27a1fe710>, 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 0x7ff27912c0d0>, 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 0x7ff27912c310>, 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 0x7ff27a1ff910>, 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 0x7ff27a1ffbe0>, 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 0x7ff27a1ff640>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff520>, 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 0x7ff27a1fcf70>, 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 0x7ff27a1fd1b0>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>), <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fcc10>, 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 0x7ff27a1fce50>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc670>, kernel=<OpOverload(op='aten.conv_tbc_backward', 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 0x7ff27a1fc9d0>, 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 0x7ff27a1fc4c0>, 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 0x7ff27a1fe0e0>, 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 0x7ff27a1fe200>, 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 0x7ff27a1fdc60>, 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 0x7ff27a1fdf30>, 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 0x7ff27a1fd870>, 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 0x7ff27a1fd7e0>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a58ca0>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>), <OpOverload(op='aten._assert_tensor_metadata', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a584c0>, kernel=<OpOverload(op='aten._assert_tensor_metadata', overload='default')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5a320>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5a200>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59000>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59900>, 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 0x7ff27a1fc1f0>, 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 0x7ff2ed450040>, 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 0x7ff2794d7760>, 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 0x7ff2ed4063b0>, 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 0x7ff2ed4516c0>, 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 0x7ff2ed451b40>, 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 0x7ff2ed451480>, 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 0x7ff279a5a4d0>, 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 0x7ff279a5b2e0>, 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 0x7ff279a5be20>, 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 0x7ff279a5bd00>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59fc0>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a58c10>, 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 0x7ff279a5a3b0>, 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 0x7ff279a5ba30>, 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 0x7ff279a597e0>, 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 0x7ff279a5b9a0>, 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 0x7ff279a5a710>, 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 0x7ff2797d6320>, 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 0x7ff27989e3b0>, 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 0x7ff27989f1c0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989ca60>, 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 0x7ff27989ecb0>, 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 0x7ff27989f010>, 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 0x7ff2797d69e0>, 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 0x7ff279a5acb0>, 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 0x7ff2797d71c0>, 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 0x7ff2797d6290>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d7130>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d72e0>, 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 0x7ff2797d7e20>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978d630>, 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 0x7ff27978eef0>, 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 0x7ff27978f910>, 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 0x7ff27978ee60>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978dfc0>, 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 0x7ff27989dbd0>, kernel=<OpOverload(op='aten.cumprod_backward', 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 0x7ff27989d6c0>, 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 0x7ff27989e7a0>, 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 0x7ff27989cdc0>, 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 0x7ff27989fc70>, 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 0x7ff27989e9e0>, 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 0x7ff27978e680>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978c5e0>, 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 0x7ff27965bb50>, 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 0x7ff27965bd00>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279637d90>, 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 0x7ff2dbd1eb00>, 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 0x7ff2794ef010>, 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 0x7ff2dbebc160>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f7f0>, 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 0x7ff27978ec20>, 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 0x7ff27978f760>, 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 0x7ff27978c1f0>, 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 0x7ff27978d360>, 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 0x7ff27978f010>, 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 0x7ff27978c670>, 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 0x7ff27978c3a0>, 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 0x7ff279a5b760>, 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 0x7ff279a59e10>, 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 0x7ff279a5b130>, 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 0x7ff279a59d80>, 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 0x7ff279a59ab0>, 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 0x7ff279a5beb0>, 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 0x7ff279a5a8c0>, 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 0x7ff279a593f0>, 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 0x7ff279a5ad40>, 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 0x7ff279a5b520>, 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 0x7ff27989e4d0>, 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 0x7ff2797d7370>, 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 0x7ff27989eef0>, 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 0x7ff27989e950>, 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 0x7ff27989d900>, 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 0x7ff27989caf0>, 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 0x7ff27989e830>, 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 0x7ff279a59750>, 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 0x7ff279a595a0>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d7880>, 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 0x7ff2797d7760>, 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 0x7ff2797d7be0>, 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 0x7ff2797d7490>, 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 0x7ff27978e8c0>, 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 0x7ff27978e560>, 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 0x7ff27978df30>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', 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 0x7ff27989cca0>, 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 0x7ff27989eb00>, 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 0x7ff27989d2d0>, 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 0x7ff27989e710>, 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 0x7ff27989e290>, 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 0x7ff27978e710>, 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 0x7ff27978c040>, 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 0x7ff27978d510>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978c820>, 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 0x7ff279659fc0>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2794eeb90>, 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 0x7ff27978fa30>, 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 0x7ff27978e200>, 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 0x7ff27978f9a0>, 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 0x7ff27978feb0>, 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 0x7ff27978c0d0>, 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 0x7ff27978d3f0>, 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 0x7ff27978c700>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912c790>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5bc70>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e4d0>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451bd0>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279b11510>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd750>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe050>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd510>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451cf0>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451a20>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5b880>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff400>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912dbd0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912ca60>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc160>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fda20>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc790>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912cd30>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd990>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc040>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fedd0>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912ea70>, kernel=<OpOverload(op='aten.rrelu', overload='default')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff370>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5a830>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fdfc0>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed407f40>, kernel=<OpOverload(op='aten.fix', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe5f0>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd480>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e5f0>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd900>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ffe20>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451f30>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279212dd0>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc5e0>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912c1f0>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d5a0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd5a0>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ffa30>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912dc60>, kernel=<OpOverload(op='aten.absolute', overload='default')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912c040>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912c8b0>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed4515a0>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd2d0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5add0>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912cc10>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978dcf0>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989f2e0>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59c60>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989cd30>, kernel=<OpOverload(op='aten.inverse', 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 0x7ff279a59360>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989ec20>, kernel=<OpOverload(op='aten._gather_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 0x7ff279a5b250>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59f30>, kernel=<OpOverload(op='aten.trace_backward', 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 0x7ff27989d000>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f640>, kernel=<OpOverload(op='aten.embedding_sparse_backward', 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 0x7ff2dbd1e7a0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989e8c0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2796589d0>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7ff27989d510>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989e680>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978fc70>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f1c0>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5af80>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989ee60>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d7d00>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e290>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd000>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279212b00>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe4d0>, 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 0x7ff27a1fdea0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd630>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5b640>, kernel=<OpOverload(op='aten.smm', 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 0x7ff27912dcf0>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912cdc0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59090>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5bf40>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d7640>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989db40>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978fb50>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978eb00>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978fd00>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978e170>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f370>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978ce50>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978e320>, 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 0x7ff27965bc70>, 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 0x7ff279a59a20>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc3a0>, 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 0x7ff27a1ff490>, 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 0x7ff27a1fe320>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451900>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5a050>, kernel=<OpOverload(op='aten.orgqr', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978ff40>, kernel=<OpOverload(op='aten.sum_to_size', 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 0x7ff27989d630>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5b400>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_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 0x7ff279a5a560>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989e0e0>, kernel=<OpOverload(op='aten.to_mkldnn_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 0x7ff279a5b6d0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5ab90>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978ef80>, kernel=<OpOverload(op='aten.quantile', 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 0x7ff279a59ea0>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', 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 0x7ff27978c4c0>, 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 0x7ff279a5aef0>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978c940>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d630>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912cf70>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912caf0>, 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 0x7ff27912c820>, 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 0x7ff27912c550>, 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 0x7ff27912e7a0>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989e170>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912c940>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.nonzero_numpy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912dfc0>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e050>, 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 0x7ff27912e170>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fcd30>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe7a0>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fef80>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc310>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe8c0>, 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 0x7ff27a1fe950>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fed40>, 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 0x7ff27a1ff5b0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', 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 0x7ff27978c430>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e9e0>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451750>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978f6d0>, 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 0x7ff27912d090>, 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 0x7ff27912df30>, 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 0x7ff27912d870>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe680>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ffeb0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978d870>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed4513f0>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten.pdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5ae60>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7ff2ef657640>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7ff2ef480310>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7ff2ef4804c0>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7ff2ef4805e0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7ff2ef4811b0>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7ff2ef4811b0>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7ff2ef4820e0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7ff2ef4820e0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7ff2ef4827a0>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7ff2ef4827a0>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe440>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7ff2ef74ed40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7ff2ef74ed40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7ff2ef74eef0>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7ff2ef74eef0>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7ff2ef74f490>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7ff2ef74f490>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2797d6440>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff0a0>, 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 0x7ff27a1ff250>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7ff2ef5760e0>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7ff2ef5760e0>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fd6c0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59480>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7ff2ef576830>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7ff2ef576830>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7ff2ef5b4e50>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7ff2ef5b4e50>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7ff2ef5b51b0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7ff2ef5b5240>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7ff2ef5b5ea0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7ff2ef5b6680>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7ff2ef5b68c0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7ff2ef5b6cb0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7ff2ef5b6cb0>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7ff2ef5b7370>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7ff2ef5b7400>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7ff2ef5b76d0>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7ff2ef5b7910>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7ff2ef5b7b50>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7ff2ef5b7f40>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7ff2ef5f4040>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7ff2ef5f41f0>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7ff2ef5f41f0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7ff2ef5f44c0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function PyCapsule.alias at 0x7ff2ef5f45e0>, <OpOverload(op='aten.alias_copy', overload='out')>: <function PyCapsule.alias at 0x7ff2ef5f45e0>, <OpOverload(op='aten.as_strided_copy', overload='default')>: <function PyCapsule.as_strided at 0x7ff2ef5f4790>, <OpOverload(op='aten.as_strided_copy', overload='out')>: <function PyCapsule.as_strided at 0x7ff2ef5f4790>, <OpOverload(op='aten.expand_copy', overload='default')>: <function PyCapsule.expand at 0x7ff2ef5f4af0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function PyCapsule.expand at 0x7ff2ef5f4af0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function PyCapsule.narrow at 0x7ff2ef5f4ca0>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function PyCapsule.narrow at 0x7ff2ef5f4ca0>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.permute_copy', overload='out')>: <function PyCapsule.permute at 0x7ff2ef5f5000>, <OpOverload(op='aten.permute_copy', overload='default')>: <function PyCapsule.permute at 0x7ff2ef5f5000>, <OpOverload(op='aten.t_copy', overload='out')>: <function PyCapsule.t at 0x7ff2ef5f51b0>, <OpOverload(op='aten.t_copy', overload='default')>: <function PyCapsule.t at 0x7ff2ef5f51b0>, <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function PyCapsule.transpose at 0x7ff2ef5f5360>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function PyCapsule.transpose at 0x7ff2ef5f5360>, <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function PyCapsule.unbind at 0x7ff2ef5f5510>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function PyCapsule.unbind at 0x7ff2ef5f5510>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function PyCapsule.unsqueeze at 0x7ff2ef5f56c0>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function PyCapsule.unsqueeze at 0x7ff2ef5f56c0>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe9e0>, 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 0x7ff27912c160>, 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 0x7ff27a1ff9a0>, 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 0x7ff27a1fd090>, 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 0x7ff27a1fc940>, 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 0x7ff27a1fe290>, 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 0x7ff27a1fdbd0>, 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 0x7ff279a5b910>, 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 0x7ff279212a70>, 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 0x7ff2ed451990>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d3f0>, 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 0x7ff27912e830>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912dea0>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff1c0>, 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 0x7ff27a1fea70>, 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 0x7ff27912c280>, 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 0x7ff27a1ffac0>, 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 0x7ff27a1fd240>, 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 0x7ff27a1fcee0>, 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 0x7ff27a1fc700>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7ff2ef654b80>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7ff2ef654b80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7ff2ef655bd0>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7ff2ef655d80>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7ff2ef655bd0>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7ff2ef655d80>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912dd80>, 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 0x7ff27912d990>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>), <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7ff2ef6569e0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7ff2ef6569e0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7ff2ef656c20>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7ff2ef656c20>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed4517e0>, kernel=<OpOverload(op='aten.pairwise_distance', overload='default')>), <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fee60>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7ff2ef6a4dc0>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7ff2ef6a4dc0>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7ff2ef6c0430>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7ff2ef6c0430>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7ff2ef6c1900>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7ff2ef6c1900>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912e200>, 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 0x7ff27912e440>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912db40>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27912d900>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ffd90>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ffc70>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ff7f0>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1ffb50>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fcb80>, kernel=<OpOverload(op='aten.var_mean', overload='correction_names')>), <OpOverload(op='aten.var_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fcdc0>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc550>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fc820>, kernel=<OpOverload(op='aten.var_mean', overload='default')>), <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7ff2ef723250>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7ff2ef74ca60>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7ff2ef74cb80>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7ff2ef74cb80>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7ff2ef74cc10>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7ff2ef74d3f0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7ff2ef74d3f0>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7ff2ef74dbd0>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451ab0>, 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 0x7ff2ed4520e0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7ff2ef771ea0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7ff2ef771ea0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7ff2ef771f30>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7ff2ef771f30>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7ff2ef7723b0>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7ff2ef7723b0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7ff2ef772440>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7ff2ef772440>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7ff2ef772830>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7ff2ef772830>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7ff2ef7729e0>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7ff2ef7729e0>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7ff2ef772b90>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7ff2ef772b90>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7ff2ef773010>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7ff2ef773010>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7ff2ef7731c0>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7ff2ef7731c0>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7ff2ef7732e0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7ff2ef7732e0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7ff2ef773760>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7ff2ef773760>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7ff2ef798040>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7ff2ef773d90>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7ff2ef773d90>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7ff2ef773f40>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7ff2ef773f40>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7ff2ef799120>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7ff2ef799120>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7ff2ef798e50>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7ff2ef798e50>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7ff2ef7991b0>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7ff2ef7991b0>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7ff2ef7992d0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7ff2ef799bd0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7ff2ef799900>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7ff2ef799900>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7ff2ef799c60>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7ff2ef799d80>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7ff2ef799e10>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7ff2ef79aa70>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7ff2ef79a9e0>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7ff2ef79ab00>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7ff2ef79bc70>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7ff2ef79ac20>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7ff2ef79ac20>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7ff2ef79bd90>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7ff2ef79b6d0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7ff2ef79bd00>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7ff2ef79beb0>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7ff2ef79beb0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7ff2ef7c05e0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7ff2ef7c05e0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7ff2ef7c0dc0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7ff2ef7c0dc0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7ff2ef7c1000>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7ff2ef7c1000>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7ff2ef7c11b0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7ff2ef7c1510>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7ff2ef7c11b0>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7ff2ef7c1090>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451510>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7ff2ef7c1480>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7ff2ef7c1480>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59cf0>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7ff2ef7c1870>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7ff2ef7c1870>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7ff2ef7c1f30>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7ff2ef7c2830>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7ff2ef7c2830>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7ff2ef7c2b90>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7ff2ef7c2b90>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7ff2ef7c2f80>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7ff2ef7c2f80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7ff2ef7c3640>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7ff2ef7c36d0>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7ff2ef7c3760>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7ff2ef7c37f0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7ff2ef7c3ac0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7ff2ef7c3b50>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7ff2ef7c3d00>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7ff2ef7c3d90>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7ff2ef7c3e20>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7ff2ef7c3eb0>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7ff2ef7e89d0>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7ff2ef7e8af0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7ff2ef7e8b80>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7ff2ef7ea170>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7ff2ef7e93f0>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7ff2ef7e93f0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7ff2ef7ea170>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7ff2ef7ea830>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7ff2ef7ea830>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb5b0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb5b0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7ff2ef7ebe20>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7ff2ef7eb490>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7ff2ef7eb490>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7ff2ef7ebe20>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7ff2ef8183a0>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7ff2ef8183a0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7ff2ef818160>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7ff2ef818160>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7ff2ef819090>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7ff2ef819090>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7ff2ef8195a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7ff2ef8195a0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7ff2ef819480>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7ff2ef819480>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819630>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef8196c0>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819750>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef8197e0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819870>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819870>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819900>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819990>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819a20>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ab0>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819b40>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819b40>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819bd0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819bd0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819cf0>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819d80>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819d80>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819e10>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819e10>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ea0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ea0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819f30>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819fc0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a050>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a0e0>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a170>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a170>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a320>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a3b0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a440>, <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a59120>, 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 0x7ff2797d7400>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff279a5bd90>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27978d480>, 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 0x7ff27989ea70>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27989d990>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7ff2ef7c2050>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7ff2ef7c2050>, <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fdb40>, kernel=<OpOverload(op='aten.linear', overload='default')>), <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7ff2ef935750>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7ff2ef935750>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7ff2ef935a20>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7ff2ef935a20>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7ff2ef935cf0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7ff2ef935cf0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7ff2ef935fc0>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7ff2ef935fc0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7ff2ef9365f0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7ff2ef9365f0>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7ff2ef936ef0>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7ff2ef936ef0>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7ff2ef936f80>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7ff2ef936f80>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7ff2ef9371c0>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7ff2ef9371c0>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7ff2ef937490>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7ff2ef937490>, <OpOverload(op='aten.mish_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2ed451630>, 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 0x7ff27912c670>, kernel=<OpOverload(op='aten.silu_backward', overload='default')>), <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7ff2ef937ac0>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7ff2ef937d90>, <OpOverload(op='aten.rrelu_with_noise', overload='default')>: <function rrelu_with_noise at 0x7ff2ef770160>, <OpOverload(op='aten.rrelu_with_noise', overload='out')>: <function rrelu_with_noise at 0x7ff2ef770160>, <OpOverload(op='aten.rrelu_with_noise_', overload='default')>: <function rrelu_with_noise_ at 0x7ff2ef7701f0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7ff2ef770430>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7ff2ef770430>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7ff2ef7704c0>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7ff2ef7704c0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7ff2ef770af0>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7ff2ef770af0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7ff2ef770d30>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7ff2ef771090>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7ff2ef771000>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7ff2ef7711b0>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7ff2ef771000>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7ff2ef7712d0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7ff2ef7713f0>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7ff2ef7715a0>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7ff2ef7715a0>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7ff2ef771870>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7ff2ef771870>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7ff2ef771ab0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7ff2ef771ab0>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7ff2ef818dc0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7ff2ef79b6d0>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27a1fe3b0>, kernel=<OpOverload(op='aten.item', overload='default')>), <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819c60>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>}
- 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='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814bb50>, 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 0x7ff27814b7f0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814b910>, 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 0x7ff27814b1c0>, 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 0x7ff278150b80>, 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 0x7ff278151000>, 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 0x7ff278150a60>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781504c0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278150040>, kernel=<OpOverload(op='aten.pad', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278149990>, 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 0x7ff278149480>, 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 0x7ff278149870>, 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 0x7ff278149750>, 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 0x7ff278149360>, 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 0x7ff2781491b0>, 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 0x7ff2781489d0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814ab90>, 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 0x7ff27814a4d0>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814a830>, 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 0x7ff27814a200>, 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 0x7ff27814a440>, 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 0x7ff27813b7f0>, 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 0x7ff27813b5b0>, 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 0x7ff27813b1c0>, 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 0x7ff27813ae60>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148310>, 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 0x7ff2781480d0>, 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 0x7ff278148160>, 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 0x7ff27813bb50>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139750>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139900>, 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 0x7ff278139090>, 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 0x7ff278138dc0>, 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 0x7ff278138f70>, 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 0x7ff278138e50>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813ac20>, 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 0x7ff27813a5f0>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a7a0>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139fc0>, 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 0x7ff278139cf0>, 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 0x7ff278139e10>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168280>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781684c0>, 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 0x7ff278168310>, 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 0x7ff278163c70>, kernel=<OpOverload(op='aten.column_stack', 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 0x7ff278163b50>, 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 0x7ff278163eb0>, 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 0x7ff278163490>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814be20>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278150d30>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781495a0>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814a710>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b520>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b0a0>, 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 0x7ff27813aef0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148430>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813bf40>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148280>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>), <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813bd00>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781395a0>, kernel=<OpOverload(op='aten._remove_batch_dim', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='other_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781397e0>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139120>, 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 0x7ff2781393f0>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278138d30>, kernel=<OpOverload(op='aten.nll_loss2d', 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 0x7ff27813ad40>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813ab90>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a680>, 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 0x7ff278139f30>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139d80>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168550>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168790>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>), <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781681f0>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781683a0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>), <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163e20>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781635b0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781696c0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169b40>, kernel=<OpOverload(op='aten.matrix_exp_backward', 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 0x7ff278169900>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781695a0>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168e50>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169120>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168a60>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168af0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162050>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162170>, 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 0x7ff278161cf0>, 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 0x7ff278161f30>, 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 0x7ff278161b40>, 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 0x7ff2781615a0>, 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 0x7ff278161750>, 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 0x7ff2781630a0>, 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 0x7ff2781632e0>, 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 0x7ff278162950>, 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 0x7ff278162c20>, 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 0x7ff278162680>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162560>, 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 0x7ff278153f40>, 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 0x7ff2781601f0>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>), <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153be0>, 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 0x7ff278153e20>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153640>, kernel=<OpOverload(op='aten.conv_tbc_backward', 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 0x7ff2781539a0>, 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 0x7ff278153490>, 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 0x7ff278161120>, 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 0x7ff278161240>, 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 0x7ff278160ca0>, 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 0x7ff278160f70>, 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 0x7ff2781608b0>, 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 0x7ff278160820>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151750>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>), <OpOverload(op='aten._assert_tensor_metadata', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151a20>, kernel=<OpOverload(op='aten._assert_tensor_metadata', overload='default')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151510>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781516c0>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781511b0>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151360>, 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 0x7ff2781531c0>, 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 0x7ff2781528c0>, 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 0x7ff278152b90>, 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 0x7ff278152950>, 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 0x7ff278152320>, 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 0x7ff2781527a0>, 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 0x7ff2781520e0>, 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 0x7ff27814bbe0>, 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 0x7ff27814ba30>, 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 0x7ff27814b880>, 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 0x7ff27814b400>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814af80>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278150ca0>, 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 0x7ff278150ee0>, 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 0x7ff278150940>, 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 0x7ff278150310>, 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 0x7ff2781505e0>, 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 0x7ff2781500d0>, 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 0x7ff278149c60>, 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 0x7ff2781493f0>, 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 0x7ff278148ee0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278149090>, 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 0x7ff2781488b0>, 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 0x7ff278148af0>, 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 0x7ff27814a9e0>, 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 0x7ff27814ac20>, 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 0x7ff27814a5f0>, 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 0x7ff27814a7a0>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814a0e0>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814a320>, 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 0x7ff278149f30>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b760>, 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 0x7ff27813b880>, 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 0x7ff27813b370>, 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 0x7ff27813b130>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b010>, 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 0x7ff278148820>, kernel=<OpOverload(op='aten.cumprod_backward', 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 0x7ff2781484c0>, 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 0x7ff278148790>, 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 0x7ff278148670>, 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 0x7ff27813be20>, 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 0x7ff2781481f0>, 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 0x7ff27813bbe0>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139a20>, 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 0x7ff278139240>, 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 0x7ff278139360>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278138ee0>, 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 0x7ff279634b80>, 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 0x7ff278138c10>, 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 0x7ff278138b80>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a9e0>, 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 0x7ff27813acb0>, 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 0x7ff27813a4d0>, 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 0x7ff27813a0e0>, 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 0x7ff27813a290>, 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 0x7ff27813a200>, 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 0x7ff278139ab0>, 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 0x7ff278139ea0>, 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 0x7ff27814bc70>, 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 0x7ff27814beb0>, 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 0x7ff27814b760>, 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 0x7ff27814b640>, 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 0x7ff27814b0a0>, 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 0x7ff27814ad40>, 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 0x7ff278150af0>, 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 0x7ff278150f70>, 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 0x7ff278150670>, 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 0x7ff278150430>, 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 0x7ff278149a20>, 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 0x7ff278149cf0>, 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 0x7ff278149630>, 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 0x7ff278148f70>, 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 0x7ff278149240>, 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 0x7ff278149120>, 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 0x7ff278148d30>, 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 0x7ff27814acb0>, 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 0x7ff27814ab00>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814a170>, 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 0x7ff27814a3b0>, 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 0x7ff278149d80>, 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 0x7ff278149ea0>, 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 0x7ff27813b6d0>, 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 0x7ff27813b910>, 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 0x7ff27813b2e0>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', 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 0x7ff27810fb50>, 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 0x7ff278313be0>, 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 0x7ff2781483a0>, 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 0x7ff278148700>, 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 0x7ff27813beb0>, 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 0x7ff27813bc70>, 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 0x7ff2781396c0>, 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 0x7ff278139870>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139510>, 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 0x7ff278139000>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278138ca0>, 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 0x7ff27813a950>, 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 0x7ff27813add0>, 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 0x7ff27813a560>, 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 0x7ff27813a830>, 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 0x7ff27813a050>, 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 0x7ff27813a320>, 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 0x7ff278139b40>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163760>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814b370>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169510>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151f30>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151bd0>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160790>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161090>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160550>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151ea0>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153ac0>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160dc0>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152c20>, 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 0x7ff2781525f0>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151240>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162440>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168c10>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163a30>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153130>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160a60>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153760>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163d00>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781609d0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153010>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161e10>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169ab0>, kernel=<OpOverload(op='aten.rrelu', overload='default')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781623b0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152d40>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151120>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153a30>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161000>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152a70>, kernel=<OpOverload(op='aten.fix', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161630>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781604c0>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151ab0>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169630>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160940>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162e60>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151d80>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152e60>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781535b0>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781631c0>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781685e0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781605e0>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162a70>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168ca0>, kernel=<OpOverload(op='aten.absolute', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151480>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162f80>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163010>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163880>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152200>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160310>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781515a0>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151b40>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163be0>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139480>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148b80>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278151090>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148940>, kernel=<OpOverload(op='aten.inverse', 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 0x7ff2781501f0>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148dc0>, kernel=<OpOverload(op='aten._gather_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 0x7ff2781503a0>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814ae60>, kernel=<OpOverload(op='aten.trace_backward', 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 0x7ff278149ab0>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813aa70>, kernel=<OpOverload(op='aten.embedding_sparse_backward', 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 0x7ff279a585e0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278149000>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781391b0>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7ff2781485e0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148040>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139990>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278139630>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781508b0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781496c0>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278149bd0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781692d0>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160040>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152dd0>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161510>, 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 0x7ff278160ee0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160670>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781512d0>, kernel=<OpOverload(op='aten.smm', 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 0x7ff278168d30>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163d90>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814bf40>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814add0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814a680>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148550>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b490>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813bd90>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813ab00>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813af80>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a710>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b400>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813bac0>, 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 0x7ff2781392d0>, 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 0x7ff278150700>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153370>, 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 0x7ff2781624d0>, 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 0x7ff278161360>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781524d0>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814b130>, kernel=<OpOverload(op='aten.orgqr', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b250>, kernel=<OpOverload(op='aten.sum_to_size', 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 0x7ff2781497e0>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278150820>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_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 0x7ff27814b250>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278148c10>, kernel=<OpOverload(op='aten.to_mkldnn_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 0x7ff2781509d0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27814bac0>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813b9a0>, kernel=<OpOverload(op='aten.quantile', 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 0x7ff27814b520>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', 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 0x7ff278139c60>, 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 0x7ff27814bd90>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a170>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278168670>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163f40>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163ac0>, 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 0x7ff2781637f0>, 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 0x7ff278163520>, 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 0x7ff2781697e0>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2780eac20>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278163910>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.nonzero_numpy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169000>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169090>, 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 0x7ff2781691b0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278153d00>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781617e0>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161fc0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781532e0>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161900>, 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 0x7ff278161990>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278161d80>, 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 0x7ff2781625f0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', 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 0x7ff278139bd0>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278169a20>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781523b0>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a8c0>, 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 0x7ff2781680d0>, 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 0x7ff278168f70>, 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 0x7ff2781688b0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff2781616c0>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278162ef0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff27813a3b0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278152050>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x7ff2ef4800d0>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7ff2ef480280>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7ff2ef480430>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7ff2ef657640>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7ff2ef480310>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7ff2ef4804c0>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7ff2ef480550>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7ff2ef4805e0>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7ff2ef480c10>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7ff2ef480c10>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7ff2ef480f70>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7ff2ef480f70>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7ff2ef4811b0>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7ff2ef4811b0>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7ff2ef4813f0>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7ff2ef4817e0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7ff2ef4817e0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7ff2ef481b40>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7ff2ef481b40>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7ff2ef481ea0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7ff2ef481ea0>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7ff2ef4820e0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7ff2ef4820e0>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7ff2ef482320>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7ff2ef482560>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7ff2ef4827a0>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7ff2ef4827a0>, <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x7ff2ef4829e0>, <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7ff2ef4829e0>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7ff2ef482c20>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7ff2ef482c20>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7ff2ef4830a0>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7ff2ef4830a0>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7ff2ef483370>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7ff2ef74eb00>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7ff2ef74e9e0>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7ff2ef74ed40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7ff2ef74ed40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7ff2ef74eef0>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7ff2ef74eef0>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7ff2ef74f490>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7ff2ef74f490>, <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7ff2ef74f520>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7ff2ef74f5b0>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7ff2ef574d30>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7ff2ef574f70>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7ff2ef575120>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7ff2ef5752d0>, <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x7ff2ef5751b0>, <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x7ff2ef5751b0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7ff2ef575750>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7ff2ef576200>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7ff2ef5760e0>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7ff2ef5760e0>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7ff2ef576680>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7ff2ef576830>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7ff2ef576830>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7ff2ef576c20>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7ff2ef576dd0>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7ff2ef5770a0>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7ff2ef5770a0>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7ff2ef5772e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7ff2ef5772e0>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7ff2ef577490>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7ff2ef5776d0>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7ff2ef5776d0>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7ff2ef577910>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7ff2ef577910>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7ff2ef577b50>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7ff2ef577b50>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7ff2ef577d90>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7ff2ef577d90>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7ff2ef577e20>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7ff2ef5b4430>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7ff2ef5b4790>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7ff2ef5b4940>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7ff2ef5b4c10>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7ff2ef5b4e50>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7ff2ef5b4e50>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7ff2ef5b5000>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7ff2ef5b4820>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7ff2ef5b4ee0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7ff2ef5b5090>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7ff2ef5b5120>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7ff2ef5b5120>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7ff2ef5b51b0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7ff2ef5b5240>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7ff2ef5b52d0>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7ff2ef5b5360>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7ff2ef5b53f0>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7ff2ef5b5480>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7ff2ef5b5510>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7ff2ef5b55a0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7ff2ef5b55a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7ff2ef5b5630>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7ff2ef5b5630>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7ff2ef5b56c0>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7ff2ef5b5750>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7ff2ef5b5750>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7ff2ef5b57e0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7ff2ef5b57e0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7ff2ef5b5870>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7ff2ef5b5870>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7ff2ef5b5900>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7ff2ef5b5990>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7ff2ef5b5990>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7ff2ef5b5a20>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7ff2ef5b5a20>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7ff2ef5b5ab0>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7ff2ef5b5ab0>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7ff2ef5b5b40>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7ff2ef5b5bd0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7ff2ef5b5bd0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7ff2ef5b5c60>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7ff2ef5b5cf0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7ff2ef5b5d80>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7ff2ef5b5d80>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7ff2ef5b5e10>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7ff2ef5b5e10>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7ff2ef5b5ea0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7ff2ef5b5f30>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7ff2ef5b5fc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7ff2ef5b6050>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7ff2ef5b6050>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7ff2ef5b60e0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7ff2ef5b6170>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7ff2ef5b6200>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7ff2ef5b6290>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7ff2ef5b6320>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7ff2ef5b63b0>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7ff2ef5b6440>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7ff2ef5b6440>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7ff2ef5b64d0>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7ff2ef5b6560>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7ff2ef5b6560>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7ff2ef5b65f0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7ff2ef5b65f0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7ff2ef5b6680>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7ff2ef5b6710>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7ff2ef5b67a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7ff2ef5b67a0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7ff2ef5b6830>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7ff2ef5b6830>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7ff2ef5b68c0>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7ff2ef5b6950>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7ff2ef5b69e0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7ff2ef5b6a70>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7ff2ef5b6b00>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7ff2ef5b6b90>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7ff2ef5b6c20>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7ff2ef5b6c20>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7ff2ef5b6cb0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7ff2ef5b6cb0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7ff2ef5b6d40>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7ff2ef5b6dd0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7ff2ef5b6e60>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7ff2ef5b6ef0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7ff2ef5b6f80>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7ff2ef5b7010>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7ff2ef5b70a0>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7ff2ef5b7130>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7ff2ef5b71c0>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7ff2ef5b7250>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7ff2ef5b7250>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7ff2ef5b72e0>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7ff2ef5b72e0>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7ff2ef5b7370>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7ff2ef5b7400>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7ff2ef5b7490>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7ff2ef5b7490>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7ff2ef5b7520>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7ff2ef5b75b0>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7ff2ef5b7640>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7ff2ef5b7640>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7ff2ef5b76d0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7ff2ef5b7760>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7ff2ef5b77f0>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7ff2ef5b77f0>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7ff2ef5b7880>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7ff2ef5b7910>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7ff2ef5b79a0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7ff2ef5b7a30>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7ff2ef5b7ac0>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7ff2ef5b7b50>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7ff2ef5b7be0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7ff2ef5b7c70>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7ff2ef5b7d00>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7ff2ef5b7d90>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7ff2ef5b7d90>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7ff2ef5b7e20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7ff2ef5b7eb0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7ff2ef5b7f40>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7ff2ef5f4040>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7ff2ef5f40d0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7ff2ef5f40d0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7ff2ef5f4160>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7ff2ef5f41f0>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7ff2ef5f41f0>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7ff2ef5f4280>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7ff2ef5f4310>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7ff2ef5f43a0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7ff2ef5f4430>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7ff2ef5f44c0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function PyCapsule.alias at 0x7ff2ef5f45e0>, <OpOverload(op='aten.alias_copy', overload='out')>: <function PyCapsule.alias at 0x7ff2ef5f45e0>, <OpOverload(op='aten.as_strided_copy', overload='default')>: <function PyCapsule.as_strided at 0x7ff2ef5f4790>, <OpOverload(op='aten.as_strided_copy', overload='out')>: <function PyCapsule.as_strided at 0x7ff2ef5f4790>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function PyCapsule.diagonal at 0x7ff2ef5f4940>, <OpOverload(op='aten.expand_copy', overload='default')>: <function PyCapsule.expand at 0x7ff2ef5f4af0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function PyCapsule.expand at 0x7ff2ef5f4af0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function PyCapsule.narrow at 0x7ff2ef5f4ca0>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function PyCapsule.narrow at 0x7ff2ef5f4ca0>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function PyCapsule.squeeze at 0x7ff2ef5f4e50>, <OpOverload(op='aten.permute_copy', overload='out')>: <function PyCapsule.permute at 0x7ff2ef5f5000>, <OpOverload(op='aten.permute_copy', overload='default')>: <function PyCapsule.permute at 0x7ff2ef5f5000>, <OpOverload(op='aten.t_copy', overload='out')>: <function PyCapsule.t at 0x7ff2ef5f51b0>, <OpOverload(op='aten.t_copy', overload='default')>: <function PyCapsule.t at 0x7ff2ef5f51b0>, <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function PyCapsule.transpose at 0x7ff2ef5f5360>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function PyCapsule.transpose at 0x7ff2ef5f5360>, <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function PyCapsule.unbind at 0x7ff2ef5f5510>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function PyCapsule.unbind at 0x7ff2ef5f5510>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function PyCapsule.unsqueeze at 0x7ff2ef5f56c0>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function PyCapsule.unsqueeze at 0x7ff2ef5f56c0>, <OpOverload(op='aten.view_copy', overload='out')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function PyCapsule.view at 0x7ff2ef5f5870>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7ff2ef5f67a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7ff2ef5f6950>, <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x7ff2ef5f7490>, <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7ff2ef5f7490>, <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x7ff2ef5f7640>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7ff2ef5f7640>, <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x7ff2ef5f77f0>, <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7ff2ef5f77f0>, <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x7ff2ef5f79a0>, <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7ff2ef5f79a0>, <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x7ff2ef5f7b50>, <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7ff2ef5f7b50>, <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x7ff2ef5f7d00>, <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7ff2ef5f7d00>, <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x7ff2ef624430>, <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7ff2ef624430>, <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x7ff2ef6245e0>, <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7ff2ef6245e0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x7ff2ef624790>, <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7ff2ef624790>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x7ff2ef624940>, <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7ff2ef624940>, <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7ff2ef624f70>, <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x7ff2ef624f70>, <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x7ff2ef625120>, <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7ff2ef625120>, <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x7ff2ef6252d0>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7ff2ef6252d0>, <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x7ff2ef625480>, <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7ff2ef625480>, <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x7ff2ef625630>, <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7ff2ef625630>, <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x7ff2ef6257e0>, <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7ff2ef6257e0>, <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x7ff2ef625990>, <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7ff2ef625990>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x7ff2ef625b40>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7ff2ef625b40>, <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x7ff2ef625bd0>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x7ff2ef625c60>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7ff2ef626290>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7ff2ef626560>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7ff2ef627eb0>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7ff2ef654430>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7ff2ef654790>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7ff2ef654b80>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7ff2ef654b80>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7ff2ef654dc0>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7ff2ef655090>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7ff2ef655bd0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7ff2ef655750>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7ff2ef655d80>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7ff2ef655bd0>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7ff2ef655d80>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x7ff2ef656290>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x7ff2ef656440>, <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7ff2ef6567a0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7ff2ef6569e0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7ff2ef6569e0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7ff2ef656c20>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7ff2ef656c20>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7ff2ef657010>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7ff2ef6575b0>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7ff2ef657c70>, <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x7ff2ef657e20>, <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x7ff2ef6825f0>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7ff2ef683b50>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7ff2ef683b50>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7ff2ef6a4dc0>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7ff2ef6a4dc0>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7ff2ef6a5480>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7ff2ef6a5bd0>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7ff2ef6a5f30>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7ff2ef6a6290>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7ff2ef6a65f0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7ff2ef6a6d40>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7ff2ef6a6d40>, <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7ff2ef6c0280>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7ff2ef6c0430>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7ff2ef6c0430>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7ff2ef6c15a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7ff2ef6c1900>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7ff2ef6c1900>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7ff2ef6c2680>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7ff2ef6d4700>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7ff2ef6d4ca0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7ff2ef6d5240>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7ff2ef6d7910>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7ff2ef6f8700>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7ff2ef6f8700>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7ff2ef6f8ca0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7ff2ef6f8f70>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7ff2ef6f9240>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7ff2ef6f95a0>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7ff2ef6f9870>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7ff2ef6f9b40>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7ff2ef6fb0a0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7ff2ef6fb7f0>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7ff2ef6fbd00>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7ff2ef6fbf40>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7ff2ef7201f0>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7ff2ef7201f0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7ff2ef7203a0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7ff2ef7203a0>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7ff2ef720550>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7ff2ef720550>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7ff2ef721510>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7ff2ef7216c0>, <OpOverload(op='aten.std', overload='default')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='dim')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7ff2ef721d80>, <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7ff2ef721c60>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x7ff2ef721ea0>, <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7ff2ef7224d0>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7ff2ef723250>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7ff2ef722cb0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7ff2ef723910>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7ff2ef723f40>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7ff2ef74c3a0>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7ff2ef74ca60>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7ff2ef74cb80>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7ff2ef74cb80>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7ff2ef74cc10>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7ff2ef74cee0>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7ff2ef74d240>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7ff2ef74d3f0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7ff2ef74d3f0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7ff2ef74d630>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7ff2ef74dbd0>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7ff2ef74e050>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7ff2ef74df30>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7ff2ef74e4d0>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7ff2ef74e950>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7ff2ef771ea0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7ff2ef771ea0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7ff2ef771f30>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7ff2ef771f30>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7ff2ef7723b0>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7ff2ef7723b0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7ff2ef772440>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7ff2ef772440>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7ff2ef772830>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7ff2ef772830>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7ff2ef7729e0>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7ff2ef7729e0>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7ff2ef772b90>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7ff2ef772b90>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7ff2ef773010>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7ff2ef773010>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7ff2ef7731c0>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7ff2ef7731c0>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7ff2ef7732e0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7ff2ef7732e0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7ff2ef773760>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7ff2ef773760>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7ff2ef7739a0>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7ff2ef798040>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7ff2ef773be0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7ff2ef773d90>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7ff2ef773d90>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7ff2ef773f40>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7ff2ef773f40>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7ff2ef798160>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7ff2ef798940>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7ff2ef798af0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7ff2ef799120>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7ff2ef798ca0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7ff2ef799120>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7ff2ef798e50>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7ff2ef798e50>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7ff2ef7991b0>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7ff2ef7991b0>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7ff2ef7992d0>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7ff2ef7996c0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7ff2ef799bd0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7ff2ef799900>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7ff2ef799900>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7ff2ef799c60>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7ff2ef799b40>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7ff2ef799d80>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7ff2ef799e10>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7ff2ef79aa70>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7ff2ef79a9e0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7ff2ef799fc0>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7ff2ef79ab00>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7ff2ef79bc70>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7ff2ef79ac20>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7ff2ef79ac20>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7ff2ef79b400>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7ff2ef79b5b0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7ff2ef79bd90>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7ff2ef79b6d0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7ff2ef79bd00>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7ff2ef79beb0>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7ff2ef79beb0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7ff2ef7c05e0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7ff2ef7c05e0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7ff2ef7c0dc0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7ff2ef7c0dc0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7ff2ef7c1000>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7ff2ef7c1000>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7ff2ef7c11b0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7ff2ef7c1510>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7ff2ef7c11b0>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7ff2ef7c1090>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x7ff2ef7c13f0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7ff2ef7c1480>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7ff2ef7c1480>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7ff2ef7c1750>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7ff2ef7c1870>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7ff2ef7c1870>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7ff2ef7c1f30>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7ff2ef7c2560>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7ff2ef7c2710>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7ff2ef7c2830>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7ff2ef7c2830>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7ff2ef7c2b90>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7ff2ef7c29e0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7ff2ef7c2b90>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7ff2ef7c2dd0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7ff2ef7c2f80>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7ff2ef7c2f80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7ff2ef7c3640>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7ff2ef7c36d0>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7ff2ef7c3760>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7ff2ef7c37f0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7ff2ef7c3ac0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7ff2ef7c3b50>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7ff2ef7c3d00>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7ff2ef7c3d90>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7ff2ef7c3e20>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7ff2ef7c3eb0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7ff2ef7e8040>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7ff2ef7e8280>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7ff2ef7e8310>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7ff2ef7e8430>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7ff2ef7e89d0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7ff2ef7e89d0>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7ff2ef7e8af0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7ff2ef7e8b80>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7ff2ef7e8d30>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7ff2ef7ea170>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7ff2ef7e93f0>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7ff2ef7e93f0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7ff2ef7ea170>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7ff2ef7ea440>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7ff2ef7ea680>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7ff2ef7ea830>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7ff2ef7ea830>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7ff2ef7ead40>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7ff2ef7eb0a0>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7ff2ef7eb2e0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7ff2ef7eb5b0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb5b0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7ff2ef7eb6d0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7ff2ef7ebe20>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7ff2ef818040>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7ff2ef7eb490>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7ff2ef7eb490>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7ff2ef7ebe20>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7ff2ef8183a0>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7ff2ef8183a0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7ff2ef818160>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7ff2ef818160>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7ff2ef818d30>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7ff2ef818ee0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7ff2ef819090>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7ff2ef819090>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7ff2ef8192d0>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7ff2ef8195a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7ff2ef8195a0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7ff2ef819480>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7ff2ef819480>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819630>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef8196c0>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819750>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef8197e0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819870>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819870>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819900>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819990>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819a20>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ab0>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819b40>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819b40>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819bd0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819bd0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819cf0>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819d80>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819d80>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819e10>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819e10>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ea0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819ea0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819f30>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819fc0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a050>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a0e0>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a170>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a170>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a200>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a320>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a3b0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef81a440>, <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7ff2ef720790>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7ff2ef7215a0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7ff2ef818f70>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160d30>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.var', overload='default')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='dim')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x7ff2ef7217e0>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7ff2ef721900>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7ff2ef721bd0>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7ff2ef721870>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7ff2ef5755a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7ff2ef575900>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278149510>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7ff2ef7c2050>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7ff2ef5b4040>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7ff2ef7c2050>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7ff2ef6d6830>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7ff2ef6d6830>, <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7ff278160b80>, kernel=<OpOverload(op='aten.linear', overload='default')>), <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7ff2ef935750>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7ff2ef935750>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7ff2ef935a20>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7ff2ef935a20>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7ff2ef935cf0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7ff2ef935cf0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7ff2ef935fc0>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7ff2ef935fc0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7ff2ef936560>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7ff2ef9365f0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7ff2ef9365f0>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7ff2ef9368c0>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7ff2ef936cb0>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7ff2ef936ef0>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7ff2ef936ef0>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7ff2ef936f80>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7ff2ef936f80>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7ff2ef9371c0>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7ff2ef9371c0>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7ff2ef937490>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7ff2ef937490>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7ff2ef937760>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7ff2ef937a30>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7ff2ef937ac0>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7ff2ef937ac0>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7ff2ef937d90>, <OpOverload(op='aten.rrelu_with_noise', overload='default')>: <function rrelu_with_noise at 0x7ff2ef770160>, <OpOverload(op='aten.rrelu_with_noise', overload='out')>: <function rrelu_with_noise at 0x7ff2ef770160>, <OpOverload(op='aten.rrelu_with_noise_', overload='default')>: <function rrelu_with_noise_ at 0x7ff2ef7701f0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7ff2ef770430>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7ff2ef770430>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7ff2ef7704c0>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7ff2ef7704c0>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7ff2ef770a60>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7ff2ef770af0>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7ff2ef770af0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7ff2ef770d30>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7ff2ef771090>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7ff2ef771000>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7ff2ef7711b0>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7ff2ef771000>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7ff2ef7712d0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7ff2ef7713f0>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7ff2ef7715a0>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7ff2ef7715a0>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7ff2ef771870>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7ff2ef771870>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7ff2ef771ab0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7ff2ef771ab0>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7ff2ef6c2320>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7ff2ef6c29e0>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7ff2ef6c2d40>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7ff2ef6c3400>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7ff2ef6c3760>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7ff2ef6c3ac0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7ff2ef6c3e20>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7ff2ef6d4160>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7ff2ef6c3e20>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7ff2ef6d4430>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7ff2ef6d49d0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7ff2ef6d4f70>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7ff2ef6d56c0>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7ff2ef6d53f0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7ff2ef6d6200>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7ff2ef6d56c0>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7ff2ef6d6200>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7ff2ef6d64d0>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7ff2ef6d67a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7ff2ef6d64d0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7ff2ef6d70a0>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7ff2ef6d67a0>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7ff2ef6d70a0>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7ff2ef6d7370>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7ff2ef6d7370>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7ff2ef6d7640>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7ff2ef6d7640>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7ff2ef6d7be0>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7ff2ef6d7eb0>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7ff2ef6d7eb0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7ff2ef6f81f0>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7ff2ef6d7be0>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7ff2ef6f81f0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7ff2ef6f89d0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7ff2ef6f89d0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7ff2ef6f9e10>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7ff2ef6f9e10>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7ff2ef6fa0e0>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7ff2ef6fa3b0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7ff2ef6fa680>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7ff2ef6fa950>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7ff2ef6fac20>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7ff2ef6fa950>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7ff2ef6fac20>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7ff2ef6d5990>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7ff2ef6faef0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7ff2ef6fb2e0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7ff2ef6fb2e0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7ff2ef74f1c0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7ff2ef74e3b0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7ff2ef723130>, <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7ff2ef723520>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7ff2ef6813f0>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7ff2ef681750>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7ff2ef681ab0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7ff2ef681e10>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7ff2ef682170>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7ff2ef682dd0>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7ff2ef683130>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7ff2ef6824d0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7ff2ef682830>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7ff2ef682a70>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7ff2ef683490>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7ff2ef683490>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7ff2ef6837f0>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7ff2ef720940>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7ff2ef683eb0>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7ff2ef6a4280>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7ff2ef6a45e0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7ff2ef6a4940>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7ff2ef6a5120>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7ff2ef6a70a0>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7ff2ef6a7400>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7ff2ef6a70a0>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7ff2ef6a7760>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7ff2ef6a7ac0>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7ff2ef6a7e20>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7ff2ef6c0ca0>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7ff2ef6c0820>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7ff2ef6c1240>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7ff2ef6c0ee0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7ff2ef6c1c60>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7ff2ef6c0ee0>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7ff2ef6c1fc0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7ff2ef6c1fc0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7ff2ef680d30>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7ff2ef681090>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7ff2ef818dc0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7ff2ef79b6d0>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7ff2ef682a70>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7ff2ef627d00>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7ff2ef74eb00>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7ff2ef574b80>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7ff2ef575cf0>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7ff2ef575b40>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7ff2ef74ff40>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7ff2ef574160>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7ff2ef5749d0>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7ff2ef5744c0>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7ff2ef574820>, <OpOverload(op='aten.item', overload='default')>: <function item at 0x7ff2ef7209d0>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7ff2ef772e60>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7ff2ef819c60>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7ff2ef723130>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7ff2ef79b5b0>}