.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.000530648001586087 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.0018618679969222285 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-QMY._add_shape_information] dynamic shapes replacements={} [GraphBuilder-QMY.optimize] start with 13 nodes [GraphBuilder-QMY.optimize] #patterns=79 [GraphBuilder-QMY.optimize] start with subgraphs [GraphBuilder-QMY.optimize] done with subgraphs [GraphBuilderPatternOptimization-QMY.optimize] start with 7 nodes, 0 initializers, 79 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-QMY.optimize] same children={'SameChildrenFromInputPattern', 'SameChildrenPattern'} [GraphBuilderPatternOptimization-QMY.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-QMY.optimize] increase priority to 1 [GraphBuilderPatternOptimization-QMY.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-QMY.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=LeakyReluPattern:0.000 [GraphBuilderPatternOptimization-QMY.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-QMY.optimize] increase priority to 3 [GraphBuilderPatternOptimization-QMY.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-QMY.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=ShapeBasedEditDistanceReshapePattern:0.000 [GraphBuilderPatternOptimization-QMY.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-QMY.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-QMY.optimize] done after 5 iterations with 4 nodes in 0.008 [GraphBuilder-QMY.optimize] done with 4 nodes in 0.009 [GraphBuilder-QMY.to_onnx] make_model 1 inits 0 params [GraphBuilder-QMY.time_evaluation_constants_] 0 [GraphBuilder-QMY._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-QMY._build_initializers] switch low/high order [GraphBuilder-QMY._build_initializers] done in 8.589995559304953e-07s with 1 initializers, 0 large initializers [GraphBuilder-QMY._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.01058758600265719s and 4 nodes, 1 initializers, 5 inputs, 1 outputs [onnx_custom_backend] to_onnx done in 0.013668027000676375 with 4 nodes and 0 local functions. [onnx_custom_backend] starts creating InferenceSession [onnx_custom_backend] InferenceSession done in 0.003358818001288455 discrepancies: 0.0
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.0004808140001841821 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.0013695279994863085 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-SAI._add_shape_information] dynamic shapes replacements={} [GraphBuilder-SAI.optimize] start with 13 nodes [GraphBuilder-SAI.optimize] #patterns=79 [GraphBuilder-SAI.optimize] start with subgraphs [GraphBuilder-SAI.optimize] done with subgraphs [GraphBuilderPatternOptimization-SAI.optimize] start with 7 nodes, 0 initializers, 79 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-SAI.optimize] same children={'SameChildrenFromInputPattern', 'SameChildrenPattern'} [GraphBuilderPatternOptimization-SAI.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-SAI.optimize] increase priority to 1 [GraphBuilderPatternOptimization-SAI.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-SAI.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=TransposeEqualReshapePattern:0.000 [GraphBuilderPatternOptimization-SAI.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-SAI.optimize] increase priority to 3 [GraphBuilderPatternOptimization-SAI.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-SAI.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000 [GraphBuilderPatternOptimization-SAI.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-SAI.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-SAI.optimize] done after 5 iterations with 4 nodes in 0.008 [GraphBuilder-SAI.optimize] done with 4 nodes in 0.009 [GraphBuilder-SAI.to_onnx] make_model 1 inits 0 params [GraphBuilder-SAI.time_evaluation_constants_] 0 [GraphBuilder-SAI._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-SAI._build_initializers] switch low/high order [GraphBuilder-SAI._build_initializers] done in 7.759990694466978e-07s with 1 initializers, 0 large initializers [GraphBuilder-SAI._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.01036352600203827s and 4 nodes, 1 initializers, 5 inputs, 1 outputs +C init7_s2_-1_1: int64:(2,):[-1, 1] +I input0: float32:(32, 10):0.23481199145317078,-0.1965004950761795,0.15395349264144897,-0.3161988854408264,0.08266819268465042... +I input1: float32:(32,):0.03187233954668045,-0.020977182313799858,0.1828019917011261,0.1572691947221756,0.2998141646385193... +I input2: float32:(3, 10):-0.5961423516273499,3.9074254035949707,-0.030041033402085304,0.1973528414964676,-0.40353211760520935... +I input3: float32:(1, 32):-0.14340795576572418,0.019093912094831467,0.17521485686302185,-0.05500273033976555,0.04134605824947357... +I input4: float32:(1,):[-0.15324781835079193] Gemm(input2, input0, input1) -> input_1 + input_1: float32:(3, 32):-1.3024532794952393,-1.2254719734191895,-0.29571467638015747,-1.145869255065918,-0.5774043202400208... Sigmoid(input_1) -> input_2 + input_2: float32:(3, 32):0.21375241875648499,0.22697491943836212,0.4266054034233093,0.2412443906068802,0.35953009128570557... 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.14340795576572418,0.019093912094831467,0.17521485686302185,-0.05500273033976555,0.04134605824947357... Gemm(input_2, l_self_modules_layers_modules_2_parameters_weight_::T10, input4) -> output_0 + output_0: float32:(3, 1):[0.21915854513645172, 0.47031933069229126, 0.20489461719989777] discrepancies: 5.960464477539063e-08
dynger_backend¶
- experimental_experiment.torch_dynamo.dynger_backend(graph_module: GraphModule, args: List[Tensor | SymInt | SymFloat], dynamic_shapes: Dict[str, Any] | Tuple[Any] | None = None, optimize: bool = True, verbose: int | Tuple[int, int] = 0) Callable[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.307147741317749,0.3953583240509033,0.026103917509317398,-0.26702654361724854,0.03850489854812622... <built-in method sigmoid of type object at 0x7f707b53ec60>((input_1,)) -> input_2 + input_2: torch.float32:torch.Size([3, 32]):0.42381107807159424,0.5975719690322876,0.5065255761146545,0.43363723158836365,0.5096250176429749... <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.5680139660835266,-0.5639958381652832,-0.6018041968345642 [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 0x7f6f9b36e750>: <function out_dtype_decomp at 0x7f6f9ac1ba60>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7f6f9abe7d80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7f6f9abe7d80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <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 0x7f6f9aa21940>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7f6f9aa22f20>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7f6f9aa22f20>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7f6f9aa231a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7f6f9aa231a0>, <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x7f6f9abe5800>, <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x7f6f9aa23c40>, <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7f6f9aa23c40>, <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x7f6f9aa21620>, <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7f6f9aa21620>, <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x7f6f9aa23f60>, <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7f6f9aa23f60>, <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x7f6f9aa5c220>, <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7f6f9aa5c220>, <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x7f6f9aa5c4a0>, <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7f6f9aa5c4a0>, <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x7f6f9aa5cd60>, <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7f6f9aa5cd60>, <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x7f6f9aa5cfe0>, <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7f6f9aa5cfe0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x7f6f9aa5d260>, <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7f6f9aa5d260>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x7f6f9aa5d4e0>, <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7f6f9aa5d4e0>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7f6f9aa5db20>, <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x7f6f9aa5dbc0>, <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7f6f9aa5dbc0>, <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x7f6f9aa5db20>, <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x7f6f9aa5de40>, <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7f6f9aa5de40>, <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x7f6f9aa5ccc0>, <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7f6f9aa5ccc0>, <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x7f6f9aa5dee0>, <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7f6f9aa5dee0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x7f6f9aa5e160>, <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7f6f9aa5e160>, <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x7f6f9aa5e3e0>, <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7f6f9aa5e3e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x7f6f9aa5e660>, <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x7f6f9aa5e700>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x7f6f9aa5e7a0>, <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7f6f9aa5ee80>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7f6f9aa5ee80>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7f6f9aa5f240>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7f6f9aa5f240>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7f6f9aa8c220>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7f6f9aa8c4a0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7f6f9aa8c4a0>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7f6f9aa8cb80>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7f6f9aa8cb80>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7f6f9aa8d800>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7f6f9aa8cfe0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7f6f9aa8cfe0>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7f6f9aa8d800>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7f6f9aa8d620>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7f6f9aa8dbc0>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7f6f9aa8dbc0>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7f6f9aa8e980>, <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x7f6f9aa8dd00>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7f6f9aa8eca0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7f6f9aa8e2a0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7f6f9aa8e2a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7f6f9aa8e980>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7f6f9aa8eca0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x7f6f9aa8f420>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x7f6f9aa8f740>, <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x7f6f9aa8fce0>, <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7f6f9aa8fce0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7f6f9aaac0e0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7f6f9aaac0e0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7f6f9aaac400>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7f6f9aaac400>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7f6f9aa8e8e0>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7f6f9aa8e8e0>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7f6f9aaacc20>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7f6f9aaacc20>, <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x7f6f9aaacea0>, <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x7f6f9aaad120>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7f6f9aaadd00>, <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7f6f9aaad760>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7f6f9aaadd00>, <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x7f6f9aaad9e0>, <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x7f6f9aaadda0>, <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7f6f9aaae020>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7f6f9aaae020>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7f6f9aaae2a0>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7f6f9aaae2a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7f6f9aaacd60>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7f6f9aaae0c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7f6f9aaae340>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7f6f9aaae480>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7f6f9aaae5c0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7f6f9aaaeca0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7f6f9aaaeca0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7f6f9aaaf060>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7f6f9aaaf060>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7f6f9aaaf420>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7f6f9aaaf420>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7f6f9aaaf9c0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7f6f9aaaf9c0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7f6f9aaafec0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7f6f9aaafec0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7f6f9aac8400>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7f6f9aac8400>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7f6f9aac87c0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7f6f9aac87c0>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7f6f9aac8b80>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7f6f9aac8b80>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7f6f9aaad940>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7f6f9aaad940>, <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x7f6f9aac9080>, <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7f6f9aac9080>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7f6f9aac9440>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7f6f9aac9440>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7f6f9aac9a80>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7f6f9aac9a80>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7f6f9ab1afc0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7f6f9ab1afc0>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7f6f9ab2c2c0>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7f6f9ab2c2c0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7f6f9ab2cb80>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7f6f9ab2cb80>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7f6f9ab2d440>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7f6f9ab2cfe0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7f6f9ab2cfe0>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7f6f9ab2d440>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7f6f9ab2d3a0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7f6f9ab2d3a0>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7f6f9ab2d620>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7f6f9ab2d620>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7f6f9ab2da80>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7f6f9ab2da80>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7f6f9ab447c0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7f6f9ab44b80>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7f6f9ab44b80>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.std', overload='default')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='dim')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7f6f9ab47ec0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7f6f9ab47ec0>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7f6f9ab75a80>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7f6f9ab74fe0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7f6f9ab75b20>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7f6f9ab751c0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7f6f9ab76200>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7f6f9ab75800>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7f6f9ab75800>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7f6f9ab8c540>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7f6f9ab76200>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7f6f9ab74400>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7f6f9ab74400>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7f6f9ab76700>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7f6f9ab76700>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7f6f9ab76d40>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7f6f9ab76d40>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7f6f9ab76fc0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7f6f9ab76fc0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7f6f9ab772e0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7f6f9ab772e0>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7f6f9ab77920>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7f6f9ab77920>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7f6f9ab77240>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7f6f9ab8d080>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7f6f9ab8c860>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7f6f9ab8c0e0>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7f6f9ab8cb80>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7f6f9ab8cb80>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7f6f9ab8cf40>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7f6f9ab8cf40>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7f6f9ab8d3a0>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7f6f9ab8d260>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7f6f9ab8d440>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7f6f9ab8d8a0>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7f6f9ab8d8a0>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7f6f9ab8db20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7f6f9ab8f240>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x7f6f9ab8fce0>, <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x7f6f9ab8fce0>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7f6f9abb47c0>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7f6f9abb49a0>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7f6f9abb49a0>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7f6f9abb5e40>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7f6f9abb5e40>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7f6f9abb5300>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7f6f9abb5300>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7f6f9abb5800>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7f6f9abb5800>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7f6f9abb5760>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7f6f9abb5a80>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7f6f9abb5a80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7f6f9abb5760>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7f6f9abb6020>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7f6f9abb6020>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7f6f9abb63e0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7f6f9abb63e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7f6f9abb67a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7f6f9abb67a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7f6f9abb6b60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7f6f9abb6c00>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7f6f9abb7560>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7f6f9abb7560>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7f6f9abb7a60>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7f6f9abb7a60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7f6f9abb5f80>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7f6f9abb5f80>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7f6f9abb7f60>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7f6f9abb7f60>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7f6f9abe59e0>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7f6f9abe42c0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7f6f9abe51c0>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7f6f9abb7c40>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7f6f9abe40e0>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7f6f9abe5440>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7f6f9abe4360>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7f6f9abe44a0>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7f6f9abe44a0>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7f6f9abe56c0>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7f6f9abe45e0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7f6f9abe56c0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7f6f9abe4720>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7f6f9abe6480>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7f6f9abe4860>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7f6f9abe49a0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7f6f9abe5940>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7f6f9abe4ae0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7f6f9abe5940>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7f6f9abe4c20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7f6f9aa20860>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7f6f9abe4d60>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7f6f9abe5bc0>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7f6f9abe4ea0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7f6f9abe4ea0>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7f6f9abe6200>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7f6f9abe4fe0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7f6f9abe4fe0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7f6f9abe5120>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7f6f9abe5260>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7f6f9abe5260>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7f6f9abe5c60>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7f6f9abe53a0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7f6f9abe53a0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7f6f9abe54e0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7f6f9abe54e0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7f6f9abe5b20>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7f6f9abe5620>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7f6f9abe5760>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7f6f9abe5760>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7f6f9abe5f80>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7f6f9abe5b20>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7f6f9abe58a0>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7f6f9abe58a0>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7f6f9abe59e0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7f6f9abe6700>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7f6f9abe4cc0>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7f6f9abe4cc0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7f6f9abe6980>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7f6f9abe4a40>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7f6f9abe47c0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7f6f9abe6c00>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7f6f9abe4540>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7f6f9abe4220>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7f6f9abe5da0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7f6f9abe6e80>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7f6f9abe5ee0>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7f6f9abe7100>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7f6f9abe6020>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7f6f9abe6020>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7f6f9abe6160>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7f6f9abe62a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7f6f9abe62a0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7f6f9abe7ec0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7f6f9abe63e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7f6f9abe63e0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7f6f9abe6520>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7f6f9abe6660>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7f6f9abe67a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7f6f9abe67a0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7f6f9abe72e0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7f6f9abe68e0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7f6f9abe68e0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7f6f9abe6a20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7f6f9abe6b60>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7f6f9aa218a0>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7f6f9abe6ca0>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7f6f9abe71a0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7f6f9abe6de0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7f6f9abe71a0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7f6f9abe6f20>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7f6f9abe7060>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7f6f9abe7c40>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7f6f9abe4040>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7f6f9abe4040>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7f6f9abe4680>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7f6f9abe4680>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7f6f9abe4b80>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7f6f9abe6340>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7f6f9abe5080>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7f6f9abe5580>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7f6f9abe5580>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7f6f9abe5e40>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7f6f9abe5a80>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7f6f9abe7420>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7f6f9abe4900>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7f6f9abe7560>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7f6f9abe7560>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7f6f9abe76a0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7f6f9abe77e0>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7f6f9abe7920>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7f6f9abe7a60>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7f6f9abe7ba0>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7f6f9abe5d00>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7f6f9abe7ce0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7f6f9abe7e20>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7f6f9abe7f60>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7f6f9abe7600>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7f6f9aa200e0>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7f6f9abe7600>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7f6f9aa20220>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7f6f9aa20360>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7f6f9aa204a0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7f6f9abe7880>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7f6f9abe7b00>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7f6f9aa205e0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7f6f9aa205e0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7f6f9abe7b00>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7f6f9aa20720>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7f6f9abe6d40>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7f6f9abe6d40>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7f6f9abe6ac0>, <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 0x7f6f9abe6ac0>, <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 0x7f6f9aa20900>, <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 0x7f6f9aa20900>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7f6f9aa202c0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7f6f9aa202c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7f6f9aa20c20>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7f6f9aa20c20>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7f6f9aa20f40>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7f6f9aa20f40>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <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 0x7f6f9aa21260>, <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 0x7f6f9aa21260>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7f6f9aa21580>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7f6f9aa21580>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7f6f9aa218a0>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7f6f9abe4e00>, <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 0x7f6f9abe4e00>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7f6f9ac1a700>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7f6f9ac19b20>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7f6f9ac19b20>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7f6f9ac1a2a0>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7f6f9ac1a2a0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac185e0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac185e0>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7f6f9ac1b9c0>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7f6f9ac1b380>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7f6f9ac1b380>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7f6f9ac1b9c0>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7f6f9ac1b7e0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7f6f9ac1b740>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7f6f9ac1b740>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7f6f9ac1bba0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7f6f9ac1bf60>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7f6f9ac1bf60>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e0c0>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7f6f9ac3ccc0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7f6f9ac3ca40>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7f6f9ac3ca40>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e020>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7f6f9ac3d3a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7f6f9ac3d3a0>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7f6f9ac3d1c0>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d440>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d580>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d6c0>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d800>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e2a0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d940>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d940>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3da80>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dbc0>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dd00>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3de40>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d080>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3df80>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3df80>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e0c0>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dda0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dda0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3db20>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d8a0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d620>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d300>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3ce00>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3ce00>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e340>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e480>, <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x7f6f9acb6200>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7f6f9acb7ba0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7f6f9acb7ba0>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7f6f9accd620>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7f6f9accd620>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7f6f9acce020>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7f6f9acce020>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7f6f9accea20>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7f6f9accdbc0>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7f6f9accdbc0>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7f6f9acced40>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7f6f9acced40>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7f6f9accf240>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7f6f9accf240>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7f6f9accfc40>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7f6f9accfc40>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7f6f9ace1620>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7f6f9ace1620>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7f6f9ace31a0>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7f6f9ace31a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7f6f9ace36a0>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7f6f9ace36a0>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7f6f9acf8ae0>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7f6f9acf8ae0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7f6f9add3d80>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7f6f9ab1a2a0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7f6f9accea20>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7f6f9ad5a340>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7f6f9ad5a340>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7f6f9ad5ae80>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7f6f9ad5ae80>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7f6f9ad5a700>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7f6f9ad5a700>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7f6f9ad5a2a0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7f6f9ad5a2a0>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7f6f9ad5b2e0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7f6f9ad5b2e0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7f6f9ad5b380>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7f6f9ad5b380>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7f6f9ad5b740>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7f6f9ad5bc40>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7f6f9ad5bc40>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7f6f9ad5ba60>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7f6f9ad5bb00>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7f6f9ad5be20>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7f6f9ad5be20>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7f6f9ad7c220>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7f6f9ad7c220>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7f6f9ad7c5e0>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7f6f9ad7ca40>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7f6f9ad7cae0>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7f6f9ad7cae0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7f6f9ad7ce00>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7f6f9ad7cea0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7f6f9ad7d3a0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7f6f9ad7d3a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7f6f9ad7d440>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7f6f9ad7d440>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7f6f9ad7f7e0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7f6f9ad7dd00>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7f6f9ad7f7e0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7f6f9ad7d940>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7f6f9ad7d940>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7f6f9ad7ccc0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7f6f9ad7dee0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7f6f9ad7dc60>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7f6f9ad7dc60>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7f6f9ad7e0c0>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7f6f9ada5580>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7f6f9ad7e2a0>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7f6f9ada49a0>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7f6f9ad7e520>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7f6f9ad7e520>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7f6f9ad7e8e0>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7f6f9ad7e8e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7f6f9ad7ec00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7f6f9ad7f1a0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7f6f9ad7f1a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7f6f9ad7f240>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7f6f9ad7f240>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7f6f9ad7f880>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7f6f9ad7f880>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7f6f9ad7de40>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7f6f9ad7de40>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7f6f9ada5260>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7f6f9ad7f600>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7f6f9ad7f600>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7f6f9ad7fb00>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7f6f9ad7ed40>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7f6f9ada4220>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7f6f9ada44a0>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7f6f9ada45e0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7f6f9ada45e0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7f6f9ada4c20>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7f6f9ada4c20>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7f6f9ada4f40>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7f6f9ada4f40>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7f6f9ada5300>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7f6f9ada5300>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7f6f9ada5580>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7f6f9ada5940>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7f6f9ada5940>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7f6f9ada5d00>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7f6f9ada5f80>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7f6f9ada5f80>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7f6f9ada6200>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7f6f9ada6200>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7f6f9ada67a0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7f6f9ada67a0>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7f6f9ada6840>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7f6f9ada6840>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7f6f9ada6980>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7f6f9ada6ac0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7f6f9ada5ee0>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7f6f9ada5c60>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7f6f9ada6ca0>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7f6f9ada6ca0>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7f6f9ada7420>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7f6f9ada6160>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7f6f9ada6160>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7f6f9ada7380>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7f6f9ada7560>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7f6f9add1800>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7f6f9add16c0>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7f6f9add0180>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7f6f9ada7d80>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7f6f9add0220>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7f6f9add04a0>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7f6f9add0360>, <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7f6f9add0720>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7f6f9add07c0>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7f6f9add0900>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7f6f9ada7ec0>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7f6f9ada7ec0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7f6f9add0400>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7f6f9add11c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7f6f9add11c0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7f6f9add1620>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7f6f9add2fc0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7f6f9add0400>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7f6f9add1f80>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7f6f9add1f80>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7f6f9add2660>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7f6f9add2660>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7f6f9add31a0>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7f6f9add31a0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7f6f9add2ac0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7f6f9add2ac0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7f6f9add2de0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7f6f9add2de0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7f6f9add3060>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7f6f9add3060>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7f6f9add2e80>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7f6f9add39c0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7f6f9add39c0>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7f6f9adf8540>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7f6f9add3ba0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7f6f9adf8d60>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7f6f9adf9080>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7f6f9adf8fe0>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7f6f9adf8540>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7f6f9adf8d60>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7f6f9adf93a0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7f6f9adf8fe0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7f6f9adf9080>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7f6f9adf96c0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7f6f9adf93a0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7f6f9adf96c0>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7f6f9add1760>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7f6f9adf9a80>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7f6f9adfa020>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7f6f9add1760>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7f6f9adf9c60>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7f6f9adf9e40>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7f6f9adfa480>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7f6f9adfa660>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7f6f9adfa980>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7f6f9adfab60>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7f6f9adfbba0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7f6f9adfb600>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7f6f9adfb420>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7f6f9adfb920>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7f6f9adfb920>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7f6f9adfbba0>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7f6f9adfa8e0>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7f6f9adfbf60>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7f6f9adfa8e0>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7f6f9ac18180>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7f6f9ac18540>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7f6f9adf9f80>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7f6f9adf9f80>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7f6f9ac1bba0>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7f6f9ac191c0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7f6f9ac191c0>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7f6f9ac1bc40>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7f6f9ac19580>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7f6f9ac19580>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7f6f9ac198a0>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7f6f9ac198a0>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7f6f9ab1b420>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7f6f9ab1b420>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7f6f9ab1b880>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7f6f9ab1b880>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7f6f9ab1bce0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7f6f9ab1bce0>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7f6f9ab46a20>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7f6f9ad5ba60>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7f6f9ab2e340>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7f6f9ab2e340>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7f6f9ab2e7a0>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7f6f9ab2e7a0>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7f6f9ad5b740>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7f6f9ab2f4c0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7f6f9ab2f4c0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7f6f9ab447c0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7f6f9adfaf20>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e2a0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7f6f9b19afc0>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7f6f9b19afc0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7f6f9ac3c900>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7f6f9ac3c900>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7f6f9ab77a60>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7f6f9ab8d300>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7f6f9ab744a0>, <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7f6f9aaad760>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7f6f9add3d80>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e020>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7f6f9ab1a2a0>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7f6f9adfad40>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7f6f9ab465c0>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7f6f9ac3c860>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7f6f9ab465c0>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7f6f9abe5800>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.var', overload='default')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='dim')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7f6f9ab46b60>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7f6f9ada5d00>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7f6f9ab46b60>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7f6f9ab46a20>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7f6f9abe7920>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7f6f9ada44a0>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7f6f9ab8ccc0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7f6f9ab8ccc0>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7f6f9abb6340>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7f6f9ab74fe0>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7f6f9ad5bb00>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7f6f9acb4d60>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7f6f9acb4d60>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7f6f9accc5e0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7f6f9accc0e0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7f6f9accc0e0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7f6f9ab8d300>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3c040>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7f6f9accc5e0>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7f6f9acf80e0>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7f6f9acf96c0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7f6f9acccae0>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7f6f9acccae0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7f6f9abe7380>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7f6f9acccfe0>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7f6f9acccfe0>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7f6f9ad5ad40>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7f6f9ad5ade0>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7f6f9accdb20>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7f6f9accdb20>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7f6f9ace0180>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7f6f9ace0180>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7f6f9ace0680>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7f6f9ace0680>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7f6f9ace0b80>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7f6f9ace0b80>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7f6f9ace1080>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7f6f9ace1580>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7f6f9ace1080>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7f6f9ace1580>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7f6f9aa5e660>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7f6f9ace23e0>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7f6f9ace23e0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7f6f9ace1da0>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7f6f9ace1da0>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7f6f9ace2ca0>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7f6f9ace2ca0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7f6f9ace3ba0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7f6f9ace3ba0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7f6f9acf80e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7f6f9ad7ec00>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7f6f9acf85e0>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7f6f9acf85e0>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7f6f9ace2700>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7f6f9ace2700>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7f6f9acf8e00>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7f6f9acf8e00>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7f6f9aa8d620>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7f6f9ada7d80>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7f6f9acf96c0>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7f6f9acf9bc0>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7f6f9ab8f240>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7f6f9acf9bc0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7f6f9ac3d1c0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7f6f9acfa0c0>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7f6f9acfa0c0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7f6f9ac3c860>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7f6f9acfaa20>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7f6f9acfaa20>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7f6f9ab19440>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7f6f9ab19440>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7f6f9adfb600>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7f6f9ab198a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7f6f9ab198a0>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7f6f9ac18680>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7f6f9ab19da0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7f6f9ab19da0>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7f6f9ac3f100>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7f6f9ac3f100>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7f6f9acb4180>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7f6f9acb4180>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7f6f9ad7ed40>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7f6f9acb47c0>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7f6f9acb47c0>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7f6f9acb4cc0>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7f6f9acb4cc0>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7f6f9acb51c0>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7f6f9acb51c0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7f6f9acb56c0>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7f6f9acb56c0>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7f6f9acb5bc0>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7f6f9acb5bc0>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7f6f9acb6e80>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7f6f9acb6e80>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7f6f9acb7380>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7f6f9acb7380>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7f6f9acb60c0>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7f6f9abe5300>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7f6f9ada6c00>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7f6f9ad7ca40>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7f6f9acb60c0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7f6f9acb65c0>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7f6f9acb65c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7f6f9acb6980>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7f6f9acb6980>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7f6f9abb4180>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7f6f9ab2f560>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7f6f9ab2f560>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7f6f9ac3f380>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7f6f9ac3f380>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7f6f9adf8540>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7f6f9ac1a340>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7f6f9ab8c860>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7f6f9ab8c0e0>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7f6f9abe42c0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7f6f9adfb420>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7f6f9ab8e7a0>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7f6f9ab744a0>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7f6f9ab8dc60>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7f6f9abb4180>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7f6f9ab8efc0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7f6f9ab8efc0>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7f6f9abb4540>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7f6f9abb4540>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7f6f9abb42c0>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7f6f9abb42c0>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7f6f9ab8e840>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7f6f9ab8e840>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7f6f9ab8eac0>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7f6f9ab8eac0>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7f6f9ab8ec00>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7f6f9ab8ec00>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7f6f9ab8da80>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7f6f9ab8da80>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7f6f9ab8c9a0>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7f6f9ab8c9a0>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <OpOverload(op='aten.item', overload='default')>: <function item at 0x7f6f9ab454e0>, <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7f6f9ac18400>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7f6f9ad7ff60>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7f6f9ad7ff60>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac1b920>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7f6f9abe72e0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7f6f9abb6b60>}
- experimental_experiment.torch_dynamo.get_decomposition_table()[source]¶
Returns the decomposition table needed to translate backward graph into onnx. It should used as follows:
import torch from torch._dynamo.backends.common import aot_autograd from experimental_experiment.torch_dynamo import get_decomposition_table aot_compiler = aot_autograd( fw_compiler=backend_debug, decompositions=get_decomposition_table() ) compiled_model = torch.compile( model, backend=aot_compiler, dynamic=dynamic, fullgraph=fullgraph, )
The value is:
<<<
import pprint from experimental_experiment.torch_dynamo import get_decomposition_table pprint.pprint(get_decomposition_table())
>>>
{<OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7f6f9ada7380>}
- 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='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfec0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebceb60>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebceca0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten.conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce660>, kernel=<OpOverload(op='aten.conv3d', overload='default')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6a20>, kernel=<OpOverload(op='aten.flipud', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc2fc0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5760>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>), <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bad440>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3a60>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60540>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>), <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6160>, kernel=<OpOverload(op='aten.pad', overload='default')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea394e0>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bafa60>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten.isclose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7420>, kernel=<OpOverload(op='aten.isclose', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38cc0>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.conv1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7a60>, kernel=<OpOverload(op='aten.conv1d', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7b00>, kernel=<OpOverload(op='aten.is_neg', overload='default')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5580>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38680>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebccae0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac2c0>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc07c0>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.greater', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baef20>, kernel=<OpOverload(op='aten.greater', 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 0x7f6ef6bad8a0>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a8e0>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a2a0>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7c40>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bacf40>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a020>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6916de0>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', 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 0x7f6eeea3b9c0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7560>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131080>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b4c0>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='aten.special_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131120>, kernel=<OpOverload(op='aten.special_softmax', overload='default')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a520>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130860>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcea20>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce980>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6f52536f20>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3ae80>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131760>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6ac1da0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4a40>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>), <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6e80>, kernel=<OpOverload(op='aten.align_as', overload='default')>), <OpOverload(op='aten.special_log_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb68e0>, kernel=<OpOverload(op='aten.special_log_softmax', overload='default')>), <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7132020>, kernel=<OpOverload(op='aten.nll_loss_nd', 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 0x7f6eeebc3880>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.log_sigmoid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7380>, kernel=<OpOverload(op='aten.log_sigmoid', overload='default')>), <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc2e80>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38c20>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc02c0>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af02c0>, kernel=<OpOverload(op='aten.special_erfinv', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af1b20>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5940>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.max_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6fc0>, kernel=<OpOverload(op='aten.max_pool2d', overload='default')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6480>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb67a0>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6f20>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>), <OpOverload(op='aten.linalg_inv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6d40>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7ec0>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60360>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4720>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5ee0>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7240>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7740>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5f80>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6020>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.subtract', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7d80>, kernel=<OpOverload(op='aten.subtract', overload='Tensor')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131940>, kernel=<OpOverload(op='aten.inverse', 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 0x7f6eeebcc220>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38b80>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>), <OpOverload(op='aten.max_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc4a0>, kernel=<OpOverload(op='aten.max_pool3d', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcca40>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39120>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60f40>, 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 0x7f6ef6bad3a0>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.special_sinc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf920>, kernel=<OpOverload(op='aten.special_sinc', overload='default')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6badee0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60a40>, kernel=<OpOverload(op='image.read_file', overload='default')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf100>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae980>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3b00>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf2e0>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bafc40>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea607c0>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac4a0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60860>, kernel=<OpOverload(op='image.decode_png', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baeb60>, kernel=<OpOverload(op='aten.special_gammaincc', 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 0x7f6eeebb4b80>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6917380>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.einsum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712cea0>, kernel=<OpOverload(op='aten.einsum', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af2520>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131b20>, kernel=<OpOverload(op='aten.smm', overload='default')>), <OpOverload(op='aten.divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130040>, kernel=<OpOverload(op='aten.divide', overload='Scalar')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71309a0>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131ee0>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130900>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.instance_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6ac1f80>, kernel=<OpOverload(op='aten.instance_norm', overload='default')>), <OpOverload(op='aten.special_erfc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb47c0>, kernel=<OpOverload(op='aten.special_erfc', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4ea0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc14e0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.scaled_dot_product_attention', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1260>, kernel=<OpOverload(op='aten.scaled_dot_product_attention', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc05e0>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc3100>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0680>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.cross_entropy_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc3b00>, kernel=<OpOverload(op='aten.cross_entropy_loss', overload='default')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af11c0>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af34c0>, kernel=<OpOverload(op='aten.take_along_dim', 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 0x7f6eeea3a7a0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af2de0>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af2660>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39580>, kernel=<OpOverload(op='aten.arcsin', 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 0x7f6eeea38220>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea396c0>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38a40>, kernel=<OpOverload(op='aten.to_sparse_bsr', 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 0x7f6eeea3aca0>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38180>, 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 0x7f6eeea38400>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_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 0x7f6eeea3ab60>, 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 0x7f6eeea3ac00>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a480>, kernel=<OpOverload(op='aten.orgqr', 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 0x7f6eeea61760>, 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 0x7f6eeea61940>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39760>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea398a0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>), <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39c60>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebceac0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5d00>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0040>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0a40>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6700>, kernel=<OpOverload(op='aten._pad_circular', 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 0x7f6eeebb5a80>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5da0>, 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 0x7f6eeebcccc0>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60c20>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>), <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60cc0>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5c60>, kernel=<OpOverload(op='aten.gather_backward', 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 0x7f6eeebcf600>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea602c0>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd4e0>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef8f3ec00>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bad120>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf380>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>), <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baccc0>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4220>, kernel=<OpOverload(op='aten._backward', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf4c0>, kernel=<OpOverload(op='aten.special_i0', overload='default')>), <OpOverload(op='aten.det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bad800>, kernel=<OpOverload(op='aten.det', overload='default')>), <OpOverload(op='aten.multiply', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea385e0>, kernel=<OpOverload(op='aten.multiply', overload='Tensor')>), <OpOverload(op='aten.special_expm1', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38860>, kernel=<OpOverload(op='aten.special_expm1', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712ff60>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4180>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7f60>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bade40>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bad1c0>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6badbc0>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0f40>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1440>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71322a0>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4d60>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1a80>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc2f20>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf380>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten.embedding_bag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebccf40>, kernel=<OpOverload(op='aten.embedding_bag', overload='default')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130720>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bafd80>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten._debug_has_internal_overlap', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb58a0>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0e00>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce7a0>, kernel=<OpOverload(op='aten.gru_cell', 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 0x7f6ef6af32e0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc3060>, kernel=<OpOverload(op='aten.rrelu', overload='default')>), <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc3ec0>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>), <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0ae0>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>), <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4ae0>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>), <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3be20>, kernel=<OpOverload(op='_test.get_first', 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 0x7f6eeea3b2e0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38ea0>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af2480>, 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 0x7f6ef6af0040>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bacc20>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af04a0>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf240>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3ce0>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6ac0>, kernel=<OpOverload(op='aten.absolute', overload='default')>), <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb51c0>, 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 0x7f6eeea60680>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten.embedding_bag', overload='padding_idx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4c20>, kernel=<OpOverload(op='aten.embedding_bag', overload='padding_idx')>), <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 0x7f6eeebb5bc0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4680>, kernel=<OpOverload(op='aten._convolution_double_backward', 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 0x7f6eeebcf740>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb59e0>, kernel=<OpOverload(op='aten.divide', overload='Tensor')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb71a0>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>), <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7600>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60220>, kernel=<OpOverload(op='c10d_functional.all_reduce', 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 0x7f6ef6baf4c0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7e20>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131f80>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7ce0>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>), <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4400>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc4c20>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7f6eeebb4040>, 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 0x7f6eeebcc040>, 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 0x7f6eeebcc720>, kernel=<OpOverload(op='aten._sparse_mm', 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 0x7f6eeebb49a0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5800>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc860>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd120>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>), <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcce00>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b060>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf1a0>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af1080>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac9a0>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39800>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7920>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>), <OpOverload(op='aten.type_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6badb20>, 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 0x7f6ef6bae5c0>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae520>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130680>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae200>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bafec0>, kernel=<OpOverload(op='aten.greater_equal', overload='Tensor')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac720>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf880>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baefc0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae0c0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bace00>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bad580>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131260>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea387c0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6914900>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae480>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc4220>, kernel=<OpOverload(op='aten.special_round', 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 0x7f6ef7130ea0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bacb80>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130180>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef69142c0>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>), <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712e520>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.concat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712e160>, kernel=<OpOverload(op='aten.concat', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc360>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b740>, kernel=<OpOverload(op='image.decode_webp', overload='default')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc09a0>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4360>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712e8e0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71300e0>, kernel=<OpOverload(op='aten.cummaxmin_backward', 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 0x7f6ef7132200>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>), <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b560>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten.log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a3e0>, kernel=<OpOverload(op='aten.log_softmax', overload='int')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130ae0>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>), <OpOverload(op='aten.logdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71316c0>, kernel=<OpOverload(op='aten.logdet', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bad300>, kernel=<OpOverload(op='aten.argsort', overload='default')>), <OpOverload(op='aten.subtract', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130360>, kernel=<OpOverload(op='aten.subtract', overload='Scalar')>), <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130e00>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>), <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6ac3f60>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>), <OpOverload(op='aten.linalg_det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc7880>, kernel=<OpOverload(op='aten.linalg_det', overload='default')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131620>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130fe0>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b100>, kernel=<OpOverload(op='image.encode_png', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb65c0>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.less_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6ca0>, kernel=<OpOverload(op='aten.less_equal', overload='Tensor')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb74c0>, kernel=<OpOverload(op='aten.data', overload='default')>), <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb62a0>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6840>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten.cross', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4fe0>, kernel=<OpOverload(op='aten.cross', overload='default')>), <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1b20>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6980>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1760>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131bc0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0220>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baee80>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1580>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten.avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5120>, kernel=<OpOverload(op='aten.avg_pool1d', 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 0x7f6eeebc3380>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='aten.atleast_2d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a5c0>, kernel=<OpOverload(op='aten.atleast_2d', overload='Sequence')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131580>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5440>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71313a0>, kernel=<OpOverload(op='aten.trapz', overload='x')>), <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc3d80>, kernel=<OpOverload(op='aten.bilinear', overload='default')>), <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1940>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>), <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af09a0>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>), <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3e20>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac860>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af2020>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6200>, kernel=<OpOverload(op='aten.special_polygamma', 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 0x7f6ef6af16c0>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf060>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc6340>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3aac0>, kernel=<OpOverload(op='aten.special_gammaln', 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 0x7f6eeebb7060>, kernel=<OpOverload(op='aten.max_pool1d_with_indices', overload='default')>), <OpOverload(op='aten.concatenate', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6520>, kernel=<OpOverload(op='aten.concatenate', overload='default')>), <OpOverload(op='aten.true_divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb63e0>, kernel=<OpOverload(op='aten.true_divide', overload='Scalar')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71305e0>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6c00>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb54e0>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea600e0>, kernel=<OpOverload(op='image.decode_gif', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3ade0>, kernel=<OpOverload(op='aten.chalf', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a340>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6660>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7100>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc3e20>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb77e0>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4f40>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb76a0>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb45e0>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb44a0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4540>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>), <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7ba0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>), <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3bc40>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc31a0>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc180>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bacea0>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc900>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcde40>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>), <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd080>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>), <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6badd00>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea384a0>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac040>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39440>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38900>, kernel=<OpOverload(op='aten.one_hot', 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 0x7f6eeebb4e00>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcff60>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38f40>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea380e0>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38360>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.layer_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6b60>, kernel=<OpOverload(op='aten.layer_norm', overload='default')>), <OpOverload(op='aten.tile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bafb00>, kernel=<OpOverload(op='aten.tile', overload='default')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3aa20>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a980>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a840>, kernel=<OpOverload(op='aten.fix', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131c60>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39e40>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39da0>, 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 0x7f6eeea39ee0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a0c0>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', 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 0x7f6eeea399e0>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39bc0>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf7e0>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3ba60>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', 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 0x7f6eeea616c0>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5e40>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd620>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce2a0>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce520>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>), <OpOverload(op='aten.group_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce340>, kernel=<OpOverload(op='aten.group_norm', overload='default')>), <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce5c0>, kernel=<OpOverload(op='aten.pinverse', 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 0x7f6eeebb5620>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcdbc0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61120>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='aten.atleast_3d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb72e0>, kernel=<OpOverload(op='aten.atleast_3d', overload='Sequence')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd3a0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfce0>, kernel=<OpOverload(op='aten.lu_solve', 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 0x7f6eeebcf920>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61300>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712e980>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf9c0>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af0720>, kernel=<OpOverload(op='aten.max_pool1d', overload='default')>), <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfb00>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>), <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcee80>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0900>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1620>, kernel=<OpOverload(op='aten.max', overload='names_dim')>), <OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf2e0>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39080>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38d60>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf1a0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebced40>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.atleast_1d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39620>, kernel=<OpOverload(op='aten.atleast_1d', overload='Sequence')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae840>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38fe0>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', 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 0x7f6eeea38ae0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130d60>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0720>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcefc0>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce700>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>), <OpOverload(op='aten.grid_sampler', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf7e0>, kernel=<OpOverload(op='aten.grid_sampler', overload='default')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfc40>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac900>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten.special_erf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae2a0>, kernel=<OpOverload(op='aten.special_erf', overload='default')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6917ec0>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef712dbc0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7131d00>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3ba0>, kernel=<OpOverload(op='aten.is_inference', overload='default')>), <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b6a0>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b1a0>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7132480>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130f40>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcda80>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6ac2020>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef1d3bb00>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc6e80>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3ad40>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc7ce0>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bada80>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60fe0>, kernel=<OpOverload(op='c10d_functional.broadcast', overload='default')>), <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc7ec0>, kernel=<OpOverload(op='aten.msort', 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 0x7f6ef6af3c40>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <OpOverload(op='aten.softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcdc60>, kernel=<OpOverload(op='aten.softmax', overload='int')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3bd80>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1bc0>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1120>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3bb00>, kernel=<OpOverload(op='_test.cat', overload='default')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6badc60>, 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 0x7f6ef6bac360>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6badf80>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc7a60>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5300>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb56c0>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef7130540>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae700>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf6a0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <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 0x7f6eeebb4900>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af07c0>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39260>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39300>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc9a0>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0b80>, kernel=<OpOverload(op='aten.isreal', 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 0x7f6ef6af2b60>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3bba0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb4860>, kernel=<OpOverload(op='aten.kron', overload='default')>), <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc1300>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebccb80>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0c20>, kernel=<OpOverload(op='aten.diagflat', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0540>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <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 0x7f6eeea3b380>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', 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 0x7f6eeebc3a60>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>), <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3380>, kernel=<OpOverload(op='aten.to_dense', overload='default')>), <OpOverload(op='aten.less', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebccfe0>, kernel=<OpOverload(op='aten.less', overload='Tensor')>), <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3ec0>, kernel=<OpOverload(op='aten.cov', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af2ca0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc680>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae8e0>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af1940>, kernel=<OpOverload(op='aten.argwhere', overload='default')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef1f39760>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea391c0>, kernel=<OpOverload(op='aten.is_conj', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb40e0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61a80>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af3100>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a660>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.is_nonzero', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38040>, kernel=<OpOverload(op='aten.is_nonzero', overload='default')>), <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd6c0>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61440>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.sym_size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a700>, kernel=<OpOverload(op='aten.sym_size', overload='int')>), <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6af0f40>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebccd60>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten._fused_rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc0e0>, kernel=<OpOverload(op='aten._fused_rms_norm', overload='default')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baf560>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc7c0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bafe20>, 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 0x7f6ef6bac7c0>, kernel=<OpOverload(op='aten.true_divide', overload='Tensor')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc5e0>, 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 0x7f6ef6baff60>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7f6f9abe7d80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7f6f9abe7d80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <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 0x7f6f9aa21940>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7f6f9aa22f20>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7f6f9aa22f20>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7f6f9aa231a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7f6f9aa231a0>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39940>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3a160>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7f6f9aa23c40>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39d00>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7f6f9aa21620>, <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce840>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7f6f9aa23f60>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce8e0>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7f6f9aa5c220>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce200>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7f6f9aa5c4a0>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce480>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7f6f9aa5cd60>, <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd9e0>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7f6f9aa5cfe0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcdf80>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7f6f9aa5d260>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce020>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7f6f9aa5d4e0>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7f6f9aa5db20>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd300>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7f6f9aa5dbc0>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf880>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd760>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7f6f9aa5de40>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfba0>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7f6f9aa5ccc0>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf240>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7f6f9aa5dee0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf560>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7f6f9aa5e160>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcede0>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7f6f9aa5e3e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf060>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf100>, 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 0x7f6eeebcec00>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7f6f9aa5ee80>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7f6f9aa5ee80>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7f6f9aa5f240>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7f6f9aa5f240>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7f6f9aa8c220>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7f6f9aa8c4a0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7f6f9aa8c4a0>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7f6f9aa8cb80>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7f6f9aa8cb80>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7f6f9aa8d800>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7f6f9aa8cfe0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7f6f9aa8cfe0>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7f6f9aa8d800>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7f6f9aa8d620>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7f6f9aa8dbc0>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7f6f9aa8dbc0>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7f6f9aa8e980>, <OpOverload(op='aten.selu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6baeca0>, kernel=<OpOverload(op='aten.selu', overload='default')>), <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7f6f9aa8eca0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7f6f9aa8e2a0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7f6f9aa8e2a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7f6f9aa8e980>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7f6f9aa8eca0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bac0e0>, kernel=<OpOverload(op='aten.margin_ranking_loss', overload='default')>), <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bae340>, 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 0x7f6ef712e840>, kernel=<OpOverload(op='aten.nll_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7f6f9aa8fce0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7f6f9aaac0e0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7f6f9aaac0e0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7f6f9aaac400>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7f6f9aaac400>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7f6f9aa8e8e0>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7f6f9aa8e8e0>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7f6f9aaacc20>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7f6f9aaacc20>, <OpOverload(op='aten.prelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef6bc7ba0>, 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 0x7f6ef7131300>, kernel=<OpOverload(op='aten.relu6', overload='default')>), <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7f6f9aaadd00>, <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7f6f9aaad760>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7f6f9aaadd00>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb6de0>, 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 0x7f6eeebc18a0>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7f6f9aaae020>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7f6f9aaae020>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7f6f9aaae2a0>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7f6f9aaae2a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7f6f9aaacd60>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7f6f9aaae0c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7f6f9aaae340>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7f6f9aaae480>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7f6f9aaae5c0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7f6f9aaaeca0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7f6f9aaaeca0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7f6f9aaaf060>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7f6f9aaaf060>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7f6f9aaaf420>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7f6f9aaaf420>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7f6f9aaaf9c0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7f6f9aaaf9c0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7f6f9aaafec0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7f6f9aaafec0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7f6f9aac8400>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7f6f9aac8400>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7f6f9aac87c0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7f6f9aac87c0>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7f6f9aac8b80>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7f6f9aac8b80>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7f6f9aaad940>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7f6f9aaad940>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea382c0>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7f6f9aac9080>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7f6f9aac9440>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7f6f9aac9440>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7f6f9aac9a80>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7f6f9aac9a80>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7f6f9ab1afc0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7f6f9ab1afc0>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7f6f9ab2c2c0>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7f6f9ab2c2c0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7f6f9ab2cb80>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7f6f9ab2cb80>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7f6f9ab2d440>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7f6f9ab2cfe0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7f6f9ab2cfe0>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7f6f9ab2d440>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7f6f9ab2d3a0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7f6f9ab2d3a0>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7f6f9ab2d620>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7f6f9ab2d620>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7f6f9ab2da80>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7f6f9ab2da80>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb7880>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7f6f9ab447c0>, <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc400>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7f6f9ab44b80>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7f6f9ab44b80>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7f6f9ab745e0>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb79c0>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.any', overload='default')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5b20>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb53a0>, 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 0x7f6ef6baf9c0>, 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 0x7f6eeebb60c0>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5260>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5080>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb42c0>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcc540>, kernel=<OpOverload(op='aten.var_mean', overload='default')>), <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd1c0>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebccc20>, 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 0x7f6eeebccea0>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7f6f9ab47ec0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7f6f9ab47ec0>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7f6f9ab75a80>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7f6f9ab74fe0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7f6f9ab75b20>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7f6f9ab751c0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7f6f9ab76200>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7f6f9ab75800>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7f6f9ab75800>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7f6f9ab8c540>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7f6f9ab76200>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7f6f9ab753a0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7f6f9ab76340>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7f6f9ab74400>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7f6f9ab74400>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7f6f9ab76700>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7f6f9ab76700>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7f6f9ab76d40>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7f6f9ab76d40>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebc0ea0>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7f6f9ab76fc0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7f6f9ab76fc0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7f6f9ab772e0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7f6f9ab772e0>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7f6f9ab77920>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7f6f9ab77920>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef71311c0>, 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 0x7f6ef71302c0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7f6f9ab77240>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7f6f9ab8d080>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7f6f9ab8c860>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7f6f9ab8c0e0>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7f6f9ab8cb80>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7f6f9ab8cb80>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7f6f9ab8cf40>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7f6f9ab8cf40>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7f6f9ab8d3a0>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7f6f9ab8d260>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7f6f9ab8d440>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7f6f9ab8d8a0>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7f6f9ab8d8a0>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6ef8f3e7a0>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea393a0>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7f6f9ab8db20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7f6f9ab8f240>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39f80>, 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 0x7f6eeea3a200>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7f6f9abb47c0>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7f6f9abb49a0>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7f6f9abb49a0>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd8a0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcdee0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7f6f9abb5e40>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7f6f9abb5e40>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7f6f9abb5300>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7f6f9abb5300>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7f6f9abb5800>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7f6f9abb5800>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7f6f9abb5760>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7f6f9abb5a80>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7f6f9abb5a80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7f6f9abb5760>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7f6f9abb6020>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7f6f9abb6020>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7f6f9abb63e0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7f6f9abb63e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7f6f9abb67a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7f6f9abb67a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7f6f9abb6b60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7f6f9abb6c00>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7f6f9abb7560>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7f6f9abb7560>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7f6f9abb7a60>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7f6f9abb7a60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7f6f9abb5f80>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7f6f9abb5f80>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7f6f9abb7f60>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7f6f9abb7f60>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7f6f9abe59e0>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7f6f9abe42c0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7f6f9abe51c0>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7f6f9abb7c40>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7f6f9abe40e0>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7f6f9abe5440>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7f6f9abe4360>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7f6f9abe44a0>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7f6f9abe44a0>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7f6f9abe56c0>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7f6f9abe45e0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7f6f9abe56c0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7f6f9abe4720>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7f6f9abe6480>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7f6f9abe4860>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7f6f9abe49a0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7f6f9abe5940>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7f6f9abe4ae0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7f6f9abe5940>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7f6f9abe4c20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7f6f9aa20860>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7f6f9abe4d60>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7f6f9abe5bc0>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7f6f9abe4ea0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7f6f9abe4ea0>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7f6f9abe6200>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7f6f9abe4fe0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7f6f9abe4fe0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7f6f9abe5120>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7f6f9abe5260>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7f6f9abe5260>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7f6f9abe5c60>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7f6f9abe53a0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7f6f9abe53a0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7f6f9abe54e0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7f6f9abe54e0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7f6f9abe5b20>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7f6f9abe5620>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7f6f9abe5760>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7f6f9abe5760>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7f6f9abe5f80>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7f6f9abe5b20>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7f6f9abe58a0>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7f6f9abe58a0>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7f6f9abe59e0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7f6f9abe6700>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7f6f9abe4cc0>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7f6f9abe4cc0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7f6f9abe6980>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7f6f9abe4a40>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7f6f9abe47c0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7f6f9abe6c00>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7f6f9abe4540>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7f6f9abe4220>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7f6f9abe5da0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7f6f9abe6e80>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7f6f9abe5ee0>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7f6f9abe7100>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7f6f9abe6020>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7f6f9abe6020>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7f6f9abe6160>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7f6f9abe62a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7f6f9abe62a0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7f6f9abe7ec0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7f6f9abe63e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7f6f9abe63e0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7f6f9abe6520>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7f6f9abe6660>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7f6f9abe67a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7f6f9abe67a0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7f6f9abe72e0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7f6f9abe68e0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7f6f9abe68e0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7f6f9abe6a20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7f6f9abe6b60>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7f6f9aa218a0>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7f6f9abe6ca0>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7f6f9abe71a0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7f6f9abe6de0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7f6f9abe71a0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7f6f9abe6f20>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7f6f9abe7060>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7f6f9abe7c40>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7f6f9abe4040>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7f6f9abe4040>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7f6f9abe4680>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7f6f9abe4680>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7f6f9abe4b80>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7f6f9abe6340>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7f6f9abe5080>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7f6f9abe5580>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7f6f9abe5580>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7f6f9abe5e40>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7f6f9abe5a80>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7f6f9abe7420>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7f6f9abe4900>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7f6f9abe7560>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7f6f9abe7560>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7f6f9abe76a0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7f6f9abe77e0>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7f6f9abe7920>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7f6f9abe7a60>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7f6f9abe7ba0>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7f6f9abe5d00>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7f6f9abe7ce0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7f6f9abe7e20>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7f6f9abe7f60>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7f6f9abe7600>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7f6f9aa200e0>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7f6f9abe7600>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7f6f9aa20220>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7f6f9aa20360>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7f6f9aa204a0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7f6f9abe7880>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7f6f9abe7b00>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7f6f9aa205e0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7f6f9aa205e0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7f6f9abe7b00>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7f6f9aa20720>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7f6f9abe6d40>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7f6f9abe6d40>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7f6f9abe6ac0>, <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 0x7f6f9abe6ac0>, <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 0x7f6f9aa20900>, <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 0x7f6f9aa20900>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7f6f9aa202c0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7f6f9aa202c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7f6f9aa20c20>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7f6f9aa20c20>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7f6f9aa20f40>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7f6f9aa20f40>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <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 0x7f6f9aa21260>, <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 0x7f6f9aa21260>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7f6f9aa21580>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7f6f9aa21580>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7f6f9aa218a0>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7f6f9abe4e00>, <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 0x7f6f9abe4e00>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7f6f9ac1a700>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7f6f9ac19b20>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7f6f9ac19b20>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7f6f9ac1a2a0>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7f6f9ac1a2a0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac185e0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac185e0>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7f6f9ac1b9c0>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7f6f9ac1b380>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7f6f9ac1b380>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7f6f9ac1b9c0>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7f6f9ac1b7e0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7f6f9ac1b740>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7f6f9ac1b740>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7f6f9ac1bba0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7f6f9ac1bf60>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7f6f9ac1bf60>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e0c0>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7f6f9ac3ccc0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7f6f9ac3ca40>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7f6f9ac3ca40>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e020>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7f6f9ac3d3a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7f6f9ac3d3a0>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7f6f9ac3d1c0>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d440>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d580>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d6c0>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d800>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e2a0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d940>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d940>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3da80>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dbc0>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dd00>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3de40>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d080>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3df80>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3df80>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e0c0>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dda0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dda0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3db20>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d8a0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d620>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d300>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3ce00>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3ce00>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e340>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e480>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39b20>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7f6f9acb7ba0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7f6f9acb7ba0>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7f6f9accd620>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7f6f9accd620>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7f6f9acce020>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7f6f9acce020>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7f6f9accea20>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7f6f9accdbc0>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7f6f9accdbc0>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfa60>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7f6f9acced40>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7f6f9acced40>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7f6f9accf240>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7f6f9accf240>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7f6f9accfc40>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7f6f9accfc40>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7f6f9ace1620>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7f6f9ace1620>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7f6f9ace31a0>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7f6f9ace31a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7f6f9ace36a0>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7f6f9ace36a0>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7f6f9acf8ae0>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7f6f9acf8ae0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7f6f9add3d80>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7f6f9ab1a2a0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7f6f9accea20>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7f6f9ad5a340>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7f6f9ad5a340>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7f6f9ad5ae80>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7f6f9ad5ae80>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7f6f9ad5a700>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7f6f9ad5a700>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7f6f9ad5a2a0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7f6f9ad5a2a0>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7f6f9ad5b2e0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7f6f9ad5b2e0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7f6f9ad5b380>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7f6f9ad5b380>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7f6f9ad5b740>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7f6f9ad5bc40>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7f6f9ad5bc40>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7f6f9ad5ba60>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7f6f9ad5bb00>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7f6f9ad5be20>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7f6f9ad5be20>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7f6f9ad7c220>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7f6f9ad7c220>, <OpOverload(op='aten.mish_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf6a0>, kernel=<OpOverload(op='aten.mish_backward', overload='default')>), <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7f6f9ad7ca40>, <OpOverload(op='aten.silu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcef20>, kernel=<OpOverload(op='aten.silu_backward', overload='default')>), <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7f6f9ad7cae0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7f6f9ad7ce00>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7f6f9ad7cea0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7f6f9ad7d3a0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7f6f9ad7d3a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7f6f9ad7d440>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7f6f9ad7d440>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7f6f9ad7f7e0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7f6f9ad7dd00>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7f6f9ad7f7e0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7f6f9ad7d940>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7f6f9ad7d940>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7f6f9ad7ccc0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7f6f9ad7dee0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7f6f9ad7dc60>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7f6f9ad7dc60>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7f6f9ad7e0c0>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7f6f9ada5580>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7f6f9ad7e2a0>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7f6f9ada49a0>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7f6f9ad7e520>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7f6f9ad7e520>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7f6f9ad7e8e0>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7f6f9ad7e8e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7f6f9ad7ec00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7f6f9ad7f1a0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7f6f9ad7f1a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7f6f9ad7f240>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7f6f9ad7f240>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7f6f9ad7f880>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7f6f9ad7f880>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7f6f9ad7de40>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7f6f9ad7de40>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7f6f9ada5260>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7f6f9ad7f600>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7f6f9ad7f600>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7f6f9ad7fb00>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7f6f9ad7fb00>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7f6f9ad7ed40>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7f6f9ada4220>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7f6f9ada4220>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7f6f9ada44a0>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7f6f9ada45e0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7f6f9ada45e0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7f6f9ada4c20>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7f6f9ada4c20>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7f6f9ada4f40>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7f6f9ada4f40>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7f6f9ada5300>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7f6f9ada5300>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7f6f9ada5580>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7f6f9ada5940>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7f6f9ada5940>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7f6f9ada5d00>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7f6f9ada5f80>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7f6f9ada5f80>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7f6f9ada6200>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7f6f9ada6200>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7f6f9ada67a0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7f6f9ada67a0>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7f6f9ada6840>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7f6f9ada6840>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7f6f9ada6980>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7f6f9ada6ac0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7f6f9ada5ee0>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7f6f9ada5c60>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7f6f9ada6ca0>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7f6f9ada6ca0>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7f6f9ada7420>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7f6f9ada6160>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7f6f9ada6160>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7f6f9ada7380>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7f6f9ada7560>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7f6f9add1800>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7f6f9add16c0>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7f6f9add0180>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7f6f9ada7d80>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7f6f9add0220>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7f6f9add04a0>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7f6f9add0360>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce0c0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7f6f9add0720>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7f6f9add07c0>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7f6f9add0900>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7f6f9ada7ec0>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7f6f9ada7ec0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7f6f9add0400>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7f6f9add11c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7f6f9add11c0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7f6f9add1620>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7f6f9add2fc0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7f6f9add0400>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7f6f9add1f80>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7f6f9add1f80>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7f6f9add2660>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7f6f9add2660>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7f6f9add31a0>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7f6f9add31a0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7f6f9add2ac0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7f6f9add2ac0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7f6f9add2de0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7f6f9add2de0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7f6f9add3060>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7f6f9add3060>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7f6f9add2e80>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd800>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7f6f9add39c0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7f6f9add39c0>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7f6f9adf8540>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7f6f9add3ba0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7f6f9adf8d60>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7f6f9adf9080>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7f6f9adf8fe0>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7f6f9adf8540>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7f6f9adf8d60>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7f6f9adf93a0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7f6f9adf8fe0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7f6f9adf9080>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7f6f9adf96c0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7f6f9adf93a0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7f6f9adf96c0>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7f6f9add1760>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7f6f9adf9a80>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7f6f9adfa020>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7f6f9add1760>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7f6f9adf9c60>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7f6f9adf9e40>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7f6f9adfa480>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7f6f9adfa660>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7f6f9adfa980>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7f6f9adfab60>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7f6f9adfbba0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7f6f9adfb600>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7f6f9adfb420>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7f6f9adfb920>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7f6f9adfb920>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7f6f9adfb420>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7f6f9adfbba0>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7f6f9adfa8e0>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7f6f9adfbf60>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7f6f9adfa8e0>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7f6f9ac18180>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7f6f9ac18540>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7f6f9adf9f80>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7f6f9adf9f80>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7f6f9ac1bba0>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7f6f9ac191c0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7f6f9ac191c0>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7f6f9ac1bc40>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7f6f9ac19580>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7f6f9ac19580>, <OpOverload(op='aten.matmul', overload='default')>: <function matmul at 0x7f6f9ac19ee0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7f6f9ac198a0>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7f6f9ac198a0>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7f6f9ab1b420>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7f6f9ab1b420>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7f6f9ab1b880>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7f6f9ab1b880>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7f6f9ab1bce0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7f6f9ab1bce0>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7f6f9ab46a20>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7f6f9ad5ba60>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7f6f9ab2e340>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7f6f9ab2e340>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7f6f9ab2e7a0>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7f6f9ab2e7a0>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7f6f9ad5b740>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea39a80>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7f6f9ab2f4c0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7f6f9ab2f4c0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7f6f9ab447c0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7f6f9adfaf20>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e2a0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7f6f9b19afc0>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7f6f9b19afc0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7f6f9ac3c900>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7f6f9ac3c900>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7f6f9ab77a60>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7f6f9ab8d300>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7f6f9ab744a0>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38e00>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7f6f9aaad760>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7f6f9add3d80>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.where', overload='ScalarOther')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea389a0>, kernel=<OpOverload(op='aten.where', overload='ScalarOther')>), <OpOverload(op='aten.where', overload='ScalarSelf')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38720>, kernel=<OpOverload(op='aten.where', overload='ScalarSelf')>), <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfd80>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.where', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcfe20>, kernel=<OpOverload(op='aten.where', overload='Scalar')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e020>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7f6f9ab1a2a0>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7f6f9adfad40>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7f6f9ab465c0>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7f6f9ac3c860>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7f6f9ab465c0>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce3e0>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7f6f9abe5800>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd940>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.var', overload='out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcdda0>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcdb20>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebce160>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7f6f9ab46b60>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7f6f9ada5d00>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7f6f9ab46b60>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7f6f9ab46a20>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7f6f9abe7920>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7f6f9ada44a0>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7f6f9ab8ccc0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7f6f9ab8ccc0>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7f6f9abb6340>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7f6f9ab74fe0>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea38540>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7f6f9ad5bb00>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7f6f9acb4d60>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7f6f9acb4d60>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7f6f9accc5e0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7f6f9accc0e0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7f6f9accc0e0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7f6f9ab8d300>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3c040>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7f6f9accc5e0>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7f6f9acf80e0>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7f6f9acf96c0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7f6f9acccae0>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7f6f9acccae0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7f6f9abe7380>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7f6f9acccfe0>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7f6f9acccfe0>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7f6f9ad5ad40>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7f6f9ad5ade0>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7f6f9accdb20>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7f6f9accdb20>, <OpOverload(op='aten.isfinite', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcf420>, kernel=<OpOverload(op='aten.isfinite', overload='default')>), <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7f6f9ace0180>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7f6f9ace0180>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7f6f9ace0680>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7f6f9ace0680>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7f6f9ace0b80>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7f6f9ace0b80>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7f6f9ace1080>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7f6f9ace1580>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7f6f9ace1080>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7f6f9ace1580>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7f6f9aa5e660>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7f6f9ace23e0>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7f6f9ace23e0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7f6f9ace1da0>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7f6f9ace1da0>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7f6f9ace2ca0>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7f6f9ace2ca0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7f6f9ace3ba0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7f6f9ace3ba0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7f6f9acf80e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7f6f9ad7ec00>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7f6f9acf85e0>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7f6f9acf85e0>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7f6f9ace2700>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7f6f9ace2700>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7f6f9acf8e00>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7f6f9acf8e00>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7f6f9aa8d620>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7f6f9ada7d80>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7f6f9acf96c0>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7f6f9acf9bc0>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7f6f9ab8f240>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7f6f9acf9bc0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7f6f9ac3d1c0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7f6f9acfa0c0>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7f6f9acfa0c0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7f6f9ac3c860>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7f6f9acfaa20>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7f6f9acfaa20>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7f6f9ab19440>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7f6f9ab19440>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7f6f9adfb600>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7f6f9ab198a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7f6f9ab198a0>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7f6f9ac18680>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7f6f9ab19da0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7f6f9ab19da0>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7f6f9ac3f100>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7f6f9ac3f100>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7f6f9acb4180>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7f6f9acb4180>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7f6f9ad7ed40>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7f6f9acb47c0>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7f6f9acb47c0>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7f6f9acb4cc0>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7f6f9acb4cc0>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7f6f9acb51c0>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7f6f9acb51c0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7f6f9acb56c0>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7f6f9acb56c0>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7f6f9acb5bc0>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7f6f9acb5bc0>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7f6f9acb6e80>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7f6f9acb6e80>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7f6f9acb7380>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7f6f9acb7380>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7f6f9acb60c0>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7f6f9abe5300>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7f6f9ada6c00>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7f6f9ad7ca40>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7f6f9acb60c0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7f6f9acb65c0>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7f6f9acb65c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7f6f9acb6980>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7f6f9acb6980>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7f6f9abb4180>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7f6f9ab2f560>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7f6f9ab2f560>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7f6f9ac3f380>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7f6f9ac3f380>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd440>, 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 0x7f6eeebcdd00>, kernel=<OpOverload(op='aten.conv2d', overload='default')>), <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7f6f9adf8540>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7f6f9ac1a340>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7f6f9ab8c860>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7f6f9ab8c0e0>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7f6f9abe42c0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7f6f9adfb420>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7f6f9ab8e7a0>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7f6f9ab744a0>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7f6f9ab8dc60>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7f6f9abb4180>, <OpOverload(op='aten.slice', overload='Tensor')>: <function slice_forward at 0x7f6f9ad7f920>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7f6f9ab8efc0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7f6f9ab8efc0>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7f6f9abb4540>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7f6f9abb4540>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7f6f9abb42c0>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7f6f9abb42c0>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7f6f9ab8e840>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7f6f9ab8e840>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7f6f9ab8eac0>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7f6f9ab8eac0>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7f6f9ab8ec00>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7f6f9ab8ec00>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7f6f9ab8da80>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7f6f9ab8da80>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7f6f9ab8c9a0>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7f6f9ab8c9a0>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebcd260>, 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 0x7f6eeebcd580>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7f6f9ac18400>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7f6f9ad7ff60>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7f6f9ad7ff60>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac1b920>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7f6f9abe72e0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7f6f9abb6b60>, <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea605e0>, 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 0x7f6eeea60400>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea604a0>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3bf60>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60b80>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60ae0>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', 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 0x7f6eeea3b920>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60180>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60d60>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b240>, kernel=<OpOverload(op='quantized.conv3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3afc0>, 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 0x7f6eeea613a0>, 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 0x7f6eeea61bc0>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea618a0>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea619e0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60720>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b420>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60e00>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b600>, kernel=<OpOverload(op='quantized.conv_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 0x7f6eeea3af20>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3bce0>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b7e0>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60ea0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61800>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61620>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61080>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60040>, kernel=<OpOverload(op='quantized.conv2d_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 0x7f6eeea61260>, 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 0x7f6eeea611c0>, kernel=<OpOverload(op='quantized.conv3d_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 0x7f6eeea614e0>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61580>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea60900>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea609a0>, kernel=<OpOverload(op='quantized.conv3d_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 0x7f6eeea61b20>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', 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 0x7f6eeea3bec0>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea3b880>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeea61c60>, 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='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440cc0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437920>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437a60>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42dc60>, kernel=<OpOverload(op='aten.flipud', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425260>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42de40>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>), <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e480>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83be20>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455300>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4422a0>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ec00>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441a80>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427ce0>, kernel=<OpOverload(op='aten.is_neg', overload='default')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c040>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441440>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4358a0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434860>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425bc0>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f240>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4436a0>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443060>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427920>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42eb60>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442de0>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83ae80>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', 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 0x7f6eec4547c0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c720>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a660>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4542c0>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4432e0>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839c60>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4377e0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437740>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839080>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443c40>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839bc0>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eedb29ee0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4268e0>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>), <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426d40>, kernel=<OpOverload(op='aten.align_as', overload='default')>), <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a5c0>, kernel=<OpOverload(op='aten.nll_loss_nd', 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 0x7f6eec425c60>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424ae0>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4419e0>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4256c0>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b7e0>, kernel=<OpOverload(op='aten.special_erfinv', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83bf60>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d9e0>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42db20>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e160>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e340>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>), <OpOverload(op='aten.linalg_inv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e2a0>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ce00>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455120>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d080>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426fc0>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d8a0>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c400>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c680>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c5e0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a3e0>, kernel=<OpOverload(op='aten.inverse', 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 0x7f6eec434ea0>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441940>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435800>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441ee0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455d00>, 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 0x7f6eec42fd80>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f380>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455800>, kernel=<OpOverload(op='image.read_file', overload='default')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f4c0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f920>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b600>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42fce0>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42fb00>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455580>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e520>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455620>, kernel=<OpOverload(op='image.decode_png', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ea20>, kernel=<OpOverload(op='aten.special_gammaincc', 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 0x7f6eec426c00>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83ade0>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b6a0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a840>, kernel=<OpOverload(op='aten.smm', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83ac00>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839ee0>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a020>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4271a0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4260c0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424b80>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424fe0>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4254e0>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83ba60>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b880>, kernel=<OpOverload(op='aten.take_along_dim', 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 0x7f6eec443560>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424360>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424720>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442340>, kernel=<OpOverload(op='aten.arcsin', 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 0x7f6eec440fe0>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442480>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441800>, kernel=<OpOverload(op='aten.to_sparse_bsr', 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 0x7f6eec443a60>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440f40>, 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 0x7f6eec4411c0>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_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 0x7f6eec443920>, 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 0x7f6eec4439c0>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443240>, kernel=<OpOverload(op='aten.orgqr', 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 0x7f6eec456520>, 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 0x7f6eec456700>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442520>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442660>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>), <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442a20>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437880>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42cfe0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424c20>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424a40>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426e80>, kernel=<OpOverload(op='aten._pad_circular', 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 0x7f6eec42d260>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ca40>, 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 0x7f6eec435a80>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4559e0>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>), <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455a80>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d300>, kernel=<OpOverload(op='aten.gather_backward', 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 0x7f6eec440400>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455080>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4362a0>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4249a0>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4347c0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4342c0>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>), <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ee80>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427420>, kernel=<OpOverload(op='aten._backward', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4402c0>, kernel=<OpOverload(op='aten.special_i0', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b4c0>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4274c0>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42cd60>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.nested_to_padded_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434ae0>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434a40>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f7e0>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426480>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425f80>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a700>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4265c0>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424f40>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425800>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440180>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a0c0>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434900>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten._debug_has_internal_overlap', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42df80>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4259e0>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437560>, kernel=<OpOverload(op='aten.gru_cell', 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 0x7f6eec424040>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425080>, kernel=<OpOverload(op='aten.rrelu', overload='default')>), <OpOverload(op='aten.not_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424d60>, kernel=<OpOverload(op='aten.not_equal', overload='Tensor')>), <OpOverload(op='aten._dim_arange', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425940>, kernel=<OpOverload(op='aten._dim_arange', overload='default')>), <OpOverload(op='aten.linalg_tensorsolve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426a20>, kernel=<OpOverload(op='aten.linalg_tensorsolve', overload='default')>), <OpOverload(op='_test.get_first', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454c20>, kernel=<OpOverload(op='_test.get_first', 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 0x7f6eec4540e0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441c60>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83bec0>, 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 0x7f6eed83bce0>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ef20>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4244a0>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f6a0>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4242c0>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42dbc0>, kernel=<OpOverload(op='aten.absolute', overload='default')>), <OpOverload(op='aten.nll_loss2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42dee0>, 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 0x7f6eec455440>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', 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 0x7f6eec42d3a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d120>, kernel=<OpOverload(op='aten._convolution_double_backward', 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 0x7f6eec440540>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d6c0>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>), <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c2c0>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454fe0>, kernel=<OpOverload(op='c10d_functional.all_reduce', 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 0x7f6eec4344a0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42cc20>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a340>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427b00>, kernel=<OpOverload(op='aten.choose_qparams_optimized', overload='default')>), <OpOverload(op='aten.conv_transpose3d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427740>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839300>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7f6eec427c40>, 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 0x7f6eec435120>, 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 0x7f6eec4354e0>, kernel=<OpOverload(op='aten._sparse_mm', 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 0x7f6eec426840>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4267a0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435620>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435ee0>, kernel=<OpOverload(op='aten.conv2d', overload='padding')>), <OpOverload(op='aten.matrix_exp_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435bc0>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443e20>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42fe20>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b740>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4340e0>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4425c0>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427d80>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4349a0>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f1a0>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83ab60>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e7a0>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e660>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f600>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f9c0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e5c0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4345e0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e840>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed8394e0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441580>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83afc0>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42efc0>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839440>, kernel=<OpOverload(op='aten.special_round', 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 0x7f6eed83a480>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42eca0>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83aca0>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83af20>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>), <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b420>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434fe0>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454540>, kernel=<OpOverload(op='image.decode_webp', overload='default')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4263e0>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427560>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b240>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839da0>, kernel=<OpOverload(op='aten.cummaxmin_backward', 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 0x7f6eed83aa20>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='tensor_qparams')>), <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454360>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839d00>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434180>, kernel=<OpOverload(op='aten.argsort', overload='default')>), <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a2a0>, kernel=<OpOverload(op='aten.is_distributed', overload='default')>), <OpOverload(op='aten.retains_grad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed7f1300>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839620>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839580>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443ec0>, kernel=<OpOverload(op='image.encode_png', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d4e0>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c7c0>, kernel=<OpOverload(op='aten.data', overload='default')>), <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426f20>, kernel=<OpOverload(op='aten._test_serialization_subcmul', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426de0>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425d00>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e200>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426020>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83aac0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425ee0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ed40>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426200>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426340>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839a80>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427ec0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a520>, kernel=<OpOverload(op='aten.trapz', overload='x')>), <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4253a0>, kernel=<OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>), <OpOverload(op='aten._cast_Int', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83bba0>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>), <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b9c0>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434680>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83bc40>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e020>, kernel=<OpOverload(op='aten.special_polygamma', 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 0x7f6eec424400>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42fec0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed838e00>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443880>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed8396c0>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d580>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c0e0>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454ea0>, kernel=<OpOverload(op='image.decode_gif', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443ba0>, kernel=<OpOverload(op='aten.chalf', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443100>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d440>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d760>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425300>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c360>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c900>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ccc0>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4272e0>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4276a0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427380>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayint')>), <OpOverload(op='aten._sparse_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4279c0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>), <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454a40>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425120>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435080>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434d60>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4356c0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436c00>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>), <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435e40>, kernel=<OpOverload(op='aten.special_gammainc', overload='default')>), <OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42ff60>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441260>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434360>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442200>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4416c0>, kernel=<OpOverload(op='aten.one_hot', 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 0x7f6eec42cf40>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440d60>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441d00>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440ea0>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441120>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4437e0>, kernel=<OpOverload(op='aten.linalg_vander', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='IntList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443740>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443600>, kernel=<OpOverload(op='aten.fix', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a160>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442c00>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442b60>, 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 0x7f6eec442ca0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442e80>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', 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 0x7f6eec4427a0>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442980>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4405e0>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454860>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', 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 0x7f6eec456480>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c9a0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4363e0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437060>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4372e0>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>), <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437380>, kernel=<OpOverload(op='aten.pinverse', 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 0x7f6eec427100>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436980>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455ee0>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436160>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440ae0>, kernel=<OpOverload(op='aten.lu_solve', 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 0x7f6eec440720>, kernel=<OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4560c0>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b060>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4407c0>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440900>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>), <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437c40>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4258a0>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425440>, kernel=<OpOverload(op='aten.max', overload='names_dim')>), <OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4400e0>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441e40>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441b20>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437f60>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437b00>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e8e0>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441da0>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', 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 0x7f6eec4418a0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed8398a0>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425b20>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437d80>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4374c0>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440a40>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e980>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83ad40>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b100>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a7a0>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4247c0>, kernel=<OpOverload(op='aten.is_inference', overload='default')>), <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4544a0>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443f60>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83a200>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436840>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed838ea0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eeebb5300>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed838d60>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443b00>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed8391c0>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434e00>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455da0>, kernel=<OpOverload(op='c10d_functional.broadcast', overload='default')>), <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed8393a0>, kernel=<OpOverload(op='aten.msort', 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 0x7f6eec4240e0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454b80>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425da0>, kernel=<OpOverload(op='aten.index_select_backward', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4262a0>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454900>, kernel=<OpOverload(op='_test.cat', overload='default')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f420>, 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 0x7f6eec42f880>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f740>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed838fe0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c4a0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427060>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed839940>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434cc0>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f2e0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <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 0x7f6eec426980>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83b920>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442020>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4420c0>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435760>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425580>, kernel=<OpOverload(op='aten.isreal', 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 0x7f6eed83bd80>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4549a0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426520>, kernel=<OpOverload(op='aten.kron', overload='default')>), <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424cc0>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435940>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425620>, kernel=<OpOverload(op='aten.diagflat', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424ea0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <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 0x7f6eec454180>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', 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 0x7f6eec4251c0>, kernel=<OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>), <OpOverload(op='aten.to_dense', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed83bb00>, kernel=<OpOverload(op='aten.to_dense', overload='default')>), <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424180>, kernel=<OpOverload(op='aten.cov', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4245e0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435440>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434040>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424220>, kernel=<OpOverload(op='aten.argwhere', overload='default')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424860>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441f80>, kernel=<OpOverload(op='aten.is_conj', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427ba0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec456840>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424680>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443420>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436480>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec456200>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424540>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435b20>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten._fused_rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434f40>, kernel=<OpOverload(op='aten._fused_rms_norm', overload='default')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434400>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435580>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec434720>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4353a0>, 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 0x7f6eec42f560>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7f6f9abe7d80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7f6f9abe7d80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7f6f9aa21940>, <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 0x7f6f9aa21940>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7f6f9aa22f20>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7f6f9aa231a0>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442700>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442f20>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7f6f9aa23c40>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442ac0>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7f6f9aa21620>, <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437600>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7f6f9aa23f60>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4376a0>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7f6f9aa5c220>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436fc0>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7f6f9aa5c4a0>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437240>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7f6f9aa5cd60>, <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4367a0>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7f6f9aa5cfe0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436d40>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7f6f9aa5d260>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436de0>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7f6f9aa5d4e0>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7f6f9aa5db20>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4360c0>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7f6f9aa5dbc0>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440680>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436520>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7f6f9aa5de40>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4409a0>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7f6f9aa5ccc0>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440040>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7f6f9aa5dee0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440360>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7f6f9aa5e160>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437ba0>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7f6f9aa5e3e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437e20>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437ec0>, 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 0x7f6eec4379c0>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7f6f9aa5ee80>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7f6f9aa5f240>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7f6f9aa8c220>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7f6f9aa8c4a0>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7f6f9aa8cb80>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7f6f9aa8d800>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7f6f9aa8cfe0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7f6f9aa8d620>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7f6f9aa8dbc0>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7f6f9aa8e980>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7f6f9aa8eca0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7f6f9aa8e2a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7f6f9aa8e980>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7f6f9aa8eca0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42e700>, kernel=<OpOverload(op='aten.margin_ranking_loss', overload='default')>), <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42f060>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7f6f9aa8fce0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7f6f9aaac0e0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7f6f9aaac0e0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7f6f9aaac400>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7f6f9aaac400>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7f6f9aa8e8e0>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7f6f9aaacc20>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7f6f9aaadd00>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7f6f9aaadd00>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec426b60>, 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 0x7f6eec425e40>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7f6f9aaae020>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7f6f9aaae2a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7f6f9aaacd60>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7f6f9aaae0c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7f6f9aaae340>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7f6f9aaae480>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7f6f9aaae5c0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7f6f9aaaeca0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7f6f9aaaeca0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7f6f9aaaf060>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7f6f9aaaf060>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7f6f9aaaf420>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7f6f9aaaf9c0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7f6f9aaaf9c0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7f6f9aac8f40>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7f6f9aaafec0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7f6f9aaafec0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7f6f9aac8400>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7f6f9aac8400>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7f6f9aac87c0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7f6f9aac87c0>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7f6f9aac8b80>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7f6f9aaad940>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7f6f9aaad940>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441080>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7f6f9aac9080>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7f6f9aac9440>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7f6f9aac9440>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7f6f9aac9a80>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7f6f9aac9a80>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7f6f9aac9ee0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7f6f9ab1afc0>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7f6f9ab2c2c0>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7f6f9ab2c2c0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7f6f9ab2cb80>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7f6f9ab2d440>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7f6f9ab2cfe0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7f6f9ab2d3a0>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7f6f9ab2d620>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7f6f9ab2da80>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7f6f9ab2fba0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7f6f9ab2ff60>, <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427e20>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4351c0>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7f6f9ab44b80>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7f6f9ab44f40>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7f6f9ab451c0>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7f6f9ab45440>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7f6f9ab745e0>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7f6f9ab46520>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7f6f9ab47240>, <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427880>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.any', overload='out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42d1c0>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7f6f9ab467a0>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427f60>, 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 0x7f6eec42e3e0>, 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 0x7f6eec42c220>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c540>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec42c860>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7f6f9ab472e0>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec427600>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7f6f9ab47100>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435300>, kernel=<OpOverload(op='aten.var_mean', overload='default')>), <OpOverload(op='aten.var_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec435f80>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4359e0>, 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 0x7f6eec435c60>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7f6f9ab47a60>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7f6f9ab47ec0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7f6f9ab75800>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7f6f9ab8c540>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7f6f9ab76200>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7f6f9ab753a0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7f6f9ab76340>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7f6f9ab74400>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7f6f9ab74400>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7f6f9ab76700>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7f6f9ab76d40>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec425a80>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7f6f9ab76fc0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7f6f9ab76fc0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7f6f9ab772e0>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7f6f9ab77920>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eed8399e0>, 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 0x7f6eed839800>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7f6f9ab77f60>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7f6f9ab77d80>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7f6f9ab8c860>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7f6f9ab8c0e0>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7f6f9ab8cb80>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7f6f9ab8cb80>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7f6f9ab8cf40>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7f6f9ab8cf40>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7f6f9ab8d8a0>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7f6f9ab8d8a0>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec424900>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7f6f9ab8d940>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442160>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7f6f9ab8d9e0>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7f6f9ab8f240>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7f6f9ab8f600>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7f6f9ab8f560>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7f6f9ab8f880>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442d40>, 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 0x7f6eec442fc0>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7f6f9abb4b80>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7f6f9abb49a0>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7f6f9abb49a0>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436660>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436ca0>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7f6f9abb5e40>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7f6f9abb5e40>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7f6f9abb5300>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7f6f9abb5300>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7f6f9abb5800>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7f6f9abb5760>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7f6f9abb5a80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7f6f9abb5760>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7f6f9abb6020>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7f6f9abb6020>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7f6f9abb63e0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7f6f9abb63e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7f6f9abb67a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7f6f9abb67a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7f6f9abb6b60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7f6f9abb6c00>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7f6f9abb7560>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7f6f9abb7a60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7f6f9abb7ce0>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7f6f9abb5f80>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7f6f9abb7f60>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7f6f9abb7f60>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7f6f9abe59e0>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7f6f9abe42c0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7f6f9abe51c0>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7f6f9abb7c40>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7f6f9abe40e0>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7f6f9abe5440>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7f6f9abe4360>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7f6f9abe44a0>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7f6f9abe44a0>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7f6f9abe56c0>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7f6f9abe45e0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7f6f9abe56c0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7f6f9abe4720>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7f6f9abe6480>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7f6f9abe4860>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7f6f9abe49a0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7f6f9abe5940>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7f6f9abe4ae0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7f6f9abe5940>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7f6f9abe4c20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7f6f9aa20860>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7f6f9abe4d60>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7f6f9abe5bc0>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7f6f9abe4ea0>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7f6f9abe4ea0>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7f6f9abe6200>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7f6f9abe4fe0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7f6f9abe4fe0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7f6f9abe5120>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7f6f9abe5260>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7f6f9abe5260>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7f6f9abe5c60>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7f6f9abe53a0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7f6f9abe53a0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7f6f9abe54e0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7f6f9abe54e0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7f6f9abe5b20>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7f6f9abe5620>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7f6f9abe5760>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7f6f9abe5760>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7f6f9abe5f80>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7f6f9abe5b20>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7f6f9abe58a0>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7f6f9abe58a0>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7f6f9abe59e0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7f6f9abe6700>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7f6f9abe4f40>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7f6f9abe4cc0>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7f6f9abe4cc0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7f6f9abe6980>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7f6f9abe4a40>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7f6f9abe47c0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7f6f9abe6c00>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7f6f9abe4540>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7f6f9abe4220>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7f6f9abe5da0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7f6f9abe6e80>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7f6f9abe5ee0>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7f6f9abe7100>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7f6f9abe6020>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7f6f9abe6020>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7f6f9abe6160>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7f6f9abe62a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7f6f9abe62a0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7f6f9abe7ec0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7f6f9abe63e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7f6f9abe63e0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7f6f9abe6520>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7f6f9abe6660>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7f6f9abe67a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7f6f9abe67a0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7f6f9abe72e0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7f6f9abe68e0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7f6f9abe68e0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7f6f9abe6a20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7f6f9abe6b60>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7f6f9aa218a0>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7f6f9abe6ca0>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7f6f9abe71a0>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7f6f9abe6de0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7f6f9abe71a0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7f6f9abe6f20>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7f6f9abe7060>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7f6f9abe7c40>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7f6f9abe4040>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7f6f9abe4040>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7f6f9abe4680>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7f6f9abe4680>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7f6f9abe4b80>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7f6f9abe6340>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7f6f9abe5080>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7f6f9abe5580>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7f6f9abe5580>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7f6f9abe5e40>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7f6f9abe5a80>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7f6f9abe7420>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7f6f9abe4900>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7f6f9abe7560>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7f6f9abe7560>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7f6f9abe76a0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7f6f9abe77e0>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7f6f9abe7920>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7f6f9abe7a60>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7f6f9abe7ba0>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7f6f9abe5d00>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7f6f9abe7ce0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7f6f9abe7e20>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7f6f9abe7f60>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7f6f9abe7600>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7f6f9aa200e0>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7f6f9abe7600>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7f6f9aa20220>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7f6f9aa20360>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7f6f9aa204a0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7f6f9abe7880>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7f6f9abe7b00>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7f6f9aa205e0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7f6f9aa205e0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7f6f9abe7b00>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7f6f9aa20720>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7f6f9abe6d40>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7f6f9abe6d40>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7f6f9abe6ac0>, <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 0x7f6f9abe6ac0>, <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 0x7f6f9aa20900>, <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 0x7f6f9aa20900>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7f6f9aa202c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7f6f9aa20c20>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7f6f9aa20c20>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7f6f9aa20f40>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7f6f9aa20f40>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7f6f9aa21260>, <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 0x7f6f9aa21260>, <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 0x7f6f9aa21260>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7f6f9aa21580>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7f6f9aa21580>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7f6f9aa218a0>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7f6f9abe4e00>, <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 0x7f6f9abe4e00>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7f6f9ac19b20>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7f6f9ac19b20>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7f6f9ac1a2a0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac1ad40>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7f6f9ac185e0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7f6f9ac185e0>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7f6f9ac1b9c0>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7f6f9ac1b380>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7f6f9ac1b380>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7f6f9ac1b740>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7f6f9ac1b740>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7f6f9ac1bba0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7f6f9ac1bf60>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7f6f9ac1bf60>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e0c0>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7f6f9ac3ca40>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7f6f9ac3ca40>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7f6f9ac3cea0>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e020>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7f6f9ac3d3a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7f6f9ac3d3a0>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7f6f9ac3d1c0>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d440>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d580>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d6c0>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d800>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e2a0>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d940>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d940>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3da80>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dbc0>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dd00>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3de40>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d080>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3df80>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3df80>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e0c0>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dda0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3dda0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3db20>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d8a0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d620>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3d300>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3ce00>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3ce00>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3cae0>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e340>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e480>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4428e0>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7f6f9acb7ba0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7f6f9acb7ba0>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7f6f9accd620>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7f6f9accd620>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7f6f9acce020>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7f6f9accdbc0>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440860>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7f6f9acced40>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7f6f9accf240>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7f6f9accfc40>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7f6f9accfc40>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7f6f9ace1620>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7f6f9ace1620>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7f6f9ace31a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7f6f9ace36a0>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7f6f9ace36a0>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7f6f9acf8ae0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7f6f9acfb2e0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7f6f9acfbba0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7f6f9acfbd80>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7f6f9add3d80>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7f6f9ab1a2a0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7f6f9accea20>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7f6f9ad5a340>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7f6f9ad5a340>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7f6f9ad5ae80>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7f6f9ad5a700>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7f6f9ad5a700>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7f6f9ad5a2a0>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7f6f9ad5a2a0>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7f6f9ad5b2e0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7f6f9ad5b380>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7f6f9ad5b380>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7f6f9ad5bc40>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7f6f9ad5ba60>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7f6f9ad5bb00>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7f6f9ad5be20>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7f6f9ad5be20>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7f6f9ad7c220>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7f6f9ad7c220>, <OpOverload(op='aten.mish_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4404a0>, kernel=<OpOverload(op='aten.mish_backward', overload='default')>), <OpOverload(op='aten.silu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec437ce0>, kernel=<OpOverload(op='aten.silu_backward', overload='default')>), <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7f6f9ad7cae0>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7f6f9ad7cea0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7f6f9ad7d3a0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7f6f9ad7d3a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7f6f9ad7d440>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7f6f9ad7d440>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7f6f9ad7f7e0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7f6f9ad7dd00>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7f6f9ad7f7e0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7f6f9ad7d940>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7f6f9ad7d940>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7f6f9ad7ccc0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7f6f9ad7dee0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7f6f9ad7dc60>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7f6f9ad7dc60>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7f6f9ad7e0c0>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7f6f9ada5580>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7f6f9ad7e2a0>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7f6f9ada49a0>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7f6f9ad7e520>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7f6f9ad7e520>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7f6f9ad7e8e0>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7f6f9ad7e8e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7f6f9ad7ec00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7f6f9ad7f1a0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7f6f9ad7f1a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7f6f9ad7f240>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7f6f9ad7f240>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7f6f9ad7f880>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7f6f9ad7f880>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7f6f9ad7de40>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7f6f9ad7de40>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7f6f9ad7f600>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7f6f9ad7f600>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7f6f9ad7fb00>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7f6f9ad7fb00>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7f6f9ad7ed40>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7f6f9ada4220>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7f6f9ada4220>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7f6f9ada44a0>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7f6f9ada45e0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7f6f9ada45e0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7f6f9ada4c20>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7f6f9ada4c20>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7f6f9ada4f40>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7f6f9ada5300>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7f6f9ada5580>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7f6f9ada5940>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7f6f9ada5f80>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7f6f9ada6200>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7f6f9ada67a0>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7f6f9ada67a0>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7f6f9ada6480>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7f6f9ada6840>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7f6f9ada6840>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7f6f9ada6ac0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7f6f9ada5ee0>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7f6f9ada5c60>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7f6f9ada6ca0>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7f6f9ada7240>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7f6f9ada6ca0>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7f6f9ada7420>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7f6f9ada6160>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7f6f9ada7380>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7f6f9ada7560>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7f6f9add1800>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7f6f9add16c0>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436e80>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7f6f9add0720>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7f6f9add07c0>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7f6f9add0900>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7f6f9ada7ec0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7f6f9add0400>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7f6f9add11c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7f6f9add11c0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7f6f9ada63e0>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7f6f9add1620>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7f6f9add2fc0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7f6f9add0400>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7f6f9add1f80>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7f6f9add1f80>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7f6f9add2660>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7f6f9add2660>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7f6f9add31a0>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7f6f9add31a0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7f6f9add2ac0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7f6f9add2ac0>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7f6f9add2de0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7f6f9add2de0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7f6f9add3060>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7f6f9add3060>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7f6f9add2e80>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4365c0>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7f6f9add14e0>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7f6f9add39c0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7f6f9add39c0>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7f6f9add3ba0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7f6f9adf9080>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7f6f9adf8fe0>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7f6f9adf8d60>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7f6f9adf8fe0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7f6f9adf89a0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7f6f9adf9080>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7f6f9adf96c0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7f6f9adf93a0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7f6f9adf96c0>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7f6f9adf9a80>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7f6f9adfa020>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7f6f9add1760>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7f6f9adf9c60>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7f6f9adf9e40>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7f6f9adfa480>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7f6f9adfa660>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7f6f9adfa980>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7f6f9adfab60>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7f6f9adfbba0>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7f6f9adfb600>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7f6f9adfb920>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7f6f9adfb420>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7f6f9adfbf60>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7f6f9adfa8e0>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7f6f9ac18360>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7f6f9ac18540>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7f6f9adf9f80>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7f6f9adf9f80>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7f6f9ac1bba0>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7f6f9ac191c0>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7f6f9ac191c0>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7f6f9ac19580>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7f6f9ac1a840>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7f6f9ac198a0>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7f6f9ab1b420>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7f6f9ab1b420>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7f6f9ab1b880>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7f6f9ab1b880>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7f6f9ab1bce0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7f6f9ab1bce0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7f6f9ab46a20>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7f6f9ab2c720>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7f6f9ad5ba60>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7f6f9ab2dee0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7f6f9ab2e340>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7f6f9ab2e7a0>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7f6f9ad5b740>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec442840>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7f6f9ab2f060>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7f6f9ab2f4c0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7f6f9ab2f4c0>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7f6f9ab447c0>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7f6f9adfaf20>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7f6f9ab2f920>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e2a0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7f6f9b19afc0>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7f6f9b19afc0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7f6f9ab2f100>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7f6f9ab8c5e0>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7f6f9ab77a60>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7f6f9ab744a0>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441bc0>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7f6f9aaad760>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7f6f9add3d80>, <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec440b80>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7f6f9ab45a80>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3e020>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7f6f9ab1a2a0>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7f6f9adfad40>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7f6f9ab465c0>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4371a0>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7f6f9abe5800>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7f6f9ab46980>, <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436700>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.var', overload='out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436b60>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4368e0>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436f20>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7f6f9ab47060>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7f6f9ada5d00>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7f6f9ab46b60>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7f6f9abe7920>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7f6f9ada44a0>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7f6f9ab8ccc0>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7f6f9abb6f20>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7f6f9abb6340>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7f6f9ab74fe0>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec441300>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7f6f9ad5bb00>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7f6f9acb4d60>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7f6f9accc5e0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7f6f9accc0e0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7f6f9ab8d300>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac3c040>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7f6f9acf80e0>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7f6f9acf96c0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7f6f9acccae0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7f6f9abe7380>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7f6f9ac19e40>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7f6f9acccfe0>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7f6f9accdb20>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7f6f9ace0180>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7f6f9ace0180>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7f6f9ace0680>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7f6f9ac1a980>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7f6f9ab8fb00>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7f6f9ab74ae0>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7f6f9ace0b80>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7f6f9ace1080>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7f6f9ace1580>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7f6f9aa5e660>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7f6f9ace23e0>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7f6f9ace1da0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7f6f9ace27a0>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7f6f9ace2ca0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7f6f9ace3ba0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7f6f9acf80e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7f6f9ad7ec00>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7f6f9acf85e0>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7f6f9ace2700>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7f6f9acf8e00>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7f6f9aa8d620>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7f6f9ada7d80>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7f6f9acf9bc0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7f6f9ac3d1c0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7f6f9acfa0c0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7f6f9ac3c860>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7f6f9abb4ae0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7f6f9acfaa20>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7f6f9acfae80>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7f6f9acfb740>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7f6f9ace1b20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7f6f9acfaac0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7f6f9ac1a660>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7f6f9acfbf60>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7f6f9ab184a0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7f6f9ab19440>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7f6f9ab19440>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7f6f9ab198a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7f6f9ab198a0>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7f6f9ac18680>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7f6f9ab19d00>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7f6f9ab19da0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7f6f9ab19da0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7f6f9ab1a700>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7f6f9ab1ab60>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7f6f9ac3f100>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7f6f9acb4180>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7f6f9ad7ed40>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7f6f9acb47c0>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7f6f9acb4cc0>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7f6f9acb51c0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7f6f9acb56c0>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7f6f9acb5bc0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7f6f9acb6e80>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7f6f9acb7380>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7f6f9acb60c0>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7f6f9abe5300>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7f6f9ad7ca40>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7f6f9acb65c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7f6f9acb6980>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7f6f9acb6980>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7f6f9abb4180>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7f6f9ab2f560>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7f6f9ac3f380>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7f6f9ac3f380>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7f6f9acfa5c0>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7f6f9abb5080>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7f6f9ab2ec00>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7f6f9ac1a340>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7f6f9ab8c860>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7f6f9ab744a0>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7f6f9ada6de0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7f6f9ab8efc0>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7f6f9abb4540>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7f6f9abb42c0>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7f6f9ab8e840>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7f6f9ab8eac0>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7f6f9ab8ec00>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7f6f9ab8da80>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7f6f9ab8c9a0>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec436020>, 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 0x7f6eec436340>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7f6f9ad7ff60>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7f6f9ac1b920>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7f6f9abe72e0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7f6f9ab18900>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7f6f9abb6b60>, <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4553a0>, 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 0x7f6eec4551c0>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455260>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454d60>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455940>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4558a0>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', 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 0x7f6eec454720>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454f40>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455b20>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454040>, kernel=<OpOverload(op='quantized.conv3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec443d80>, 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 0x7f6eec456160>, 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 0x7f6eec456980>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec456660>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4567a0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4554e0>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454220>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455bc0>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454400>, kernel=<OpOverload(op='quantized.conv_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 0x7f6eec443ce0>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454ae0>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4545e0>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455c60>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4565c0>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4563e0>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455e40>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454e00>, kernel=<OpOverload(op='quantized.conv2d_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 0x7f6eec456020>, 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 0x7f6eec455f80>, kernel=<OpOverload(op='quantized.conv3d_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 0x7f6eec4562a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec456340>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec4556c0>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec455760>, kernel=<OpOverload(op='quantized.conv3d_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 0x7f6eec4568e0>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', 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 0x7f6eec454cc0>, kernel=<OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec454680>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7f6eec456a20>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}