.torch_dynamo¶
modules
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 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.0005243599989626091 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.0020082280025235377 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-XRK._add_shape_information] dynamic shapes replacements={} [GraphBuilder-XRK.optimize] start with 13 nodes [GraphBuilder-XRK.optimize] #patterns=79 [GraphBuilder-XRK.optimize] start with subgraphs [GraphBuilder-XRK.optimize] done with subgraphs [GraphBuilderPatternOptimization-XRK.optimize] start with 7 nodes, 0 initializers, 79 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-XRK.optimize] same children={'SameChildrenPattern', 'SameChildrenFromInputPattern'} [GraphBuilderPatternOptimization-XRK.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-XRK.optimize] increase priority to 1 [GraphBuilderPatternOptimization-XRK.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-XRK.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.002 | max_time=TransposeMatMulPattern:0.000 [GraphBuilderPatternOptimization-XRK.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-XRK.optimize] increase priority to 3 [GraphBuilderPatternOptimization-XRK.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-XRK.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=ShapeBasedEditDistanceReshapePattern:0.000 [GraphBuilderPatternOptimization-XRK.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-XRK.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-XRK.optimize] done after 5 iterations with 4 nodes in 0.014 [GraphBuilder-XRK.optimize] done with 4 nodes in 0.015 [GraphBuilder-XRK.to_onnx] make_model 1 inits 0 params [GraphBuilder-XRK.time_evaluation_constants_] 0 [GraphBuilder-XRK._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-XRK._build_initializers] switch low/high order [GraphBuilder-XRK._build_initializers] done in 1.4080032997298986e-06s with 1 initializers, 0 large initializers [GraphBuilder-XRK._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.016679090000252472s and 4 nodes, 1 initializers, 5 inputs, 1 outputs [onnx_custom_backend] to_onnx done in 0.020001591998152435 with 4 nodes and 0 local functions. [onnx_custom_backend] starts creating InferenceSession [onnx_custom_backend] InferenceSession done in 0.002432140001474181 discrepancies: 2.9802322387695312e-08
onnx_debug_backend¶
- experimental_experiment.torch_dynamo.onnx_debug_backend(graph_module: torch.fx.GraphModule, args: List[torch.Tensor | torch.SymInt | torch.SymFloat], target_opset: int | None = None, backend: str | Callable[[ModelProto, bool | None], Any] = 'ort', verbose: int | Tuple[int, int] = 0, dump_prefix: 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
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.0008714779978618026 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.0018855080015782733 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-SAA._add_shape_information] dynamic shapes replacements={} [GraphBuilder-SAA.optimize] start with 13 nodes [GraphBuilder-SAA.optimize] #patterns=79 [GraphBuilder-SAA.optimize] start with subgraphs [GraphBuilder-SAA.optimize] done with subgraphs [GraphBuilderPatternOptimization-SAA.optimize] start with 7 nodes, 0 initializers, 79 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-SAA.optimize] same children={'SameChildrenPattern', 'SameChildrenFromInputPattern'} [GraphBuilderPatternOptimization-SAA.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-SAA.optimize] increase priority to 1 [GraphBuilderPatternOptimization-SAA.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-SAA.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=TransposeMatMulPattern:0.000 [GraphBuilderPatternOptimization-SAA.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-SAA.optimize] increase priority to 3 [GraphBuilderPatternOptimization-SAA.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-SAA.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=ShapeBasedEditDistanceReshapePattern:0.000 [GraphBuilderPatternOptimization-SAA.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-SAA.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-SAA.optimize] done after 5 iterations with 4 nodes in 0.012 [GraphBuilder-SAA.optimize] done with 4 nodes in 0.013 [GraphBuilder-SAA.to_onnx] make_model 1 inits 0 params [GraphBuilder-SAA.time_evaluation_constants_] 0 [GraphBuilder-SAA._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-SAA._build_initializers] switch low/high order [GraphBuilder-SAA._build_initializers] done in 1.243999577127397e-06s with 1 initializers, 0 large initializers [GraphBuilder-SAA._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.014142234998871572s and 4 nodes, 1 initializers, 5 inputs, 1 outputs +C init7_s2_-1_1: int64:(2,):[-1, 1] +I input0: float32:(32, 10):0.14938974380493164,-0.1857481449842453,-0.2702493965625763,0.30747660994529724,0.18371255695819855... +I input1: float32:(32,):-0.022222701460123062,-0.13555514812469482,-0.059476252645254135,0.0011606995249167085,0.1664729118347168... +I input2: float32:(3, 10):0.8376047611236572,0.33753421902656555,0.15947884321212769,-1.080767273902893,1.0693334341049194... +I input3: float32:(1, 32):-0.12194164097309113,0.07021715492010117,0.17091533541679382,-0.13579481840133667,-0.10677402466535568... +I input4: float32:(1,):[0.0815974771976471] Gemm(input2, input0, input1) -> input_1 + input_1: float32:(3, 32):-0.05730454623699188,0.7103391289710999,-0.28403976559638977,0.5995802283287048,-0.7611668705940247... Sigmoid(input_1) -> input_2 + input_2: float32:(3, 32):0.4856777787208557,0.6704760789871216,0.429463654756546,0.6455602645874023,0.3183929920196533... Reshape(input3, init7_s2_-1_1) -> l_self_modules_layers_modules_2_parameters_weight_::T10 + l_self_modules_layers_modules_2_parameters_weight_::T10: float32:(32, 1):-0.12194164097309113,0.07021715492010117,0.17091533541679382,-0.13579481840133667,-0.10677402466535568... Gemm(input_2, l_self_modules_layers_modules_2_parameters_weight_::T10, input4) -> output_0 + output_0: float32:(3, 1):[0.28920072317123413, 0.7067922353744507, 0.43465593457221985] 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.6993283033370972,-0.1905573010444641,0.695584237575531,-0.1493944525718689,0.17199116945266724... <built-in method sigmoid of type object at 0x7bf828526d60>((input_1,)) -> input_2 + input_2: torch.float32:torch.Size([3, 32]):0.33196115493774414,0.45250430703163147,0.6672080159187317,0.46272069215774536,0.5428920984268188... <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.36701521277427673,0.27723151445388794,0.24850811064243317 [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 0x7bf7495d0860>: <function out_dtype_decomp at 0x7bf748e71c60>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7bf748cf2ca0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7bf748d10ea0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7bf748d10ea0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7bf748d11260>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7bf748d11260>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7bf748d11620>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7bf748d11620>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7bf748d11bc0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7bf748d11bc0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7bf748d120c0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7bf748d120c0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7bf748d125c0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7bf748d125c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7bf748d12980>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7bf748d12980>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7bf748d12d40>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7bf748d12d40>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7bf748d12160>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7bf748d12160>, <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x7bf748d13240>, <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7bf748d13240>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7bf748d13600>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7bf748d13c40>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7bf748d13c40>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7bf748dfee80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7bf748dfee80>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7bf748dfeb60>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7bf748dfeb60>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7bf748dff4c0>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7bf748dff9c0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7bf748dff9c0>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7bf748dffc40>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7bf748dffc40>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7bf748c38220>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7bf748c38220>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7bf748c385e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7bf748c389a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7bf748c389a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7bf748c38d60>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7bf748c38d60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7bf748c38e00>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7bf748c3b100>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7bf748c39760>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7bf748c39760>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7bf748c39c60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7bf748c38900>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7bf748c38900>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7bf748c3b4c0>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7bf748c3a200>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7bf748c3a200>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7bf748c3a480>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7bf748dff1a0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7bf748c3b740>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7bf748c3b740>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7bf748c3a2a0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7bf748c3a520>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7bf748c3a660>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7bf748c3a660>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7bf748c3a7a0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7bf748c3b9c0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7bf748c3b9c0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7bf748c3a8e0>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7bf748c3aa20>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7bf748c3ab60>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7bf748c3aca0>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7bf748c3bc40>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7bf748c3ade0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7bf748c38cc0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7bf748c3bec0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7bf748c3af20>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7bf748c3b060>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7bf748c38cc0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7bf748c3b1a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7bf748c3b1a0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7bf748c3b2e0>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7bf748c3be20>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7bf748c3b420>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7bf748c3b420>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7bf748c3be20>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7bf748c3b560>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7bf748c3b560>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7bf748c3b6a0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7bf748c3b6a0>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7bf748c3b7e0>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7bf748c3b920>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7bf748c3b920>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7bf748c3bce0>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7bf748c3ba60>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7bf748c3ba60>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7bf748c3bba0>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7bf748c3bba0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7bf748c3b240>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7bf748c3ad40>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7bf748c3ad40>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7bf748c5d080>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7bf748c3aac0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7bf748c3a840>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7bf748c5d300>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7bf748c5e340>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7bf748c3a5c0>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7bf748c3a160>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7bf748c3bf60>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7bf748c5d580>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7bf748c5c0e0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7bf748c5e0c0>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7bf748c5c220>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7bf748c5c220>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7bf748c5d3a0>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7bf748c5c360>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7bf748c5c4a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7bf748c5c4a0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7bf748c5c5e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7bf748c5c5e0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7bf748c3bb00>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7bf748c5c720>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7bf748c3b600>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7bf748c5c860>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7bf748c5c9a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7bf748c5c9a0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7bf748c5cae0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7bf748c5cae0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7bf748c5de40>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7bf748c5cc20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7bf748c5cd60>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7bf748c5cea0>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7bf748c3ac00>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7bf748c3a3e0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7bf748c5cfe0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7bf748c3a3e0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7bf748c5d120>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7bf748c3a700>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7bf748c5d260>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7bf748c5ce00>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7bf748c5ce00>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7bf748c5cb80>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7bf748c5cb80>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7bf748c5e5c0>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7bf748c5c900>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7bf748c5c680>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7bf748c5e840>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7bf748c5c400>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7bf748c5c400>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7bf748c5c180>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7bf748c5d620>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7bf748c5eac0>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7bf748c5d760>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7bf748c5d760>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7bf748c5eac0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7bf748c5d9e0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7bf748c5db20>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7bf748c5db20>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7bf748c5eb60>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7bf748c5dc60>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7bf748c5dda0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7bf748c5dee0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7bf748c3ae80>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7bf748c5e020>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7bf748c3ae80>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7bf748c5e160>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7bf748c5e2a0>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7bf748c3b380>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7bf748c5e3e0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7bf748c5e520>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7bf748c3bd80>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7bf748c5e660>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7bf748c5e7a0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7bf748c5e7a0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7bf748c5e8e0>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7bf748c5ea20>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7bf748c5fa60>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7bf748c5d940>, <OpOverload(op='aten.as_strided_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x7bf748c5c540>, <OpOverload(op='aten.as_strided_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x7bf748c5c540>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7bf748c5d1c0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7bf748c5d1c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7bf748c5ede0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7bf748c5ede0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7bf748c5f100>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7bf748c5f100>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7bf748c5f740>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7bf748c5f740>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7bf748c5ec00>, <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7bf748c3b880>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7bf748c3b880>, <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x7bf748c5ec00>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7bf748c5fe20>, <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7bf748c5fe20>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7bf748c5ee80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7bf748c5ee80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7bf748c99120>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7bf748c99120>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7bf748c993a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7bf748c993a0>, <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x7bf748c99e40>, <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7bf748c99e40>, <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x7bf748c99300>, <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7bf748c99300>, <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x7bf748c9a160>, <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7bf748c9a160>, <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x7bf748c9a3e0>, <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7bf748c9a3e0>, <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x7bf748c9a660>, <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7bf748c9a660>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7bf748c9b100>, <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x7bf748c9af20>, <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7bf748c9af20>, <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x7bf748c9b100>, <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x7bf748c9b1a0>, <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7bf748c9b1a0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x7bf748c9b420>, <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7bf748c9b420>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x7bf748c9b6a0>, <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7bf748c9b6a0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x7bf748c9bd80>, <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7bf748c9bd80>, <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x7bf748c9bf60>, <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7bf748c9bf60>, <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x7bf748c9a020>, <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7bf748c9a020>, <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x7bf748cc40e0>, <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7bf748cc40e0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x7bf748cc4360>, <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7bf748cc4360>, <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x7bf748cc45e0>, <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7bf748cc45e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x7bf748cc4860>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7bf748cc4860>, <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x7bf748cc4900>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x7bf748cc49a0>, <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7bf748cc5080>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7bf748cc5080>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7bf748cc5440>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7bf748cc5440>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7bf748cc63e0>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7bf748cc6660>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7bf748cc6660>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7bf748cc6d40>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7bf748cc6d40>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7bf748cc71a0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7bf748cc71a0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7bf748cc79c0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7bf748cc77e0>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7bf748cc77e0>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7bf748cc79c0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7bf748cc7d80>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7bf748cc7d80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7bf748cf0b80>, <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x7bf748cc7ec0>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7bf748cf0ea0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7bf748cf04a0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7bf748cf04a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7bf748cf0b80>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7bf748cf0ea0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x7bf748cf1620>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7bf748cf25c0>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7bf748cf25c0>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x7bf748cf1940>, <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x7bf748cf1ee0>, <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7bf748cf1ee0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7bf748cf22a0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7bf748cf22a0>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7bf748cf2ca0>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7bf748cf2200>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7bf748cf2200>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7bf748cf2de0>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7bf748cf2de0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x7bf748cf3060>, <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x7bf748cf32e0>, <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7bf748cf3920>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7bf748cf3920>, <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x7bf748cf3ba0>, <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x7bf748cf3f60>, <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7bf748d10220>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7bf748d10220>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7bf748d104a0>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7bf748d104a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7bf748cf2f20>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7bf748d102c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7bf748d10540>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7bf748d10680>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7bf748d107c0>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7bf748dfd440>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7bf748d25da0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7bf748d25da0>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7bf748d39e40>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7bf748d277e0>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7bf748d277e0>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7bf748d38220>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7bf748d38220>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7bf748d38c20>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7bf748d38c20>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7bf748e96a20>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7bf748e96a20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7bf748dfd440>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7bf748d38f40>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7bf748d38f40>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7bf748d39440>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7bf748d39440>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7bf748d39e40>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7bf748d3a3e0>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7bf748d3a3e0>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7bf748d4d3a0>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7bf748d4d3a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7bf748d4d8a0>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7bf748d4d8a0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7bf748c5d940>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7bf748d4eca0>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7bf748d4eca0>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7bf748c5fa60>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7bf748c5d8a0>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7bf748d751c0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7bf748d751c0>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7bf748d92980>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7bf748d77600>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7bf748d77600>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7bf748d76480>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7bf748d76480>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7bf748d76d40>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7bf748d771a0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7bf748d771a0>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7bf748d75f80>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7bf748d75f80>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7bf748d777e0>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7bf748d777e0>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7bf748d77c40>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7bf748d77c40>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7bf748d92980>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7bf748d92d40>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7bf748d92d40>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std', overload='default')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='dim')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7bf748dbfec0>, <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7bf748dbfec0>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7bf748dbe0c0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7bf748dbe0c0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7bf748dbf1a0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7bf748dbf1a0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7bf748dbfce0>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7bf748dbf380>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7bf748dbf9c0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7bf748dbf9c0>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7bf748dbf560>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7bf748dbd620>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7bf748dbd620>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7bf748ddc900>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7bf748ddc900>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7bf748ddcf40>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7bf748ddd1c0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7bf748ddd1c0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7bf748ddd4e0>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7bf748dbf920>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7bf748dddb20>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7bf748dddb20>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7bf748ddcea0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7bf748ddf240>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7bf748dde840>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7bf748ddea20>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7bf748ddca40>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7bf748dded40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7bf748dded40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7bf748ddf100>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7bf748ddf100>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7bf748ddf420>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7bf748ddf600>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7bf748ddfa60>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7bf748ddfa60>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7bf748ddfce0>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7bf748dfe980>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7bf748dfff60>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7bf748dfff60>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x7bf748dfdee0>, <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x7bf748dfdee0>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7bf749002ac0>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7bf749002a20>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7bf7490013a0>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7bf7490034c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7bf7490034c0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7bf7490039c0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7bf749003880>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7bf749001800>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7bf749003920>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7bf749003920>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7bf749003c40>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7bf748e34860>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7bf748e34860>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7bf748e36840>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7bf748e34cc0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7bf748e34cc0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7bf748e34fe0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7bf748e34fe0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7bf7490005e0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7bf748e35260>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7bf748e35260>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7bf748e35080>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7bf748e36840>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7bf748e36700>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7bf748e35bc0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7bf748e35bc0>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7bf748e502c0>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7bf748e35da0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7bf748e36f20>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7bf748e37240>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7bf748e37ec0>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7bf748e36700>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7bf748e500e0>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7bf748e36f20>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7bf748e37ce0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7bf748e37560>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7bf748e37560>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7bf748e37880>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7bf748e37880>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7bf748e374c0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7bf748e374c0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7bf748e50720>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7bf748e50e00>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7bf748e50c20>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7bf748e50fe0>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7bf748e511c0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7bf748e518a0>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7bf748e518a0>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7bf748e52980>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7bf748e51e40>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7bf748e51bc0>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7bf748e52980>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7bf748e516c0>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7bf748e51bc0>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7bf748e51e40>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7bf748e52200>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7bf748e50f40>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7bf748e523e0>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7bf748e50f40>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7bf748e527a0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7bf748e528e0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7bf748e53420>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7bf748e537e0>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7bf748e537e0>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7bf748e53b00>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7bf748e53b00>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7bf748e709a0>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7bf748e53d80>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7bf748e53d80>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7bf748e70540>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7bf748e70540>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e705e0>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7bf748e71bc0>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7bf748e71580>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7bf748e71bc0>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7bf748e719e0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7bf748e71940>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7bf748e71940>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7bf748e728e0>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7bf748e71da0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7bf748e72200>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7bf748e72200>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7bf748e728e0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7bf748e72e80>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ba0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ba0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7bf748e70fe0>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7bf748e70fe0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7bf748e73560>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7bf748e73560>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7bf748e73380>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7bf748e73380>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73600>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73e20>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73740>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73880>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73e20>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e739c0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73b00>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73b00>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73c40>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73d80>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ec0>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94040>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73f60>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94180>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94180>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73f60>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e942c0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e942c0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94540>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73920>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e736a0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e731a0>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70e00>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70c20>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70c20>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94680>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e945e0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94360>, <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x7bf748d24400>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7bf748e71b20>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7bf748e71580>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7bf748fd6de0>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='default')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7bf748fafd80>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7bf748dbcd60>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7bf749003c40>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7bf748dbcd60>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7bf748dbcc20>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7bf748e53420>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7bf748fd4220>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7bf748dfe340>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7bf7493876a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7bf748dfd760>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7bf748faeb60>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7bf748dfd760>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7bf748fd4540>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7bf748fd4a40>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7bf748e35f80>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7bf748fae020>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7bf748ddd4e0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7bf748fd5300>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e705e0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7bf748e50900>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7bf748d63e20>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7bf748d63e20>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7bf748fac540>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7bf748fac540>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7bf748fac900>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7bf748fac900>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7bf748fac4a0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7bf748fac4a0>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7bf748fad4e0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7bf748fad4e0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7bf748fad580>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7bf748fad080>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7bf748fad080>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7bf748fad940>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7bf748fad940>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7bf748fafe20>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7bf748fadc60>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7bf748fadc60>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7bf748fafe20>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7bf748fadd00>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7bf748fae020>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7bf748fae3e0>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7bf748fae3e0>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7bf748faec00>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7bf748faec00>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7bf748faeca0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7bf748faefc0>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7bf748faf060>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7bf748faf560>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7bf748faf560>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7bf748faf600>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7bf748faf600>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7bf748fad1c0>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7bf748fafd80>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7bf748faf100>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7bf748fac0e0>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7bf748fac0e0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7bf748fd4400>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7bf748fd4680>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7bf748fd4680>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7bf748fd4a40>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7bf748fd4d60>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7bf748fd4d60>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7bf748fd53a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7bf748fd53a0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7bf748faff60>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7bf748fd59e0>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7bf748fd59e0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7bf748faff60>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7bf748fd5260>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7bf748fd5260>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7bf748fd4540>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7bf748fd5d00>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7bf748faeb60>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7bf748fd63e0>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7bf748fd6660>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7bf748fd7420>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7bf748fd67a0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7bf748fd67a0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7bf748fd6de0>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7bf748fd7100>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7bf748fd7100>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7bf748fd74c0>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7bf748fd74c0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7bf748fd7740>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7bf748fd7740>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7bf748fd7b00>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7bf748fd7b00>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7bf748fd7ec0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7bf748fd7ec0>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7bf749000180>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7bf749000180>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7bf749000360>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7bf749000400>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7bf749000400>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7bf7490009a0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7bf7490009a0>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7bf749000680>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7bf749000680>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7bf749000a40>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7bf749000a40>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7bf749000b80>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7bf749000e00>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7bf749000cc0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7bf749000ea0>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7bf749000d60>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7bf748fd7e20>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7bf748fd7e20>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7bf749000fe0>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7bf7490000e0>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7bf7490018a0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7bf749001800>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7bf749000c20>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7bf749002480>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7bf749002520>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7bf7490020c0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7bf7490020c0>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7bf749002660>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7bf7490027a0>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7bf749003880>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7bf748d60c20>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7bf748d60c20>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7bf748fadd00>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7bf748d13600>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7bf748d63600>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7bf748d63600>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7bf748d63a60>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7bf748d63a60>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7bf748d62b60>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7bf748d62b60>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7bf748d75620>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7bf748d75620>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7bf748d75a80>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7bf748d75a80>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7bf748d75ee0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7bf748d75ee0>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7bf748d916c0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7bf748d90540>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7bf748d90540>, <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7bf748d909a0>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7bf748d909a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7bf748fad580>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7bf748d916c0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7bf748e35f80>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7bf749000fe0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7bf748c385e0>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7bf748d76d40>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7bf748fd65c0>, <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7bf748e716c0>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7bf748e716c0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7bf748dbec00>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7bf748ddf4c0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7bf748ddf4c0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7bf748dbe660>, <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.var', overload='dim')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7bf748fd4040>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7bf749002c00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7bf748fd5300>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7bf748dbc7c0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7bf748dbc7c0>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7bf748e71b20>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7bf748e97d80>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7bf748e97d80>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7bf748d25080>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7bf748d25580>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7bf748d25580>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7bf748d242c0>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7bf748d3b240>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7bf748d242c0>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7bf748d24860>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7bf748d247c0>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7bf748d247c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7bf748d24b80>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7bf748d24b80>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7bf748d93ec0>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7bf748d93ec0>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7bf748d24860>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7bf748c39c60>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7bf748e97420>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7bf748d262a0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7bf748d262a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7bf748e37240>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7bf748d267a0>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7bf748d267a0>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7bf748d26ca0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7bf748d26ca0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7bf748d271a0>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7bf748d271a0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7bf748facf40>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7bf748d27ce0>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7bf748facfe0>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7bf748d27ce0>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7bf748d3a340>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7bf748d3a340>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7bf748d3a840>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7bf748d3a840>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7bf748dbcc20>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7bf748d3ad40>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7bf748d3ad40>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7bf748d3bf60>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7bf748d3b240>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7bf748d3b740>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7bf748d3b740>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7bf748d4fd80>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7bf748d4c5e0>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7bf748d4c5e0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7bf748d3bf60>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7bf748d4cea0>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7bf748d4cea0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7bf748d4dda0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7bf748d4dda0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7bf748d4e2a0>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7bf748d4e7a0>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7bf748d4e7a0>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7bf748dff4c0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7bf748d4e340>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7bf748d4e340>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7bf748d4efc0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7bf748c3b060>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7bf748d4f880>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7bf748d4f880>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7bf748d4fd80>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7bf748ddcf40>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7bf748d602c0>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7bf748e95580>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7bf748e96480>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7bf748e96980>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7bf748e96980>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7bf748d4e2a0>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7bf748e96e80>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7bf748e96e80>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7bf748e97380>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7bf748e97380>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7bf748e97880>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7bf7493876a0>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7bf748d602c0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7bf748e95580>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7bf748e97420>, <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7bf748e36700>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7bf748fae7a0>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x7bf7490016c0>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7bf748faeca0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7bf748e516c0>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7bf748e96480>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7bf748e72a20>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7bf748dfd1c0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7bf748dfd1c0>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7bf748dfe700>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7bf748dfe700>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7bf748dfe480>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7bf748dfca40>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7bf748dfca40>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7bf748dfccc0>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7bf748dfccc0>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7bf748dfce00>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7bf748dfce00>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7bf748ddee80>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7bf748ddee80>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7bf748dfcc20>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7bf748dfcc20>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.item', overload='default')>: <function item at 0x7bf748d93740>, <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7bf748e52660>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7bf748fd6160>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7bf748fd6160>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94400>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7bf748ddea20>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7bf748ddca40>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7bf748c3a480>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7bf748fd6660>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7bf748dbe660>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7bf748ddf6a0>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7bf748ddfe20>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7bf748dfe340>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7bf748e97880>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7bf748dfe480>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7bf748d4efc0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7bf748d25080>}
- 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.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7bf749000360>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7bf749000680>}
- 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 – sent to
create_onnx_friendly_decomposition_table
The value is:
<<<
import pprint from experimental_experiment.torch_dynamo import get_decomposition_table_dynamo pprint.pprint(get_decomposition_table_dynamo())
>>>
{<OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03ba0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4540>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7740>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf42c0>, kernel=<OpOverload(op='image.decode_png', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01b20>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf77e0>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='image._is_compiled_against_turbo', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01f80>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5580>, kernel=<OpOverload(op='image.encode_png', overload='default')>), <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00d60>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb6b560>, kernel=<OpOverload(op='aten.to_sparse_bsr', 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 0x7bf75046ab60>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da7740>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5da0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771580>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec036a0>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da4720>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5440>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4900>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb6fc0>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55440>, kernel=<OpOverload(op='aten.take_along_dim', 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 0x7bf74f88efc0>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e2a0>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e480>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0ca40>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e8e0>, 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 0x7bf74ea0cd60>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0cb80>, kernel=<OpOverload(op='aten.smm', 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 0x7bf74f5f4220>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6a20>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5e40>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f6660>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e1620>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba4b80>, kernel=<OpOverload(op='aten.inverse', 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 0x7bf750d1f920>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', 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 0x7bf74dcf79c0>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6200>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff54c20>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce1940>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6520>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3ba0>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56480>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57d80>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57060>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750448040>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1d1c0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', 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 0x7bf74f4e1440>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb69760>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc887c0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750185440>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc891c0>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>), <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750186520>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.affine_grid_generator_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750187240>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750186de0>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501867a0>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501871a0>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c2340>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4f40>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4fe0>, kernel=<OpOverload(op='aten.orgqr', 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 0x7bf74ffb5bc0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6de0>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02660>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>), <OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02a20>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c05e0>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff54180>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff56c00>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1f420>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba3560>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0f40>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6ac0>, kernel=<OpOverload(op='aten.absolute', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f40ba60>, kernel=<OpOverload(op='aten.argsort', 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 0x7bf74f4e3420>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec25bc0>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc889a0>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3740>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>), <OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec022a0>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec825c0>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0ea0>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff56160>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc8a660>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55ee0>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750185760>, 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 0x7bf74fc88680>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74eb26b60>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02f20>, kernel=<OpOverload(op='_test.get_first', overload='default')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3100>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750184220>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772480>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01620>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750186e80>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>), <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02520>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb47c0>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6340>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7740>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3560>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57ec0>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec565c0>, kernel=<OpOverload(op='aten.outer', overload='default')>), <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 0x7bf74ec54900>, 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 0x7bf74ec54c20>, 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 0x7bf74ec55d00>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb407c0>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec560c0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7506b72e0>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3600>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf63e0>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71da0>, kernel=<OpOverload(op='aten.special_erfinv', 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 0x7bf74dcf4d60>, 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 0x7bf74dcf5440>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70040>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4a40>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', 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 0x7bf74ec579c0>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7d80>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046bf60>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046aac0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4360>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', 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 0x7bf74ffb5800>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', 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 0x7bf74ffb77e0>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770540>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd2a20>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74ea0>, kernel=<OpOverload(op='aten.data', 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 0x7bf752b72e80>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56840>, kernel=<OpOverload(op='aten.pinverse', overload='default')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0d60>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da7920>, kernel=<OpOverload(op='aten.is_inference', overload='default')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3d80>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd32e0>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4098a0>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55080>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3e20>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27060>, 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 0x7bf74fdd0a40>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00cc0>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76980>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba54e0>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>), <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76c00>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b09a0>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56fc0>, 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 0x7bf74ec542c0>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72de0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf752b72ac0>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf752b73060>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c31a0>, kernel=<OpOverload(op='aten.argwhere', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2020>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046b100>, kernel=<OpOverload(op='aten.to_dense', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77380>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf752b714e0>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772840>, kernel=<OpOverload(op='aten.msort', overload='default')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcb9bc0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd27a0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba2e80>, kernel=<OpOverload(op='aten.trapz', overload='x')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55e40>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507719e0>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7ba0>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba3ba0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773240>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70c20>, 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 0x7bf74fe70720>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0b80>, kernel=<OpOverload(op='aten.linalg_ldl_factor', 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 0x7bf74dcf5b20>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770a40>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507739c0>, kernel=<OpOverload(op='aten.special_round', overload='default')>), <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb76a0>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72c00>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772b60>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6d40>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772660>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73880>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750808720>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba65c0>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046ae80>, 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 0x7bf75046b1a0>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5b20>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773a60>, kernel=<OpOverload(op='aten.cov', overload='default')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75080b100>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0d6c0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0dda0>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c3d80>, 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 0x7bf74ebb4860>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750186200>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4680>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88f7e0>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773380>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec260c0>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88d440>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4680>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5260>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb47c0>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e200>, kernel=<OpOverload(op='aten.one_hot', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff54400>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f880>, 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 0x7bf74ea0ff60>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771260>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5a80>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f6de160>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773d80>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4400>, 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 0x7bf74dcf7600>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec016c0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>), <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02de0>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03880>, kernel=<OpOverload(op='c10d_functional.broadcast', 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 0x7bf74ec55800>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', 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 0x7bf74ec57ba0>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', 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 0x7bf74fc88b80>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4860>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507736a0>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb6b1a0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3e20>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e36a0>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', 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 0x7bf74ffb4860>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', 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 0x7bf74f4e34c0>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3060>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7ce0>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e37e0>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb63e0>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0360>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4720>, kernel=<OpOverload(op='aten._sparse_compressed_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 0x7bf74fc89260>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc88f40>, 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 0x7bf750185940>, 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 0x7bf750187420>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0ea0>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71d00>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73420>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7ce0>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27600>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd07c0>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff558a0>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4e00>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71080>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6700>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0fe0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0c20>, 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 0x7bf74fdd0ae0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf54e0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57560>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501872e0>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec82fc0>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88ccc0>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec81a80>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408b80>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1f740>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27420>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e70ee80>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', 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 0x7bf74ec571a0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea75120>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea762a0>, 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 0x7bf74ea765c0>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01760>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772020>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55f80>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046aca0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec572e0>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5760>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea744a0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77d80>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73c40>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe702c0>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2700>, 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 0x7bf752b728e0>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb42d40>, 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 0x7bf74fba2d40>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73a60>, 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 0x7bf74fe71260>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72340>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba42c0>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72fc0>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb42c0>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507700e0>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7ce0>, kernel=<OpOverload(op='aten.histogramdd', 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 0x7bf750771440>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.linalg_inv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770cc0>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5620>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7c40>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4540>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507728e0>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750809080>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88e8e0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046b2e0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750184b80>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec563e0>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56ca0>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e32e0>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4ea0>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56b60>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb5080>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb53a0>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff559e0>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7c40>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0cc20>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea760c0>, kernel=<OpOverload(op='aten._backward', overload='default')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3100>, kernel=<OpOverload(op='aten.special_log1p', 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 0x7bf74fdd02c0>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec263e0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec24c20>, kernel=<OpOverload(op='aten._is_zerotensor', 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 0x7bf74ea77a60>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e2ac0>, kernel=<OpOverload(op='aten.conv_transpose1d', 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 0x7bf74f4e0040>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', 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 0x7bf74ff55c60>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea75620>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f40a160>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5800>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7060>, kernel=<OpOverload(op='aten.flipud', 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 0x7bf74ffb4c20>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', 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 0x7bf74dcf5940>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba4220>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa3e20>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7380>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e7a0>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01120>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', 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 0x7bf75046bb00>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88cd60>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77f60>, kernel=<OpOverload(op='aten.linalg_matrix_rank', 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 0x7bf74f5f4d60>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>), <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 0x7bf74ea0e3e0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4180>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>), <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88ef20>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57600>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <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 0x7bf74f88f2e0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88f4c0>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72a20>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec54220>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5a80>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4ea0>, kernel=<OpOverload(op='aten.rrelu', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5e40>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0c720>, kernel=<OpOverload(op='aten.arcsinh', 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 0x7bf74e6c1b20>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57740>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56d40>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf67a0>, kernel=<OpOverload(op='aten.lstm_cell', 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 0x7bf74ebb6ac0>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5080>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff54680>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff554e0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba60c0>, kernel=<OpOverload(op='aten.fix', 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 0x7bf74f5f4c20>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0c2c0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f65c0>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57ce0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4040>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>), <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7100>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>), <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f47c0>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd31a0>, kernel=<OpOverload(op='aten.kron', overload='default')>), <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f380>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5260>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7920>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b1c60>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56f20>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408ea0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f77e0>, kernel=<OpOverload(op='aten.chalf', overload='default')>), <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e14e0>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7b00>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408220>, 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 0x7bf74f40ade0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046b240>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57c40>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf53a0>, kernel=<OpOverload(op='_test.cat', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408400>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74eb24fe0>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55300>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74ae0>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7880>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72660>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf65c0>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4ae0>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.sym_size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773060>, kernel=<OpOverload(op='aten.sym_size', overload='int')>), <OpOverload(op='aten.special_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcb91c0>, kernel=<OpOverload(op='aten.special_softmax', overload='default')>), <OpOverload(op='aten.avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e0c0>, kernel=<OpOverload(op='aten.avg_pool1d', overload='default')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea742c0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.subtract', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71f80>, kernel=<OpOverload(op='aten.subtract', overload='Tensor')>), <OpOverload(op='aten.special_erfc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4ea0>, kernel=<OpOverload(op='aten.special_erfc', overload='default')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb43060>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten.tile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507734c0>, kernel=<OpOverload(op='aten.tile', overload='default')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1eca0>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.max_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5c60>, kernel=<OpOverload(op='aten.max_pool3d', overload='default')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02340>, kernel=<OpOverload(op='image.decode_webp', overload='default')>), <OpOverload(op='aten.linalg_det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f920>, kernel=<OpOverload(op='aten.linalg_det', overload='default')>), <OpOverload(op='aten.det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0fba0>, kernel=<OpOverload(op='aten.det', overload='default')>), <OpOverload(op='aten.cross_entropy_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f4ea0>, kernel=<OpOverload(op='aten.cross_entropy_loss', overload='default')>), <OpOverload(op='aten.log_sigmoid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7920>, kernel=<OpOverload(op='aten.log_sigmoid', overload='default')>), <OpOverload(op='aten.max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6c00>, kernel=<OpOverload(op='aten.max_pool1d', overload='default')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70540>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten.max_pool1d_with_indices', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6020>, kernel=<OpOverload(op='aten.max_pool1d_with_indices', overload='default')>), <OpOverload(op='aten.max_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6ac0>, kernel=<OpOverload(op='aten.max_pool2d', overload='default')>), <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02980>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce00e0>, kernel=<OpOverload(op='image.read_file', overload='default')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74a40>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.special_erf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb6520>, kernel=<OpOverload(op='aten.special_erf', overload='default')>), <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf72e0>, kernel=<OpOverload(op='image.decode_gif', overload='default')>), <OpOverload(op='aten.special_expm1', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff547c0>, kernel=<OpOverload(op='aten.special_expm1', overload='default')>), <OpOverload(op='aten.log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e696980>, kernel=<OpOverload(op='aten.log_softmax', overload='int')>), <OpOverload(op='aten.special_log_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f600>, kernel=<OpOverload(op='aten.special_log_softmax', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f6340>, kernel=<OpOverload(op='aten.greater_equal', overload='Tensor')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771e40>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.less', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7a60>, kernel=<OpOverload(op='aten.less', overload='Tensor')>), <OpOverload(op='aten.atleast_1d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f9c0>, kernel=<OpOverload(op='aten.atleast_1d', overload='Sequence')>), <OpOverload(op='aten.atleast_2d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5300>, kernel=<OpOverload(op='aten.atleast_2d', overload='Sequence')>), <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba4860>, kernel=<OpOverload(op='aten.bilinear', overload='default')>), <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7420>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750184f40>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>), <OpOverload(op='aten.concat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56700>, kernel=<OpOverload(op='aten.concat', overload='default')>), <OpOverload(op='aten.concatenate', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec567a0>, kernel=<OpOverload(op='aten.concatenate', overload='default')>), <OpOverload(op='aten.greater', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7d80>, kernel=<OpOverload(op='aten.greater', overload='Tensor')>), <OpOverload(op='aten.conv1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e2f20>, kernel=<OpOverload(op='aten.conv1d', overload='default')>), <OpOverload(op='aten.cross', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750184400>, kernel=<OpOverload(op='aten.cross', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4ae0>, kernel=<OpOverload(op='aten.divide', overload='Tensor')>), <OpOverload(op='aten.true_divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6a20>, kernel=<OpOverload(op='aten.true_divide', overload='Scalar')>), <OpOverload(op='aten.einsum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c13a0>, kernel=<OpOverload(op='aten.einsum', overload='default')>), <OpOverload(op='aten.embedding_bag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c0e00>, kernel=<OpOverload(op='aten.embedding_bag', overload='default')>), <OpOverload(op='aten.embedding_bag', overload='padding_idx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4220>, kernel=<OpOverload(op='aten.embedding_bag', overload='padding_idx')>), <OpOverload(op='aten.special_sinc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f060>, kernel=<OpOverload(op='aten.special_sinc', overload='default')>), <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b16c0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>), <OpOverload(op='aten.grid_sampler', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5940>, kernel=<OpOverload(op='aten.grid_sampler', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88fc40>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten.subtract', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507722a0>, kernel=<OpOverload(op='aten.subtract', overload='Scalar')>), <OpOverload(op='aten.instance_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf51c0>, kernel=<OpOverload(op='aten.instance_norm', overload='default')>), <OpOverload(op='aten.is_nonzero', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7ec0>, kernel=<OpOverload(op='aten.is_nonzero', overload='default')>), <OpOverload(op='aten.isclose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0900>, kernel=<OpOverload(op='aten.isclose', overload='default')>), <OpOverload(op='aten.layer_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57b00>, kernel=<OpOverload(op='aten.layer_norm', overload='default')>), <OpOverload(op='aten.less_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec54360>, kernel=<OpOverload(op='aten.less_equal', overload='Tensor')>), <OpOverload(op='aten.logdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3380>, kernel=<OpOverload(op='aten.logdet', overload='default')>), <OpOverload(op='aten.multiply', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7f60>, kernel=<OpOverload(op='aten.multiply', overload='Tensor')>), <OpOverload(op='aten.softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb431a0>, kernel=<OpOverload(op='aten.softmax', overload='int')>), <OpOverload(op='aten.type_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750469bc0>, kernel=<OpOverload(op='aten.type_as', overload='default')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7100>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7f60>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>), <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf49a0>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56200>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.atleast_3d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5da0>, kernel=<OpOverload(op='aten.atleast_3d', overload='Sequence')>), <OpOverload(op='aten.scaled_dot_product_attention', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea759e0>, kernel=<OpOverload(op='aten.scaled_dot_product_attention', overload='default')>), <OpOverload(op='aten.group_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0900>, kernel=<OpOverload(op='aten.group_norm', overload='default')>), <OpOverload(op='aten.conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb65c0>, kernel=<OpOverload(op='aten.conv3d', overload='default')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74b80>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4f40>, kernel=<OpOverload(op='aten.pad', overload='default')>), <OpOverload(op='aten.divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6c00>, kernel=<OpOverload(op='aten.divide', overload='Scalar')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02b60>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4b80>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408a40>, kernel=<OpOverload(op='aten.diag', overload='default')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55a80>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76fc0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b0fe0>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b1b20>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.true_divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57880>, kernel=<OpOverload(op='aten.true_divide', overload='Tensor')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76f20>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770ea0>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772ac0>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0cc0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf56c0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5a80>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>), <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7bf748cf2ca0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7bf748d10ea0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7bf748d10ea0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7bf748d11260>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7bf748d11260>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7bf748d11620>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7bf748d11620>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7bf748d11bc0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7bf748d11bc0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7bf748d120c0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7bf748d120c0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7bf748d125c0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7bf748d125c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7bf748d12980>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7bf748d12980>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7bf748d12d40>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7bf748d12d40>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7bf748d12160>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7bf748d12160>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501877e0>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7bf748d13240>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7bf748d13600>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7bf748d13c40>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7bf748d13c40>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7bf748dfee80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7bf748dfee80>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7bf748dfeb60>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7bf748dfeb60>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0fa60>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0dd00>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7bf748dff4c0>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7bf748dff9c0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7bf748dff9c0>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7bf748dffc40>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7bf748dffc40>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7bf748c38220>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7bf748c38220>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7bf748c385e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7bf748c389a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7bf748c389a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7bf748c38d60>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7bf748c38d60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7bf748c38e00>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7bf748c3b100>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7bf748c39760>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7bf748c39760>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7bf748c39c60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7bf748c38900>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7bf748c38900>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7bf748c3b4c0>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7bf748c3a200>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7bf748c3a200>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7bf748c3a480>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7bf748dff1a0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7bf748c3b740>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7bf748c3b740>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7bf748c3a2a0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7bf748c3a520>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7bf748c3a660>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7bf748c3a660>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7bf748c3a7a0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7bf748c3b9c0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7bf748c3b9c0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7bf748c3a8e0>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7bf748c3aa20>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7bf748c3ab60>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7bf748c3aca0>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7bf748c3bc40>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7bf748c3ade0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7bf748c38cc0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7bf748c3bec0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7bf748c3af20>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7bf748c3b060>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7bf748c38cc0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7bf748c3b1a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7bf748c3b1a0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7bf748c3b2e0>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7bf748c3be20>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7bf748c3b420>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7bf748c3b420>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7bf748c3be20>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7bf748c3b560>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7bf748c3b560>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7bf748c3b6a0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7bf748c3b6a0>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7bf748c3b7e0>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7bf748c3b920>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7bf748c3b920>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7bf748c3bce0>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7bf748c3ba60>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7bf748c3ba60>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7bf748c3bba0>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7bf748c3bba0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7bf748c3b240>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7bf748c3ad40>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7bf748c3ad40>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7bf748c5d080>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7bf748c3aac0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7bf748c3a840>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7bf748c5d300>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7bf748c5e340>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7bf748c3a5c0>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7bf748c3a160>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7bf748c3bf60>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7bf748c5d580>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7bf748c5c0e0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7bf748c5e0c0>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7bf748c5c220>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7bf748c5c220>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7bf748c5d3a0>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7bf748c5c360>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7bf748c5c4a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7bf748c5c4a0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7bf748c5c5e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7bf748c5c5e0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7bf748c3bb00>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7bf748c5c720>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7bf748c3b600>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7bf748c5c860>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7bf748c5c9a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7bf748c5c9a0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7bf748c5cae0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7bf748c5cae0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7bf748c5de40>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7bf748c5cc20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7bf748c5cd60>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7bf748c5cea0>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7bf748c3ac00>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7bf748c3a3e0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7bf748c5cfe0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7bf748c3a3e0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7bf748c5d120>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7bf748c3a700>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7bf748c5d260>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7bf748c5ce00>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7bf748c5ce00>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7bf748c5cb80>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7bf748c5cb80>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7bf748c5e5c0>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7bf748c5c900>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7bf748c5c680>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7bf748c5e840>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7bf748c5c400>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7bf748c5c400>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7bf748c5c180>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7bf748c5d620>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7bf748c5eac0>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7bf748c5d760>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7bf748c5d760>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7bf748c5eac0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7bf748c5d9e0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7bf748c5db20>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7bf748c5db20>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7bf748c5eb60>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7bf748c5dc60>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7bf748c5dda0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7bf748c5dee0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7bf748c3ae80>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7bf748c5e020>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7bf748c3ae80>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7bf748c5e160>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7bf748c5e2a0>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7bf748c3b380>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7bf748c5e3e0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7bf748c5e520>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7bf748c3bd80>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7bf748c5e660>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7bf748c5e7a0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7bf748c5e7a0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7bf748c5e8e0>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7bf748c5ea20>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7bf748c5fa60>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7bf748c5d940>, <OpOverload(op='aten.as_strided_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x7bf748c5c540>, <OpOverload(op='aten.as_strided_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x7bf748c5c540>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7bf748c5d1c0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7bf748c5d1c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7bf748c5ede0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7bf748c5ede0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7bf748c5f100>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7bf748c5f100>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7bf748c5f740>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7bf748c5f740>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7bf748c5ec00>, <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7bf748c3b880>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7bf748c3b880>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd36a0>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7bf748c5fe20>, <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7bf748c5fe20>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7bf748c5ee80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7bf748c5ee80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7bf748c99120>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7bf748c99120>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7bf748c993a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7bf748c993a0>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0720>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7bf748c99e40>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3f60>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7bf748c99300>, <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd14e0>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7bf748c9a160>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd1c60>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7bf748c9a3e0>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f40bd80>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7bf748c9a660>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7bf748c9b100>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f409940>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7bf748c9af20>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b0ae0>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27d80>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7bf748c9b1a0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0220>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7bf748c9b420>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74400>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7bf748c9b6a0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea75300>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7bf748c9bd80>, <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77ec0>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7bf748c9bf60>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b1940>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7bf748c9a020>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec576a0>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7bf748cc40e0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55940>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7bf748cc4360>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56340>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7bf748cc45e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56520>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7bf748cc4860>, <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73100>, 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 0x7bf74faa2ac0>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7bf748cc5080>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7bf748cc5080>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7bf748cc5440>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7bf748cc5440>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7bf748cc63e0>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7bf748cc6660>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7bf748cc6660>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7bf748cc6d40>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7bf748cc6d40>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7bf748cc71a0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7bf748cc71a0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7bf748cc79c0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7bf748cc77e0>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7bf748cc77e0>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7bf748cc79c0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7bf748cc7d80>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7bf748cc7d80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7bf748cf0b80>, <OpOverload(op='aten.selu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046ade0>, kernel=<OpOverload(op='aten.selu', overload='default')>), <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7bf748cf0ea0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7bf748cf04a0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7bf748cf04a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7bf748cf0b80>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7bf748cf0ea0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da5b20>, kernel=<OpOverload(op='aten.margin_ranking_loss', overload='default')>), <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7bf748cf25c0>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7bf748cf25c0>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5d00>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4e00>, kernel=<OpOverload(op='aten.nll_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7bf748cf1ee0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7bf748cf22a0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7bf748cf22a0>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7bf748cf2ca0>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7bf748cf2200>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7bf748cf2200>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7bf748cf2de0>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7bf748cf2de0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.prelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f7c40>, kernel=<OpOverload(op='aten.prelu', overload='default')>), <OpOverload(op='aten.relu6', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f4180>, kernel=<OpOverload(op='aten.relu6', overload='default')>), <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7bf748cf3920>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7bf748cf3920>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba77e0>, kernel=<OpOverload(op='aten.pairwise_distance', overload='default')>), <OpOverload(op='aten.pdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb400e0>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7bf748d10220>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7bf748d10220>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7bf748d104a0>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7bf748d104a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7bf748cf2f20>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7bf748d102c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7bf748d10540>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7bf748d10680>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7bf748d107c0>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7bf748dfd440>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7bf748d25da0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7bf748d25da0>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7bf748d39e40>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7bf748d277e0>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7bf748d277e0>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7bf748d38220>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7bf748d38220>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7bf748d38c20>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7bf748d38c20>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7bf748e96a20>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7bf748e96a20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7bf748dfd440>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7bf748d38f40>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7bf748d38f40>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7bf748d39440>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7bf748d39440>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7bf748d39e40>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf62a0>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7bf748d3a3e0>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7bf748d3a3e0>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7bf748d4d3a0>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7bf748d4d3a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7bf748d4d8a0>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7bf748d4d8a0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7bf748c5d940>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7bf748d4eca0>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7bf748d4eca0>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7bf748c5fa60>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7bf748c5d8a0>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7bf748d751c0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7bf748d751c0>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7bf748d92980>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7bf748d77600>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7bf748d77600>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7bf748d76480>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7bf748d76480>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7bf748d76d40>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7bf748d771a0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7bf748d771a0>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7bf748d75f80>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7bf748d75f80>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7bf748d777e0>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7bf748d777e0>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7bf748d77c40>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7bf748d77c40>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7bf748d92980>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7bf748d92d40>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7bf748d92d40>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7bf748dbe7a0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507727a0>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.any', overload='default')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71440>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770040>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507714e0>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773f60>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba3c40>, 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 0x7bf74fe709a0>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70ea0>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73560>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771bc0>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7bf748dbe5c0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7bf748dbfec0>, <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770f40>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd00e0>, 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 0x7bf74fdd0cc0>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7bf748dbfec0>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7bf748dbe0c0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7bf748dbe0c0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5120>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7bf748dbf1a0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7bf748dbf1a0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7bf748dbfce0>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7bf748dbf380>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7bf748dbf9c0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7bf748dbf9c0>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7bf748dbf560>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7bf748ddc540>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7bf748dbd620>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7bf748dbd620>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7bf748ddc900>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7bf748ddc900>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7bf748ddcf40>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7bf748ddd1c0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7bf748ddd1c0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7bf748ddd4e0>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7bf748dbf920>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7bf748dddb20>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7bf748dddb20>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f6dff60>, 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 0x7bf74f5f51c0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7bf748ddcea0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7bf748ddf240>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7bf748dde840>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7bf748ddea20>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7bf748ddca40>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7bf748dded40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7bf748dded40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7bf748ddf100>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7bf748ddf100>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7bf748ddf420>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7bf748ddf600>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7bf748ddfa60>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7bf748ddfa60>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec544a0>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec562a0>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7bf748ddfce0>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7bf748dfe980>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7bf748dfff60>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7bf748dfff60>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6980>, 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 0x7bf74ffb74c0>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7bf749002ac0>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7bf749002a20>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7bf7490013a0>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7bf7490034c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7bf7490034c0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7bf7490039c0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7bf749003880>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7bf749001800>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7bf749003920>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7bf749003920>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7bf749003c40>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7bf748e34860>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7bf748e34860>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7bf748e36840>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7bf748e34cc0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7bf748e34cc0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7bf748e34fe0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7bf748e34fe0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7bf7490005e0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea756c0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7bf748e35260>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7bf748e35260>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7bf748e35080>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7bf748e36840>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7bf748e36700>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77e20>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7bf748e35bc0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7bf748e35bc0>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7bf748e502c0>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7bf748e35da0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7bf748e36f20>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7bf748e37240>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7bf748e37ec0>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7bf748e36700>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7bf748e500e0>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7bf748e36f20>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7bf748e37ce0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7bf748e37560>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7bf748e37560>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7bf748e37880>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7bf748e37880>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7bf748e374c0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7bf748e374c0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7bf748e50720>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7bf748e50e00>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7bf748e50c20>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7bf748e50fe0>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7bf748e511c0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7bf748e518a0>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7bf748e518a0>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7bf748e52980>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7bf748e51e40>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7bf748e51bc0>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7bf748e52980>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7bf748e516c0>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7bf748e51bc0>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7bf748e516c0>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7bf748e51e40>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7bf748e52200>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7bf748e50f40>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7bf748e523e0>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7bf748e50f40>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7bf748e527a0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7bf748e528e0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7bf748e53420>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7bf748e537e0>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7bf748e537e0>, <OpOverload(op='aten.matmul', overload='default')>: <function matmul at 0x7bf748e70180>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7bf748e53b00>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7bf748e53b00>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7bf748e709a0>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7bf748e53d80>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7bf748e53d80>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7bf748e70540>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7bf748e70540>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e705e0>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7bf748e71bc0>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7bf748e71580>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7bf748e71bc0>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7bf748e719e0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7bf748e71940>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7bf748e71940>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7bf748e728e0>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7bf748e71da0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7bf748e72200>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7bf748e72200>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7bf748e728e0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7bf748e72e80>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ba0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ba0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7bf748e70fe0>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7bf748e70fe0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7bf748e73560>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7bf748e73560>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7bf748e73380>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7bf748e73380>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73600>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73e20>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73740>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73880>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73e20>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e739c0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73b00>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73b00>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73c40>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73d80>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ec0>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94040>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73f60>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94180>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94180>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73f60>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e942c0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e942c0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94540>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73920>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e736a0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e731a0>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70e00>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70c20>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70c20>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94680>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e945e0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94360>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5080>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7bf748e71b20>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f100>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7bf748e71580>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7bf748fd6de0>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4720>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf47c0>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7bf748fafd80>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7bf748dbcd60>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7bf749003c40>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7bf748dbcd60>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7bf748dbcc20>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7bf748e53420>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7bf748fd4220>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7bf748dfe340>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7bf7493876a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7bf748dfd760>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7bf748faeb60>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7bf748dfd760>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e13a0>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7bf748fd4540>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7bf748fd4a40>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7bf748e35f80>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7bf748fae020>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7bf748ddd4e0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7bf748fd5300>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e705e0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7bf748e50900>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7bf748d63e20>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7bf748d63e20>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7bf748fac540>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7bf748fac540>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7bf748fac900>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7bf748fac900>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7bf748fac4a0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7bf748fac4a0>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7bf748fad4e0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7bf748fad4e0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7bf748fad580>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7bf748fad080>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7bf748fad080>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7bf748fad940>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7bf748fad940>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7bf748fafe20>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7bf748fadc60>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7bf748fadc60>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7bf748fafe20>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7bf748fadd00>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7bf748fae020>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7bf748fae3e0>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7bf748fae3e0>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7bf748faec00>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7bf748faec00>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7bf748faeca0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7bf748faefc0>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7bf748faf060>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7bf748faf560>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7bf748faf560>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7bf748faf600>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7bf748faf600>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7bf748fad1c0>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7bf748fafd80>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7bf748faf100>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7bf748fac0e0>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7bf748fac0e0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7bf748fd4400>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7bf748fd4680>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7bf748fd4680>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7bf748fd4a40>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7bf748fd4d60>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7bf748fd4d60>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7bf748fd53a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7bf748fd53a0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7bf748faff60>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7bf748fd59e0>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7bf748fd59e0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7bf748faff60>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7bf748fd5260>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7bf748fd5260>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7bf748fd4540>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7bf748fd5d00>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7bf748fd5d00>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7bf748faeb60>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7bf748fd63e0>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7bf748fd63e0>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7bf748fd6660>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7bf748fd7420>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7bf748fd67a0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7bf748fd67a0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7bf748fd6de0>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7bf748fd7100>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7bf748fd7100>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7bf748fd74c0>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7bf748fd74c0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7bf748fd7740>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7bf748fd7740>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7bf748fd7b00>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7bf748fd7b00>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7bf748fd7ec0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7bf748fd7ec0>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7bf749000180>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7bf749000180>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7bf749000360>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7bf749000400>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7bf749000400>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7bf7490009a0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7bf7490009a0>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7bf749000680>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7bf749000680>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7bf749000a40>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7bf749000a40>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7bf749000b80>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7bf749000e00>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7bf749000cc0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7bf749000ea0>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7bf749000d60>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7bf748fd7e20>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7bf748fd7e20>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7bf749000fe0>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7bf7490000e0>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7bf7490018a0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7bf749001800>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7bf749000c20>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7bf749002480>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7bf749002520>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7bf7490020c0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7bf7490020c0>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7bf749002660>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7bf7490027a0>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7bf749003880>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7bf748d60c20>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7bf748d60c20>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7bf748fadd00>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7bf748d13600>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.where', overload='ScalarSelf')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772700>, kernel=<OpOverload(op='aten.where', overload='ScalarSelf')>), <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7bf748d63600>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7bf748d63600>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7bf748d63a60>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7bf748d63a60>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7bf748d62b60>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7bf748d62b60>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7bf748d75620>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7bf748d75620>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7bf748d75a80>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7bf748d75a80>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7bf748d75ee0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7bf748d75ee0>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7bf748d916c0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7bf748d90540>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7bf748d90540>, <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5580>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7bf748d909a0>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7bf748d909a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7bf748fad580>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7bf748d916c0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7bf748e35f80>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7bf749000fe0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7bf748c385e0>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7bf748d76d40>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7bf748fd65c0>, <OpOverload(op='aten.where', overload='ScalarOther')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773ce0>, kernel=<OpOverload(op='aten.where', overload='ScalarOther')>), <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7bf748e716c0>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7bf748e716c0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7bf748dbec00>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7bf748ddf4c0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7bf748ddf4c0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7bf748dbe660>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72e80>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771d00>, kernel=<OpOverload(op='aten.var_mean', overload='default')>), <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.where', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772fc0>, kernel=<OpOverload(op='aten.where', overload='Scalar')>), <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5440>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507711c0>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7bf748fd4040>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7bf749002c00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7bf748fd5300>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7bf748dbc7c0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7bf748dbc7c0>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff540e0>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7bf748e71b20>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7bf748e97d80>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7bf748e97d80>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7bf748d25080>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7bf748d25580>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7bf748d25580>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7bf748d242c0>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7bf748d3b240>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7bf748d242c0>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7bf748d24860>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7bf748d247c0>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7bf748d247c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7bf748d24b80>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7bf748d24b80>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7bf748d93ec0>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7bf748d93ec0>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7bf748d24860>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7bf748c39c60>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7bf748e97420>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7bf748d262a0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7bf748d262a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7bf748e37240>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7bf748d267a0>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7bf748d267a0>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7bf748d26ca0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7bf748d26ca0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7bf748d271a0>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7bf748d271a0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7bf748facf40>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7bf748d27ce0>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7bf748facfe0>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7bf748d27ce0>, <OpOverload(op='aten.isfinite', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6480>, kernel=<OpOverload(op='aten.isfinite', overload='default')>), <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7bf748d3a340>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7bf748d3a340>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7bf748d3a840>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7bf748d3a840>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7bf748dbcc20>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7bf748d3ad40>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7bf748d3ad40>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7bf748d3bf60>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7bf748d3b240>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7bf748d3b740>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7bf748d3b740>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7bf748d4fd80>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7bf748d4c5e0>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7bf748d4c5e0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7bf748d3bf60>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7bf748d4cea0>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7bf748d4cea0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7bf748d4dda0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7bf748d4dda0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7bf748d4e2a0>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7bf748d4e7a0>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7bf748d4e7a0>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7bf748dff4c0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7bf748d4e340>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7bf748d4e340>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7bf748d4efc0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7bf748c3b060>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7bf748d4f880>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7bf748d4f880>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7bf748d4fd80>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7bf748ddcf40>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7bf748d602c0>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7bf748e95580>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7bf748e96480>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7bf748e96980>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7bf748e96980>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7bf748d4e2a0>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7bf748e96e80>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7bf748e96e80>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7bf748e97380>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7bf748e97380>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7bf748e97880>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7bf7493876a0>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7bf748d602c0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7bf748e95580>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7bf748e97420>, <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea779c0>, kernel=<OpOverload(op='aten.linear', overload='default')>), <OpOverload(op='aten.conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74860>, kernel=<OpOverload(op='aten.conv2d', overload='default')>), <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7bf748e36700>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7bf748fae7a0>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x7bf7490016c0>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7bf748faeca0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7bf748e516c0>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7bf748e96480>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7bf748e72a20>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7bf748dfd1c0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7bf748dfd1c0>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7bf748dfe700>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7bf748dfe700>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7bf748dfe480>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7bf748dfca40>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7bf748dfca40>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7bf748dfccc0>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7bf748dfccc0>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7bf748dfce00>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7bf748dfce00>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7bf748ddee80>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7bf748ddee80>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7bf748dfcc20>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7bf748dfcc20>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76ca0>, kernel=<OpOverload(op='aten.item', overload='default')>), <OpOverload(op='aten.nonzero_numpy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74540>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7bf748e52660>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7bf748fd6160>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7bf748fd6160>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94400>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7bf748ddea20>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7bf748ddca40>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7bf748c3a480>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7bf748fd6660>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7bf748dbe660>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7bf748ddf6a0>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7bf748ddfe20>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7bf748dfe340>, <OpOverload(op='aten.slice', overload='Tensor')>: <function slice_forward at 0x7bf748fd4360>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7bf748e97880>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7bf748dfe480>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7bf748d4efc0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7bf748d25080>, <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce3100>, kernel=<OpOverload(op='quantized.conv3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03ce0>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce2de0>, kernel=<OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02c00>, kernel=<OpOverload(op='quantized.conv3d_transpose', overload='default')>), <OpOverload(op='quantized.conv3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03ec0>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03240>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec019e0>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec018a0>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce2160>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7d80>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03600>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7060>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec031a0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6e80>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce0360>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec005e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf59e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7240>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02fc0>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01da0>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dce16c0>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>), <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00ae0>, kernel=<OpOverload(op='quantized.linear_unpack_fp16', overload='default')>), <OpOverload(op='quantized.conv3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03b00>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03100>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02d40>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>), <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7e20>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5d00>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02e80>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>), <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03f60>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6980>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4cc0>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf76a0>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01ee0>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00fe0>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf74c0>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01c60>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01800>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}
- experimental_experiment.torch_dynamo.get_decomposition_table_onnxscript()[source]¶
Returns the decomposition table used by
torch.onnx.export().The value is:
<<<
import pprint from experimental_experiment.torch_dynamo import get_decomposition_table_onnxscript pprint.pprint(get_decomposition_table_onnxscript())
>>>
{<OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72fc0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba3560>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72e80>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71760>, kernel=<OpOverload(op='image.decode_png', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa0180>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71440>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='image._is_compiled_against_turbo', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73880>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507728e0>, kernel=<OpOverload(op='image.encode_png', overload='default')>), <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72de0>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7c40>, kernel=<OpOverload(op='aten.to_sparse_bsr', 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 0x7bf750186de0>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea744a0>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507711c0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750186200>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2b60>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea75300>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e3e0>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0dda0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f380>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5620>, kernel=<OpOverload(op='aten.take_along_dim', 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 0x7bf74f40bd80>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408ea0>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec82fc0>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b0fe0>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77f60>, 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 0x7bf74ea779c0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74540>, kernel=<OpOverload(op='aten.smm', 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 0x7bf74fdd0f40>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046b240>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0cc0>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3e20>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf62a0>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd36a0>, kernel=<OpOverload(op='aten.inverse', 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 0x7bf74dcf6480>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', 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 0x7bf750773f60>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773ce0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88efc0>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb42d40>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046aac0>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4b80>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6d40>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03ba0>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7740>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5940>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7ce0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', 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 0x7bf74dcf4860>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5c60>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f65c0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f4ea0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f4d60>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>), <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf47c0>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.affine_grid_generator_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6a20>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4680>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7ec0>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5440>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74b80>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5e40>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7ce0>, kernel=<OpOverload(op='aten.orgqr', 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 0x7bf750ba7f60>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7d80>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2ac0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>), <OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa3880>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea75620>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0cc20>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f880>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4ae0>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4f40>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e13a0>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f4220>, kernel=<OpOverload(op='aten.absolute', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56200>, kernel=<OpOverload(op='aten.argsort', 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 0x7bf74dcf7b00>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55800>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5580>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf79c0>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>), <OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71d00>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55080>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6200>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0c720>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f77e0>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7d80>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4040>, 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 0x7bf74f5f7c40>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c1b20>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f949da0>, kernel=<OpOverload(op='_test.get_first', overload='default')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6ac0>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5a80>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750184b80>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f94b6a0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4f40>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>), <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f94bba0>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88cd60>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507736a0>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7380>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6340>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02b60>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02e80>, kernel=<OpOverload(op='aten.outer', overload='default')>), <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 0x7bf74dcf77e0>, 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 0x7bf74dcf5080>, 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 0x7bf74dcf5580>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff54180>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf72e0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0360>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4ea0>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750469bc0>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb42c0>, kernel=<OpOverload(op='aten.special_erfinv', 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 0x7bf75046b100>, 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 0x7bf75046b1a0>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5080>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e200>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', 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 0x7bf74dcf6e80>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba54e0>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1d1c0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1eca0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71080>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', 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 0x7bf750ba5800>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', 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 0x7bf74f6de160>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb74c0>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3ba0>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec572e0>, kernel=<OpOverload(op='aten.data', 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 0x7bf74ff56160>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7060>, kernel=<OpOverload(op='aten.pinverse', overload='default')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3600>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76980>, kernel=<OpOverload(op='aten.is_inference', overload='default')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e37e0>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e1440>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec54360>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec031a0>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e1620>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec544a0>, 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 0x7bf74ec55d00>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74eb26b60>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec018a0>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0720>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>), <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57880>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55a80>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03f60>, 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 0x7bf74ec01f80>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb6fc0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff55c60>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da7920>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea742c0>, kernel=<OpOverload(op='aten.argwhere', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da5b20>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501867a0>, kernel=<OpOverload(op='aten.to_dense', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57c40>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff56c00>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc89260>, kernel=<OpOverload(op='aten.msort', overload='default')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c31a0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0900>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6a20>, kernel=<OpOverload(op='aten.trapz', overload='x')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7240>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7060>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507700e0>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6c00>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb76a0>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7100>, 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 0x7bf74ebb4220>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0ea0>, kernel=<OpOverload(op='aten.linalg_ldl_factor', 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 0x7bf750773d80>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750185940>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750185440>, kernel=<OpOverload(op='aten.special_round', overload='default')>), <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5940>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb5080>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc887c0>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba2e80>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6980>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4a40>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb6b560>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd2a20>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750184f40>, 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 0x7bf750186e80>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0d60>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750187420>, kernel=<OpOverload(op='aten.cov', overload='default')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb69760>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea75120>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76f20>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74860>, 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 0x7bf74ea77d80>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf63e0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f100>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408a40>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501877e0>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55440>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408b80>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046ae80>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e2a0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0cd60>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b16c0>, kernel=<OpOverload(op='aten.one_hot', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0fa60>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77a60>, 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 0x7bf74ea77ec0>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7ce0>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046ade0>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd27a0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc88f40>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770040>, 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 0x7bf7507714e0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e884ea0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>), <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2660>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73420>, kernel=<OpOverload(op='c10d_functional.broadcast', 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 0x7bf74dcf42c0>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', 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 0x7bf74dcf76a0>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', 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 0x7bf74f5f5e40>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772840>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc88b80>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf67a0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5b20>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7600>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', 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 0x7bf74f5f6660>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', 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 0x7bf74dcf5120>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5da0>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27d80>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6c00>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f47c0>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7100>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f6dff60>, kernel=<OpOverload(op='aten._sparse_compressed_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 0x7bf74f5f4c20>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f5440>, 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 0x7bf74dcf5a80>, 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 0x7bf74dcf6520>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3e20>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb53a0>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb47c0>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f51c0>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56480>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3380>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88f2e0>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88f4c0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4e00>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba4b80>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e14e0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e34c0>, 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 0x7bf74f4e2f20>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773a60>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02340>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4d60>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec563e0>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f40ade0>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec562a0>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec567a0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7ba0>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57ba0>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec571a0>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', 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 0x7bf74dcf4cc0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01da0>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55940>, 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 0x7bf74ec56ca0>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73c40>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6700>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec019e0>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7501871a0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03240>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0cb80>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03ec0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02660>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4e00>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4180>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750da7740>, 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 0x7bf74ff54400>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff547c0>, 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 0x7bf74ffb5bc0>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4fe0>, 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 0x7bf74ffb7740>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4ae0>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec263e0>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5260>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f060>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750185760>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773380>, kernel=<OpOverload(op='aten.histogramdd', 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 0x7bf74fc8a660>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.linalg_inv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb63e0>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba2d40>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771bc0>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f600>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb5800>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb6b1a0>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f409940>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1f740>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f5f6340>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00fe0>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00d60>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf65c0>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772660>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03600>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88f7e0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0d6c0>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0ca40>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0fe0>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b0ae0>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56520>, kernel=<OpOverload(op='aten._backward', overload='default')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e36a0>, kernel=<OpOverload(op='aten.special_log1p', 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 0x7bf74f4e0cc0>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec579c0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57b00>, kernel=<OpOverload(op='aten._is_zerotensor', 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 0x7bf74ec57600>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf74c0>, kernel=<OpOverload(op='aten.conv_transpose1d', 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 0x7bf74dcf59e0>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', 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 0x7bf74ea0e0c0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01c60>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55ee0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0a40>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba4860>, kernel=<OpOverload(op='aten.flipud', 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 0x7bf74f88ccc0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', 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 0x7bf750771580>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27600>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4680>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec25bc0>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b1940>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73560>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', 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 0x7bf7501872e0>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4098a0>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56b60>, kernel=<OpOverload(op='aten.linalg_matrix_rank', 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 0x7bf74fdd0b80>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>), <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 0x7bf74e7b1c60>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e7a0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>), <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408220>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec022a0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <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 0x7bf74f40a160>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f40ba60>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5a80>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec00ae0>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75080b100>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e696980>, kernel=<OpOverload(op='aten.rrelu', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec24c20>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea762a0>, kernel=<OpOverload(op='aten.arcsinh', 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 0x7bf74ea756c0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf7420>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02520>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507722a0>, kernel=<OpOverload(op='aten.lstm_cell', 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 0x7bf74ea0ff60>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507734c0>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88d440>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0e8e0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec27060>, kernel=<OpOverload(op='aten.fix', 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 0x7bf74fdd32e0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e7b09a0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd07c0>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec016c0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046b2e0>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>), <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88fc40>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>), <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0c20>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3100>, kernel=<OpOverload(op='aten.kron', overload='default')>), <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea77e20>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771d00>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba77e0>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec565c0>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02c00>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57ec0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd1c60>, kernel=<OpOverload(op='aten.chalf', overload='default')>), <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5260>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750771440>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57d80>, 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 0x7bf74ec56840>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750d1f920>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02f20>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73a60>, kernel=<OpOverload(op='_test.cat', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56700>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e6c3d80>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01800>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03ce0>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba7c40>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb47c0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770ea0>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770a40>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01b20>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff54680>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf54e0>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe702c0>, kernel=<OpOverload(op='image.decode_webp', overload='default')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb6520>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70720>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb407c0>, kernel=<OpOverload(op='image.read_file', overload='default')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02d40>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772ac0>, kernel=<OpOverload(op='image.decode_gif', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750187240>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72340>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf51c0>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f408400>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507739c0>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec260c0>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>), <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba3ba0>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf5d00>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec005e0>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe709a0>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770540>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57740>, kernel=<OpOverload(op='aten.diag', overload='default')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01760>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56340>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56fc0>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55f80>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57ce0>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc88680>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6de0>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf6980>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507727a0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0f920>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>), <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7bf748cf2ca0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7bf748d10ea0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7bf748d10ea0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7bf748d11260>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7bf748d11260>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7bf748d11620>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7bf748d11bc0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7bf748d11bc0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7bf748d120c0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7bf748d120c0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7bf748d125c0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7bf748d125c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7bf748d12980>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7bf748d12980>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7bf748d12d40>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7bf748d12160>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7bf748d12160>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4720>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7bf748d13240>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7bf748d13600>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7bf748d13c40>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7bf748d13c40>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7bf748b380e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7bf748dfee80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7bf748dfee80>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7bf748dfed40>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7bf748dfeb60>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7bf748dfeb60>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea765c0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74ea0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7bf748dff240>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7bf748dff4c0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7bf748dff9c0>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7bf748dffc40>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7bf748dfcae0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7bf748c38220>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7bf748c38220>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7bf748c385e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7bf748c389a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7bf748c389a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7bf748c38d60>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7bf748c38d60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7bf748c38e00>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7bf748c3b100>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7bf748c39760>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7bf748c39ee0>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7bf748c38900>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7bf748c3b4c0>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7bf748c3a200>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7bf748c3a200>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7bf748c3a480>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7bf748dff1a0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7bf748c3b740>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7bf748c3b740>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7bf748c3a2a0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7bf748c3a520>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7bf748c3a660>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7bf748c3a660>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7bf748c3a7a0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7bf748c3b9c0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7bf748c3b9c0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7bf748c3a8e0>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7bf748c3aa20>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7bf748c3ab60>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7bf748c3aca0>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7bf748c3bc40>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7bf748c3ade0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7bf748c38cc0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7bf748c3bec0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7bf748c3af20>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7bf748c3b060>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7bf748c38cc0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7bf748c3b1a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7bf748c3b1a0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7bf748c3b2e0>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7bf748c3be20>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7bf748c3b420>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7bf748c3b420>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7bf748c3be20>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7bf748c3b560>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7bf748c3b560>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7bf748c3b6a0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7bf748c3b6a0>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7bf748c3b7e0>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7bf748c3b920>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7bf748c3b920>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7bf748c3bce0>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7bf748c3ba60>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7bf748c3ba60>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7bf748c3bba0>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7bf748c3bba0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7bf748c3b240>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7bf748c3afc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7bf748c3ad40>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7bf748c3ad40>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7bf748c5d080>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7bf748c3aac0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7bf748c3a840>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7bf748c5d300>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7bf748c5e340>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7bf748c3a5c0>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7bf748c3a160>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7bf748c3bf60>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7bf748c5d580>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7bf748c5c0e0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7bf748c5e0c0>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7bf748c5c220>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7bf748c5c220>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7bf748c5d3a0>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7bf748c5c360>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7bf748c5c4a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7bf748c5c4a0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7bf748c5c5e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7bf748c5c5e0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7bf748c3bb00>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7bf748c5c720>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7bf748c3b600>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7bf748c5c860>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7bf748c5c9a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7bf748c5c9a0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7bf748c5cae0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7bf748c5cae0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7bf748c5de40>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7bf748c5cc20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7bf748c5cd60>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7bf748c5cea0>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7bf748c3ac00>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7bf748c3a3e0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7bf748c5cfe0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7bf748c3a3e0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7bf748c5d120>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7bf748c3a700>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7bf748c5d260>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7bf748c5ce00>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7bf748c5ce00>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7bf748c5cb80>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7bf748c5cb80>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7bf748c5e5c0>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7bf748c5c900>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7bf748c5c680>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7bf748c5e840>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7bf748c5c400>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7bf748c5c400>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7bf748c5c180>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7bf748c5d620>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7bf748c5eac0>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7bf748c5d760>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7bf748c5d760>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7bf748c5eac0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7bf748c5d9e0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7bf748c5db20>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7bf748c5db20>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7bf748c5eb60>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7bf748c5dc60>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7bf748c5dda0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7bf748c5dee0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7bf748c3ae80>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7bf748c5e020>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7bf748c3ae80>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7bf748c5e160>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7bf748c5e2a0>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7bf748c3b380>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7bf748c5e3e0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7bf748c5e520>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7bf748c3bd80>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7bf748c5e660>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7bf748c5e7a0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7bf748c5e7a0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7bf748c5e8e0>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7bf748c5ea20>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7bf748c5fa60>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7bf748c5d940>, <OpOverload(op='aten.as_strided_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x7bf748c5c540>, <OpOverload(op='aten.as_strided_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.as_strided at 0x7bf748c5c540>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7bf748c5d1c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7bf748c5ede0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7bf748c5ede0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7bf748c5f100>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7bf748c5f100>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dim_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.squeeze_copy', overload='dims_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7bf748c5f420>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7bf748c5f740>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7bf748c5f740>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7bf748c5ec00>, <OpOverload(op='aten.transpose_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7bf748c3b880>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7bf748c3b880>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e0040>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7bf748c5fe20>, <OpOverload(op='aten.unbind_copy', overload='int_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7bf748c5fe20>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7bf748c5ee80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7bf748c5ee80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.view_copy', overload='dtype_out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7bf748c5dd00>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7bf748c99120>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7bf748c993a0>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e2ac0>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7bf748c99e40>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3740>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7bf748c99300>, <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3060>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7bf748c9a160>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e3420>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7bf748c9a3e0>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57060>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7bf748c9a660>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7bf748c9b100>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec54900>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7bf748c9af20>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec57560>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55e40>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7bf748c9b1a0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec560c0>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7bf748c9b420>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02de0>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7bf748c9b6a0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03100>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7bf748c9bd80>, <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56d40>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7bf748c9bf60>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec576a0>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7bf748c9a020>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01120>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7bf748cc40e0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03880>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7bf748cc4360>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec01620>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7bf748cc45e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec03b00>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7bf748cc4860>, <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb6ac0>, 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 0x7bf750da4720>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7bf748cc5080>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7bf748cc5440>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7bf748cc63e0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7bf748cc6660>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7bf748cc6d40>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7bf748cc71a0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7bf748cc79c0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7bf748cc77e0>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7bf748cc77e0>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7bf748cc7d80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7bf748cf0b80>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7bf748cf0ea0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7bf748cf04a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7bf748cf0b80>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7bf748cf0ea0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74400>, kernel=<OpOverload(op='aten.margin_ranking_loss', overload='default')>), <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7bf748cf25c0>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7bf748cf25c0>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea0dd00>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7bf748cf1ee0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7bf748cf22a0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7bf748cf22a0>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7bf748cf2ca0>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7bf748cf2200>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7bf748cf2de0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7bf748d13100>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7bf748cf3920>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0900>, kernel=<OpOverload(op='aten.pairwise_distance', overload='default')>), <OpOverload(op='aten.pdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ff540e0>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7bf748d10220>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7bf748d104a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7bf748cf2f20>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7bf748d102c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7bf748d10540>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7bf748d10680>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7bf748d107c0>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7bf748d25da0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7bf748d25da0>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7bf748d39e40>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7bf748d277e0>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7bf748d277e0>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7bf748d38220>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7bf748d38c20>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7bf748e96a20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7bf748dfd440>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7bf748d38f40>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7bf748d39440>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7bf748d39e40>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7bf748d3bce0>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772700>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7bf748d3a3e0>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7bf748d3a3e0>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7bf748d4d3a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7bf748d4d8a0>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7bf748d4d8a0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7bf748c5d940>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7bf748d4eca0>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7bf748c5fa60>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7bf748d61da0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7bf748d61f80>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7bf748c5d8a0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7bf748d751c0>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7bf748d77600>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7bf748d76480>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7bf748d76480>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7bf748d76d40>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7bf748d771a0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7bf748d75f80>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7bf748d777e0>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7bf748d77c40>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7bf748d91da0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7bf748d92200>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7bf748d92980>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7bf748d92d40>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7bf748d93100>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7bf748d93380>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7bf748d93600>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7bf748dbc720>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7bf748dbe7a0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb65c0>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.any', overload='out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4ea0>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7bf748dbc9a0>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7bf748d936a0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb6ac0>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fc889a0>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4720>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7f60>, 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 0x7bf74ebb4540>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb5440>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ebb4860>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7bf748dbd4e0>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750186520>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7bf748dbd300>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7bf748dbe5c0>, <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7d80>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f4e32e0>, 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 0x7bf74f4e3560>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7bf748dbdc60>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7bf748dbfec0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7bf748dbe0c0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770cc0>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7bf748dbf1a0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7bf748dbf9c0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7bf748ddc540>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7bf748dbd620>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7bf748dbd620>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7bf748ddc900>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7bf748ddd1c0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7bf748ddd1c0>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7bf748dddb20>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3f60>, 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 0x7bf74fdd3d80>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7bf748dde160>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7bf748dddf80>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7bf748ddcea0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7bf748dde7a0>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7bf748ddea20>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7bf748ddca40>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7bf748dded40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7bf748dded40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7bf748ddf100>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7bf748ddf100>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7bf748ddfa60>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7bf748ddfa60>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf49a0>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7bf748ddfb00>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02980>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7bf748ddfba0>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7bf748ddf7e0>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7bf748dfff60>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7bf748dfff60>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7bf748dfd800>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7bf748dfda80>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750ba5b20>, 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 0x7bf750ba65c0>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7bf749002ac0>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7bf749002a20>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7bf7490013a0>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7bf7490034c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7bf7490034c0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7bf7490039c0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7bf749001800>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7bf749003920>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7bf749003920>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7bf749003c40>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7bf748e34860>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7bf748e34860>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7bf748e36840>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7bf748e34cc0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7bf748e34cc0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7bf748e34fe0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7bf748e34fe0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7bf7490005e0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec036a0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7bf748e35260>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7bf748e35260>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7bf748e35080>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7bf748e36840>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec55300>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7bf748e34c20>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7bf748e35bc0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7bf748e35bc0>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7bf748e502c0>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7bf748e35da0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7bf748e37240>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7bf748e37ec0>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7bf748e500e0>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7bf748e36f20>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7bf748e37ce0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7bf748e37560>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7bf748e36b60>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7bf748e37880>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7bf748e37880>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7bf748e374c0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7bf748e50720>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7bf748e50e00>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7bf748e50c20>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7bf748e50fe0>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7bf748e511c0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7bf748e518a0>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7bf748e51e40>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7bf748e52980>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7bf748e51bc0>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7bf748e516c0>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7bf748e52200>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7bf748e525c0>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7bf748e50f40>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7bf748e50f40>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7bf748e527a0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7bf748e528e0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7bf748e53420>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7bf748e537e0>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7bf748e53b00>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7bf748e53d80>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7bf748e53d80>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7bf748e70540>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7bf748e50b80>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7bf748e53740>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e70f40>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7bf748e705e0>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7bf748e71580>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7bf748e71bc0>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7bf748e71940>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7bf748e71940>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7bf748e728e0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7bf748e72200>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7bf748e72200>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7bf748e728e0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7bf748e71d00>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ba0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ba0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7bf748e70fe0>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7bf748e70fe0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7bf748e73560>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7bf748e73560>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7bf748e73380>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7bf748e73380>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73600>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73e20>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73740>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73880>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73e20>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e739c0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73b00>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73b00>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73c40>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73d80>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73ec0>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94040>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73f60>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94180>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94180>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73f60>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e942c0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e942c0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94540>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e73920>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e736a0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e731a0>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70e00>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70c20>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e70c20>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94680>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e945e0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94360>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea74ae0>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7bf748e71b20>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ea76fc0>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7bf748e71580>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7bf748fd6de0>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7bf748dbcb80>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf75046bf60>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750808720>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7bf748fafd80>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7bf748dbcd60>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7bf749003c40>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7bf748e53420>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7bf748fd4220>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e71e40>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7bf748dfe340>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7bf7493876a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7bf748dfd760>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7bf748faeb60>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74dcf4400>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7bf748fd4540>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7bf748fd4a40>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7bf748e35f80>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7bf748fae020>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7bf748ddd4e0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7bf748fd5300>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7bf748e705e0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7bf748e50900>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7bf748d63e20>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7bf748d63e20>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7bf748fac540>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7bf748fac540>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7bf748fac900>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7bf748fac900>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7bf748fac4a0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7bf748fac4a0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7bf748fad4e0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7bf748fad580>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7bf748fad080>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7bf748fad940>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7bf748fafe20>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7bf748fadc60>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7bf748fadc60>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7bf748fafe20>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7bf748fadd00>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7bf748fae020>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7bf748fae3e0>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7bf748fae3e0>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7bf748faec00>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7bf748faeca0>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7bf748faf060>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7bf748faf560>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7bf748faf560>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7bf748faf600>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7bf748faf600>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7bf748fad1c0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7bf748faf100>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7bf748fac0e0>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7bf748fac0e0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7bf748fd4400>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7bf748fd4680>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7bf748fd4680>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7bf748fd4a40>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7bf748fd4d60>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7bf748fd4d60>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7bf748fd53a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7bf748fd53a0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7bf748faff60>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7bf748fd59e0>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7bf748fd59e0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7bf748faff60>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7bf748fd5260>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7bf748fd5260>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7bf748e35620>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7bf748fd4540>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7bf748fd5d00>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7bf748fd5d00>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7bf748faeb60>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7bf748fd63e0>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7bf748fd63e0>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7bf748fd6660>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7bf748fd7420>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7bf748fd67a0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7bf748fd67a0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7bf748fd6de0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7bf748fd7100>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7bf748fd74c0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7bf748fd7740>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7bf748fd7740>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7bf748fd7b00>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7bf748fd7ec0>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7bf749000180>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7bf749000360>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7bf749000400>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7bf7490009a0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7bf7490009a0>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7bf749000680>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7bf749000680>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7bf749000a40>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7bf749000a40>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7bf749000cc0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7bf749000ea0>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7bf749000d60>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7bf749001440>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7bf748fd7e20>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7bf748fd7e20>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7bf7490000e0>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7bf7490018a0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7bf749001800>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7bf749000c20>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7bf7490020c0>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7bf749003880>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7bf748d607c0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7bf748d60c20>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7bf748fadd00>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7bf748d13600>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7bf748d61080>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7bf748d62200>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7bf748d62660>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7bf748d63600>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7bf748d63600>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7bf748d63a60>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7bf748d63a60>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7bf748d63ec0>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7bf748d62b60>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7bf748d62b60>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7bf748d74d60>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7bf748d75620>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7bf748d75620>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7bf748d75a80>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7bf748d75a80>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7bf748d75ee0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7bf748d75ee0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7bf748d768e0>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7bf748d916c0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7bf748d900e0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7bf748d90540>, <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd3100>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7bf748d909a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7bf748fad580>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7bf748d91260>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7bf748d916c0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7bf748e35f80>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7bf749000fe0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7bf748d62ac0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7bf748d91b20>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7bf748c385e0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7bf748d75b20>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7bf748dbec00>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7bf748ddf4c0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7bf748dbe660>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb4c20>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7880>, kernel=<OpOverload(op='aten.var_mean', overload='default')>), <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7bf748dbeca0>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fdd0ea0>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ffb7a60>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7bf748d93c40>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7bf748fd4040>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7bf748dfdd00>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7bf748c39120>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7bf749002c00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7bf748fd5300>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7bf748dbc7c0>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f88ef20>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7bf748e97d80>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7bf748d25580>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7bf748e73060>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7bf748d242c0>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7bf748d3b240>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7bf748d24860>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7bf748d247c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7bf748d24b80>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7bf748d24b80>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7bf748d93ec0>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7bf748c39c60>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7bf748e97420>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7bf748d262a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7bf748e37240>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7bf748d267a0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7bf748d26ca0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7bf748d271a0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7bf748d74900>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7bf748d61940>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7bf748d27ce0>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7bf748d3a340>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7bf748d3a340>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7bf748d3a840>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7bf748dbcc20>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7bf748d3ad40>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7bf748d3bf60>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7bf748d3b740>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7bf748d4c5e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7bf748d619e0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7bf748d4c9a0>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7bf748dbd260>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7bf748d4cea0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7bf748d4dda0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7bf748e70ae0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7bf748d4e2a0>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7bf748d4e7a0>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7bf748dff4c0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7bf748d4e340>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7bf748c3b060>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7bf748d4f880>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7bf748d4fd80>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7bf748ddcf40>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7bf748d602c0>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7bf748e96980>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7bf748d4e2a0>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7bf748e96e80>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7bf748e97380>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7bf7493876a0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7bf748e95580>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7bf748e97420>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7bf748d90e00>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7bf748fae7a0>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x7bf7490016c0>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7bf748faeca0>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7bf748e96480>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7bf748e72a20>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7bf748dfd1c0>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7bf748dfe700>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7bf748dfe480>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7bf748dfca40>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7bf748dfccc0>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7bf748dfce00>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7bf748ddee80>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7bf748dfcc20>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec02a20>, kernel=<OpOverload(op='aten.item', overload='default')>), <OpOverload(op='aten.nonzero_numpy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74ec56f20>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7bf748fd6160>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7bf748e94400>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7bf748ddea20>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7bf748fd6660>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7bf748dbe660>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7bf749002de0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7bf748e97880>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7bf748d4efc0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7bf748d614e0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7bf748d25080>, <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb431a0>, kernel=<OpOverload(op='quantized.conv3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa3e20>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb400e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e886f20>, kernel=<OpOverload(op='quantized.conv3d_transpose', overload='default')>), <OpOverload(op='quantized.conv3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2700>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2c00>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70040>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa0720>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb42160>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72c00>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe73100>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70c20>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71da0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fba3c40>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb414e0>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa3a60>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750772020>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71940>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa0900>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa0220>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fb43060>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>), <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74eb24fe0>, kernel=<OpOverload(op='quantized.linear_unpack_fp16', overload='default')>), <OpOverload(op='quantized.conv3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74e887920>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa0ae0>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2020>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>), <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71f80>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72660>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe72a20>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>), <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70540>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750773240>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf7507719e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe71260>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa2ca0>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74fe70ea0>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf750770f40>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74faa0360>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7bf74f94b9c0>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}