.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.0006570490004378371 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.001336726999397797 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-COU._add_shape_information] dynamic shapes replacements={} [GraphBuilder-COU.optimize] start with 13 nodes [GraphBuilder-COU.optimize] #patterns=84 [GraphBuilder-COU.optimize] start with subgraphs [GraphBuilder-COU.optimize] done with subgraphs [GraphBuilderPatternOptimization-COU.optimize] start with 7 nodes, 0 initializers, 84 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-COU.optimize] same children={'SameChildrenFromInputPattern', 'SameChildrenPattern'} [GraphBuilderPatternOptimization-COU.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-COU.optimize] increase priority to 1 [GraphBuilderPatternOptimization-COU.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-COU.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=TransposeMatMulPattern:0.000 [GraphBuilderPatternOptimization-COU.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-COU.optimize] increase priority to 3 [GraphBuilderPatternOptimization-COU.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-COU.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000 [GraphBuilderPatternOptimization-COU.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-COU.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-COU.optimize] done after 5 iterations with 4 nodes in 0.008 [GraphBuilder-COU.optimize] done with 4 nodes in 0.009 [GraphBuilder-COU.to_onnx] make_model 1 inits 0 params [GraphBuilder-COU.time_evaluation_constants_] 0 [GraphBuilder-COU._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-COU._build_initializers] switch low/high order [GraphBuilder-COU._build_initializers] done in 8.78999344422482e-07s with 1 initializers, 0 large initializers [GraphBuilder-COU._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.009857406999799423s and 4 nodes, 1 initializers, 5 inputs, 1 outputs [onnx_custom_backend] to_onnx done in 0.01245469600053184 with 4 nodes and 0 local functions. [onnx_custom_backend] starts creating InferenceSession [onnx_custom_backend] InferenceSession done in 0.0028903979991810047 discrepancies: 4.470348358154297e-08
onnx_debug_backend¶
- experimental_experiment.torch_dynamo.onnx_debug_backend(graph_module: torch.fx.GraphModule, args: List[torch.Tensor | torch.SymInt | torch.SymFloat], target_opset: int | None = None, backend: str | Callable[[ModelProto, bool | None], Any] = 'ort', verbose: int | Tuple[int, int] = 0, dump_prefix: None = None, dump_patterns: str | None = None, providers: Tuple[str] | None = None, raise_exc: bool = True, storage: Dict[str, Any] | None = None, raise_list: Set[str] | None = None, enable_pattern: str | List[str | type] | None = 'default', disable_pattern: str | List[str | type] | None = None, pre_ort_model_transforms: Callable[[ModelProto], ModelProto] | List[Callable[[ModelProto], ModelProto]] | None = None, ort_optimization_level: str | None = None, dispatcher: Dispatcher | None = None, rename_inputs: bool = True, optimize: bool = True, processor: str = 'CPU', order_algorithm: str | None = None) Callable[source]¶
Custom backend to export torch models into onnx (see torch.compiler). This backend is not meant to be efficient, it is more to check the conversion is ok. It relies either on onnxruntime or the python reference implementation.
- Parameters:
graph_module – graph to export
args – arguments
target_opset – opset to use for the conversion
backend – after the conversion, the model is executed with a runtime, onnxruntime or the reference implementation, it must be a value among ‘ort’, ‘ref’ or a class, it can be a function as well which returns an object behaving the same way
verbose – adjust verbosity, if tuple, if gives different verbosity level to the exporter and the runtime
dump_prefix – prefix used to dump the model generated by the backend
dump_patterns – dump the patterns as well
providers – where to run the model, by default
raise_exc – raise an exception whenever something goes wrong
storage – to store any interesting objects during the process
raise_list – the builder stops any time a name falls into that list, this is a debbuging tool
enable_pattern – optimization patterns to enable
disable_pattern – optimization patterns to disable
pre_ort_model_transforms – list of transformations applied on the final ModelProto
ort_optimization_level – graph optimization level for onnxruntime, the default value is the same as what onnxruntime defines
dispatcher – see
experimental_experiment.torch_interpreter.Dispatcherrename_inputs – rename inputs into
input_{i}optimize – enable or disable the optimization
processor – specifies the processor it is optimized for
order_algorithm – algorithm optimizing the order the onnx node, none by default
- Returns:
Callable
If not empty, storage keeps the memory of the data generated, onnx models, graph module as well the inputs and outputs when the model is run.
The following example shows how to use the reference implementation (
experimental_experiment.reference.ExtendedReferenceEvaluator) to run the onnx model and display the intermediate results.<<<
import torch from experimental_experiment.torch_dynamo import onnx_debug_backend class MLP(torch.nn.Module): def __init__(self): super().__init__() self.layers = torch.nn.Sequential( torch.nn.Linear(10, 32), torch.nn.Sigmoid(), torch.nn.Linear(32, 1), ) def forward(self, x): return self.layers(x) x = torch.randn(3, 10, dtype=torch.float32) mlp = MLP() expected = mlp(x) compiled_model = torch.compile( mlp, backend=lambda *args, **kwargs: onnx_debug_backend( *args, verbose=(1, 10), backend="ref", **kwargs ), dynamic=False, fullgraph=True, ) got = compiled_model(x) diff = (expected - got).max() print(f"discrepancies: {diff}")
>>>
[to_onnx] build the graph module from <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'>, type(args)=<class 'tuple'> [to_onnx] build the graph module with input_names=['input0', 'input1', 'input2', 'input3', 'input4'] [_make_builder_interpreter] use existing <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'> [to_onnx] graph module done in 0.0005451010001706891 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.0014949149999665678 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-FDG._add_shape_information] dynamic shapes replacements={} [GraphBuilder-FDG.optimize] start with 13 nodes [GraphBuilder-FDG.optimize] #patterns=84 [GraphBuilder-FDG.optimize] start with subgraphs [GraphBuilder-FDG.optimize] done with subgraphs [GraphBuilderPatternOptimization-FDG.optimize] start with 7 nodes, 0 initializers, 84 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-FDG.optimize] same children={'SameChildrenFromInputPattern', 'SameChildrenPattern'} [GraphBuilderPatternOptimization-FDG.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-FDG.optimize] increase priority to 1 [GraphBuilderPatternOptimization-FDG.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-FDG.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=ShapeBasedSameChildrenPattern:0.000 [GraphBuilderPatternOptimization-FDG.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-FDG.optimize] increase priority to 3 [GraphBuilderPatternOptimization-FDG.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-FDG.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000 [GraphBuilderPatternOptimization-FDG.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-FDG.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-FDG.optimize] done after 5 iterations with 4 nodes in 0.011 [GraphBuilder-FDG.optimize] done with 4 nodes in 0.012 [GraphBuilder-FDG.to_onnx] make_model 1 inits 0 params [GraphBuilder-FDG.time_evaluation_constants_] 0 [GraphBuilder-FDG._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-FDG._build_initializers] switch low/high order [GraphBuilder-FDG._build_initializers] done in 1.0579997251625173e-06s with 1 initializers, 0 large initializers [GraphBuilder-FDG._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.01397530799931701s and 4 nodes, 1 initializers, 5 inputs, 1 outputs +C init7_s2_-1_1: int64:(2,):[-1, 1] +I input0: float32:(32, 10):0.08864717185497284,0.17967171967029572,0.31054040789604187,-0.10735508799552917,0.2581270933151245... +I input1: float32:(32,):0.003046581754460931,-0.2638026773929596,-0.24146156013011932,0.011907667852938175,0.0475366935133934... +I input2: float32:(3, 10):-0.5730117559432983,-2.840912342071533,-0.3145088851451874,0.9781247973442078,1.3747093677520752... +I input3: float32:(1, 32):0.11449513584375381,0.08343535661697388,0.05976271256804466,-0.04982006177306175,-0.024628784507513046... +I input4: float32:(1,):[0.12051684409379959] Gemm(input2, input0, input1) -> input_1 + input_1: float32:(3, 32):-0.7163377404212952,-0.30506330728530884,0.673978328704834,-0.8275230526924133,-0.5177251696586609... Sigmoid(input_1) -> input_2 + input_2: float32:(3, 32):0.32819995284080505,0.42432019114494324,0.6623933911323547,0.30416905879974365,0.3733843266963959... 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.11449513584375381,0.08343535661697388,0.05976271256804466,-0.04982006177306175,-0.024628784507513046... Gemm(input_2, l_self_modules_layers_modules_2_parameters_weight_::T10, input4) -> output_0 + output_0: float32:(3, 1):[0.11518534272909164, 0.25112447142601013, 0.30678293108940125] discrepancies: 0.0
dynger_backend¶
- experimental_experiment.torch_dynamo.dynger_backend(graph_module: GraphModule, args: List[Tensor | SymInt | SymFloat], dynamic_shapes: Dict[str, Any] | Tuple[Any] | None = None, optimize: bool = True, verbose: int | Tuple[int, int] = 0) Callable[source]¶
Eager backend for dynamo.
- Parameters:
graph_module – graph to export
args – arguments
optimize – optimize or not, those optimization would be done on the graph module itself
verbose – adjust verbosity, if tuple, if gives different verbosity level to the exporter and the runtime
- Returns:
Callable
Next examples shows how to display intermediate results while executing the graph produced by torch dynamo.
<<<
import torch from experimental_experiment.torch_dynamo import dynger_backend class MLP(torch.nn.Module): def __init__(self): super().__init__() self.layers = torch.nn.Sequential( torch.nn.Linear(10, 32), torch.nn.Sigmoid(), torch.nn.Linear(32, 1), ) def forward(self, x): return self.layers(x) x = torch.randn(3, 10, dtype=torch.float32) mlp = MLP() expected = mlp(x) compiled_model = torch.compile( mlp, backend=lambda *args, **kwargs: dynger_backend(*args, verbose=10, **kwargs), dynamic=False, fullgraph=True, ) got = compiled_model(x) diff = (expected - got).max() print(f"discrepancies: {diff}")
>>>
[dynger_backend] use existing <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'> [dynger_backend] begin execution with 9 nodes <built-in function linear>((l_x_, l_self_modules_layers_modules_0_parameters_weight_, l_self_modules_layers_modules_0_parameters_bias_)) -> input_1 + input_1: torch.float32:torch.Size([3, 32]):-1.6570229530334473,0.9128591418266296,0.7375975847244263,-1.0163816213607788,0.5495316386222839... <built-in method sigmoid of type object at 0x752f59d2eb20>((input_1,)) -> input_2 + input_2: torch.float32:torch.Size([3, 32]):0.1601620316505432,0.713584840297699,0.6764703392982483,0.2657328248023987,0.6340269446372986... <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.11560054123401642,-0.17230308055877686,-0.06600581109523773 [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 0x752eed47cef0>: <function out_dtype_decomp at 0x752eecd35ee0>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x752eecb19da0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x752eecb19ee0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x752eecb1a020>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x752eecb1a160>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x752eecb1a2a0>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x752eecb1a3e0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x752eecb19e40>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x752eecb1a520>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x752eecb1a660>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x752eecb1a340>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x752eecb1a7a0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x752eecb1a7a0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x752eecb1a0c0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x752eecb1a8e0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x752eecb19940>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x752eecb19940>, <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 0x752eecb18540>, <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 0x752eecb18540>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x752eecb191c0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x752eecb191c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x752eecb1ade0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x752eecb1ade0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x752eecb1b100>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x752eecb1b100>, <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 0x752eecce7880>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <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 0x752eecb1b420>, <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 0x752eecb1b420>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x752eecce7880>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x752eecb1b740>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x752eecb1ba60>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x752eecb1be20>, <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 0x752eecb1be20>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x752eecb1ae80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x752eecb1ae80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <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 0x752eecb19d00>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x752eecb55120>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x752eecb55120>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x752eecb553a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x752eecb553a0>, <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x752eecb55e40>, <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x752eecb55e40>, <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x752eecb1ac00>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x752eecb1ac00>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x752eecb57100>, <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x752eecb55300>, <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x752eecb55300>, <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x752eecb57100>, <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x752eecb56160>, <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x752eecb56160>, <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x752eecb563e0>, <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x752eecb563e0>, <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x752eecb56660>, <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x752eecb56660>, <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x752eecb56f20>, <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x752eecb56f20>, <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x752eecb571a0>, <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x752eecb571a0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x752eecb57420>, <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x752eecb57420>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x752eecb576a0>, <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x752eecb576a0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x752eecb57d80>, <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x752eecb57d80>, <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x752eecb57f60>, <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x752eecb57f60>, <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x752eecb56020>, <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x752eecb56020>, <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x752eecb780e0>, <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x752eecb780e0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x752eecb78360>, <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x752eecb78360>, <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x752eecb785e0>, <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x752eecb785e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x752eecb78860>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x752eecb78860>, <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x752eecb78900>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x752eecb789a0>, <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x752eecb79080>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x752eecb79080>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x752eecb79440>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x752eecb79440>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x752eecb7a3e0>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x752eecb7a660>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x752eecb7a660>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x752eecb7ad40>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x752eecb7ad40>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x752eecb7b1a0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x752eecb7b1a0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x752eecb7b9c0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x752eecb7b7e0>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x752eecb7b7e0>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x752eecb7b9c0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x752eecb7bd80>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x752eecb7bd80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x752eecbb4b80>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x752eecbb65c0>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x752eecbb65c0>, <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x752eecb7bec0>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x752eecbb4ea0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x752eecbb44a0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x752eecbb44a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x752eecbb4b80>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x752eecbb4ea0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x752eecbb5620>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x752eecbb5940>, <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x752eecbb5ee0>, <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x752eecbb5ee0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x752eecbb62a0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x752eecbb62a0>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x752eecbb6200>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x752eecbb6200>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x752eecbb6de0>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x752eecbb6de0>, <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x752eecbb7060>, <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x752eecbb72e0>, <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x752eecbb7920>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x752eecbb7920>, <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x752eecbb7ba0>, <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x752eecbb7f60>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x752eecbb6ca0>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x752eecbb6ca0>, <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x752eecbcc220>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x752eecbcc220>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x752eecbcc4a0>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x752eecbcc4a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x752eecbb6f20>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x752eecbcc2c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x752eecbcc540>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x752eecbcc680>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x752eecbcc7c0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x752eecbccea0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x752eecbccea0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x752eecbcd260>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x752eecbcd260>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x752eecbcd620>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x752eecbcd620>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x752eecbcdbc0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x752eecbcdbc0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x752eecbce0c0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x752eecbce0c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x752eecbce5c0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x752eecbce5c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x752eecbce980>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x752eecbce980>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x752eecbced40>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x752eecbced40>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x752eecbce160>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x752eecbce160>, <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x752eecbcf240>, <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x752eecbcf240>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x752eecbcf600>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x752eecbcf600>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x752eecbcfc40>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x752eecbcfc40>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x752eecc376a0>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x752eecc35260>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x752eecc35260>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x752eecc376a0>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x752eecc36520>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x752eecc36520>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x752eecc36de0>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x752eecc36de0>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x752eecc37240>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x752eecc37240>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x752eecc36020>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x752eecc36020>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x752eecc37880>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x752eecc37880>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x752eecc37ce0>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x752eecc37ce0>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x752eecc4aa20>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x752eecc4aa20>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x752eecc4ade0>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x752eecc4ade0>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std', overload='default')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='dim')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x752eecc77920>, <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x752eecc760c0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x752eecc760c0>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x752eecc771a0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x752eecc771a0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x752eecc77ce0>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x752eecc77380>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x752eecc77560>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x752eecc779c0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x752eecc779c0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x752eecc77ec0>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x752eecc77ec0>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x752eecc75620>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x752eecc75620>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x752eecc92840>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x752eecc90900>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x752eecc90900>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x752eecc90f40>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x752eecc90f40>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x752eecc90ea0>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x752eecc911c0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x752eecc911c0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x752eecc914e0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x752eecc914e0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x752eecc91b20>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x752eecc91b20>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x752eecc93240>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x752eecc92a20>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x752eecc90a40>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x752eecc92d40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x752eecc92d40>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x752eeccb9440>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x752eecc93100>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x752eecc93100>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x752eecc93420>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x752eecc93600>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x752eecc93a60>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x752eecc93a60>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x752eeccb9440>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x752eecc93ce0>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x752eeccba980>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x752eeccbbf60>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x752eeccbbf60>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x752eeccb9ee0>, <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x752eeccb9ee0>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x752eeccbae80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x752eeccbae80>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x752eeccbab60>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x752eeccbab60>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x752eeccbb4c0>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x752eeccbb4c0>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x752eeccbb9c0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x752eeccbb9c0>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x752eeccbbc40>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x752eeccbbc40>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x752eecce4220>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x752eecce4220>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x752eecce45e0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x752eecce45e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x752eecce49a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x752eecce49a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x752eecce4d60>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x752eecce4d60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x752eecce4e00>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x752eecce5760>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x752eecce5760>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x752eecce5c60>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x752eecce5c60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x752eecce4cc0>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x752eecce4cc0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x752eecce7ec0>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x752eecce6200>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x752eecce6200>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x752eecce6480>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x752eeccbb1a0>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x752eecce7e20>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x752eecce62a0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x752eecce7e20>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x752eecce6520>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x752eecce6660>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x752eecce6660>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x752eecce67a0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x752eecce68e0>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x752eecce7ce0>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x752eecce6a20>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x752eecb1a5c0>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x752eecce6b60>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x752eecce6ca0>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x752eecce6de0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x752eecce6f20>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x752eecce7ba0>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x752eecce7060>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x752eecce7060>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x752eecce71a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x752eecce71a0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x752eecce7240>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x752eecce72e0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x752eecb1ab60>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x752eecce7420>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x752eecce7420>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x752eecce74c0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x752eecce7560>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x752eecce7560>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x752eecce76a0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x752eecce76a0>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x752eecce7740>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x752eecb19300>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x752eecce77e0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x752eecce7740>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x752eecce7c40>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x752eecce7920>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x752eecce7920>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x752eecce7a60>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x752eecce7a60>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x752eecce79c0>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x752eecce7ba0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x752eecce79c0>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x752eecb19580>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x752eecce6d40>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x752eecce6d40>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x752eecce6ac0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x752eecb193a0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x752eecce6840>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x752eecce65c0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x752eecce7b00>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x752eecce6160>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x752eecb1aac0>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x752eecce7f60>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x752eecb180e0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x752eecce7600>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x752eecb18220>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x752eecb18220>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x752eecce7100>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x752eecb18360>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x752eecb1aac0>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x752eecb184a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x752eecb184a0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x752eecb185e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x752eecb185e0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x752eecb18720>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x752eecce6c00>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x752eecb18860>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x752eecb189a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x752eecb189a0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x752eecce6700>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x752eecb18ae0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x752eecb18ae0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x752eecb18c20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x752eecb18d60>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x752eecb1a840>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x752eecb18ea0>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x752eecb1ba60>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x752eecce63e0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x752eecce5080>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x752eecb18fe0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x752eecb19120>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x752eecce5080>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x752eecb19260>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x752eecb19080>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x752eecb18e00>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x752eecb18b80>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x752eecce6e80>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x752eecb18900>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x752eecce7380>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x752eecb18680>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x752eecb18400>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x752eecb18400>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x752eecb18180>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x752eecce7d80>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x752eecb19620>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x752eecb19760>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x752eecb19760>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x752eecb198a0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x752eecb199e0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x752eecb19b20>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x752eecb19b20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x752eecb1aa20>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x752eecb19c60>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x752eecd22340>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x752eecd207c0>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x752eecd207c0>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x752eecd22c00>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x752eecd22c00>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x752eecd236a0>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x752eecd23a60>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x752eecd23a60>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x752eecd23d80>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x752eecd23d80>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x752eecd23f60>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x752eecd23f60>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x752eecd347c0>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x752eecd347c0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x752eecd23600>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x752eecef6480>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x752eecd34c20>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd34fe0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd34fe0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x752eecd35e40>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x752eecd35800>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x752eecd35800>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x752eecd35e40>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x752eecd35c60>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x752eecd35bc0>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x752eecd35bc0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x752eecef6480>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x752eecd23600>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37a60>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x752eecd36fc0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x752eecd36d40>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x752eecd36d40>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37ce0>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x752eecd376a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x752eecd376a0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x752eecd374c0>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x752eecd374c0>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37f60>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37740>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37880>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37ec0>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd379c0>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37b00>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37c40>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37c40>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37d80>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd377e0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd377e0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36f20>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36ca0>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36ca0>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36980>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36980>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd363e0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd363e0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36020>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd700e0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70220>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70360>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd704a0>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x752eecd704a0>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70720>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70860>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd709a0>, <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x752eecdd84a0>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x752eecdd9e40>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x752eecdd9e40>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x752eecddb880>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x752eecddb880>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x752eecdec2c0>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x752eecdec2c0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x752eecdeccc0>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x752eecd734c0>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x752eecd734c0>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x752eecdecfe0>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x752eecdecfe0>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x752eecded4e0>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x752eecded4e0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x752eecdedee0>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x752eecdedee0>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x752eecdee480>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x752eecdee480>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x752eece09440>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x752eece09440>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x752eece09940>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x752eece09940>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x752eece0ad40>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x752eece0ad40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x752eecea2480>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x752eecea18a0>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x752eecea2e80>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x752eecea1a80>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37a60>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x752eecea2e80>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x752eececa700>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x752eecd20f40>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x752eece78860>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x752eed253560>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x752eecc23ec0>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x752eecd236a0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x752eecef6160>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x752eecea23e0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x752eecef6160>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x752eecea1da0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x752eecea3f60>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x752eecc23ec0>, <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x752eecce63e0>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x752eece79da0>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x752eece784a0>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x752eece784a0>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x752eece78860>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x752eece78400>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x752eece78400>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x752eece79440>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x752eece79440>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x752eece7b6a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x752eece793a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x752eece793a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x752eece7b6a0>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x752eece78040>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x752eece78040>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x752eece799e0>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x752eece799e0>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x752eece79d00>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x752eece79d00>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x752eece79da0>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x752eece7a0c0>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x752eece7a0c0>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x752eece7a480>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x752eece7a480>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x752eece7aca0>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x752eece7aca0>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x752eece7ad40>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x752eece7b060>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x752eece7b100>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x752eece7b600>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x752eece7b600>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x752eecea00e0>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x752eece79580>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x752eece79580>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x752eece7bec0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x752eece78d60>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x752eece78d60>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x752eece7b9c0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x752eece7be20>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x752eece7be20>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x752eecea02c0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x752eecea04a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x752eece7bd80>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x752eece7bd80>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x752eecea0720>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x752eecea0720>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x752eecea0ae0>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x752eecea0ae0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x752eecea0e00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x752eecea13a0>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x752eecea0c20>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x752eecea0c20>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x752eecea0900>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x752eecea0900>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x752eecea18a0>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x752eecea1b20>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x752eecea1b20>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x752eecea32e0>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x752eecea2700>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x752eecea2700>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x752eecea2840>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x752eecea2840>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x752eecea23e0>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x752eecea31a0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x752eecea31a0>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x752eecea1a80>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x752eecea3240>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x752eecea3240>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x752eecea3560>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x752eecea3ba0>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x752eecea3ba0>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x752eecea3f60>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x752eecec8220>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x752eecec84a0>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x752eecec84a0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x752eecec8a40>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x752eecec8a40>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x752eecec8720>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x752eecec8720>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x752eecec8ae0>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x752eecec8ae0>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x752eecec8c20>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x752eecec8d60>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x752eecea3ce0>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x752eecec9300>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x752eecec93a0>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x752eecec8ea0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x752eecec8ea0>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x752eecec9620>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x752eecec9260>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x752eecec9260>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x752eecec9940>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x752eecec96c0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x752eececb6a0>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x752eececa520>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x752eececa160>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x752eececa160>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x752eececa5c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x752eececa660>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x752eececa840>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x752eececaac0>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x752eececa660>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x752eececab60>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x752eececaca0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x752eecef6520>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x752eececb880>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x752eecec9580>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x752eecec9580>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x752eececb920>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x752eececba60>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x752eececb6a0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x752eecef51c0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x752eececbc40>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x752eececbc40>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x752eececbce0>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x752eececbce0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x752eecef4900>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x752eecef4900>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x752eecef4d60>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x752eecd222a0>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x752eecef6520>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x752eecec9b20>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x752eecec9b20>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x752eecec8180>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x752eecef7600>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x752eecef5120>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x752eecef5120>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x752eecef5da0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x752eecef5da0>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x752eecef68e0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x752eecef5f80>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x752eecef7100>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x752eececb9c0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x752eecef4ae0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x752eecef7600>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x752eecef7100>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x752eecef6ca0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x752eecef5620>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x752eececb9c0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x752eecef5620>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x752eecef4ae0>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x752eecef7d80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x752eecef7e20>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x752eecd20040>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x752eecd20220>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x752eecd20400>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x752eecef4cc0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x752eecd20860>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x752eecd20a40>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x752eecd20d60>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x752eecd21120>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x752eecd21300>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x752eecd219e0>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x752eecd219e0>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x752eecd21f80>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x752eecd21d00>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x752eecd21800>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x752eecd21d00>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x752eecd21f80>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x752eecc356c0>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x752eecc356c0>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x752eecc35b20>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x752eecc35b20>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x752eecb18e00>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x752eecc35f80>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x752eecc35f80>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x752eecc485e0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x752eecc485e0>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x752eecc48a40>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x752eecc48a40>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x752eeccb9760>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x752eecce6e80>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x752eecc49760>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x752eecc49760>, <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x752eecef68e0>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x752eecd36c00>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x752eecc934c0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x752eecc934c0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x752eecc76660>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x752eecdeccc0>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x752eecc74d60>, <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x752eecd36ac0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x752eecea13a0>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x752eecec8220>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x752eecd36ac0>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x752eecc747c0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x752eecc747c0>, <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x752eeccb9760>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='dim')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='default')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x752eecc74d60>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x752eecc74c20>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x752eecc74c20>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x752eecef4d60>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x752eecb1b740>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x752eecc4bf60>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x752eecdd9b20>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x752eecdd9b20>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x752eecdef7e0>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x752eecdd8b80>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x752eecdda340>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x752eecdda340>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x752eece0be20>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x752eecdda840>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x752eecddb240>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x752eecdda840>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x752eecddad40>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x752eecddad40>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x752eecddb240>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x752eece78ea0>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x752eecddbd80>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x752eecddbd80>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x752eecea0e00>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x752eecdee3e0>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x752eecdee8e0>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x752eecdee8e0>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x752eecdeede0>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x752eecdeede0>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x752eecb18b80>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x752eecdef2e0>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x752eecdef2e0>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x752eecdef7e0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x752eecc76c00>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x752eeccba340>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x752eece08680>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x752eece08680>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x752eecdefec0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x752eecdefec0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x752eece08f40>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x752eece08f40>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x752eece09e40>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x752eece09e40>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x752eece0a340>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x752eece0a340>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x752eece0a840>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x752eece0a840>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x752eece0a3e0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x752eece0a3e0>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x752eece0b060>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x752eece0b060>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x752eecd36c00>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x752eece0b920>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x752eece0b920>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x752eece0be20>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x752eecc20360>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x752eecc20360>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x752eecc20cc0>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x752eecc20cc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x752eecc236a0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x752eecc236a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x752eecc23b00>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x752eed253560>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x752eecc23b00>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x752eecc227a0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x752eecc227a0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x752eecd70180>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x752eecd70180>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x752eecd71b20>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x752eecc4bf60>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x752eecd72a20>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x752eecd72a20>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x752eecd72f20>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x752eecd72f20>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x752eecd73420>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x752eecd73920>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x752eecd73920>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x752eecd73e20>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x752eecd73e20>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x752eecdd9120>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x752eecdd9120>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x752eecdd9620>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x752eecdd9620>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x752eecdd8860>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x752eecdd8c20>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x752eecdee3e0>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x752eecdd8360>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x752eecdd8360>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x752eecdd8860>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x752eecdd8b80>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x752eecdd8c20>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x752eecef68e0>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x752eece7a840>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x752eecec9800>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x752eece7ad40>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x752eecd36840>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x752eece78f40>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x752eecc92a20>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x752eecc90a40>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x752eecd73420>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x752eecd21800>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x752eecd71b20>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x752eecc93e20>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x752eecc76660>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x752eecc936a0>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x752eeccba340>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x752eeccb91c0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x752eeccb91c0>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x752eeccba700>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x752eeccba700>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x752eeccba480>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x752eeccba480>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x752eeccb8a40>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x752eeccb8a40>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x752eeccb8cc0>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x752eeccb8cc0>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x752eeccb8e00>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x752eeccb8e00>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x752eecc92e80>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x752eecc92e80>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x752eeccb8c20>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x752eeccb8c20>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <OpOverload(op='aten.item', overload='default')>: <function item at 0x752eecc4b7e0>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x752eecea2200>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x752eecea2200>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd372e0>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x752eecce6480>, <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x752eecef5ee0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x752eecc349a0>}
- 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 0x752eecec8720>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x752eecec9620>}
- 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.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b420>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e20946fc0>, kernel=<OpOverload(op='aten.is_inference', 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 0x752e17e43ce0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42ac0>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded940>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19580>, kernel=<OpOverload(op='aten.inverse', overload='default')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df53a0>, kernel=<OpOverload(op='aten.is_leaf', overload='default')>), <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24360>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f220c0>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4860>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5620>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.special_erfc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b2e0>, kernel=<OpOverload(op='aten.special_erfc', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c244a0>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deeb60>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>), <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df58a0>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18cc0>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4680>, kernel=<OpOverload(op='aten._sparse_softmax', 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 0x752e17deec00>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e402c0>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee660>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a700>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.max_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defa60>, kernel=<OpOverload(op='aten.max_pool3d', overload='default')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18220>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defb00>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec180>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6c00>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee5c0>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40ea0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.grid_sampler', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def9c0>, kernel=<OpOverload(op='aten.grid_sampler', overload='default')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4e00>, 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 0x752e17e40720>, 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 0x752e17decd60>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21b20>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee0c0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e187c0>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dedb20>, kernel=<OpOverload(op='aten.trapz', overload='x')>), <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18ae0>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec680>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22520>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>), <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22660>, 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 0x752e17f214e0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e49940>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f091c0>, kernel=<OpOverload(op='aten.index_select_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 0x752e17f21da0>, 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 0x752e17e48900>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df40e0>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23600>, kernel=<OpOverload(op='aten.special_round', overload='default')>), <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e23bd65c0>, kernel=<OpOverload(op='aten.msort', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23c40>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6520>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23ec0>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22160>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6020>, kernel=<OpOverload(op='_test.cat', 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 0x752e17e1a340>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>), <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df63e0>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1ae80>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23e20>, 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 0x752e17f23920>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21bc0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b600>, kernel=<OpOverload(op='aten.special_digamma', 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 0x752e17f094e0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b560>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22840>, kernel=<OpOverload(op='aten.diagflat', overload='default')>), <OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f222a0>, 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 0x752e17e43b00>, 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 0x752e17f21800>, 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 0x752e17f23100>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4900>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23d80>, kernel=<OpOverload(op='aten.is_conj', overload='default')>), <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e20911bc0>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7600>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21120>, kernel=<OpOverload(op='aten._weight_norm', 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 0x752e17f08ea0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <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 0x752e23bd7560>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5940>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0a840>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e439c0>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5080>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23560>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0a340>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f207c0>, kernel=<OpOverload(op='aten.one_hot', overload='default')>), <OpOverload(op='aten.special_expm1', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20d60>, kernel=<OpOverload(op='aten.special_expm1', overload='default')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43ec0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e49800>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21e40>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee020>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee340>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21440>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e496c0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e48ea0>, kernel=<OpOverload(op='aten.chalf', 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 0x752e17e43c40>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f234c0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4360>, kernel=<OpOverload(op='aten.argsort', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20860>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.atleast_2d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e209467a0>, kernel=<OpOverload(op='aten.atleast_2d', overload='Sequence')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec4a0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e185e0>, 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 0x752e17e19940>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e419e0>, kernel=<OpOverload(op='aten.orgqr', overload='default')>), <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18ea0>, 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 0x752e17df5260>, 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 0x752e17dee7a0>, 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 0x752e17e423e0>, kernel=<OpOverload(op='aten.pinverse', overload='default')>), <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dedbc0>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5440>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b240>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40400>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41e40>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6e80>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42840>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41ee0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4cc0>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e437e0>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deea20>, kernel=<OpOverload(op='aten.is_neg', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee3e0>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19bc0>, 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 0x752e17df7b00>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18b80>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41c60>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42b60>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e182c0>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5760>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>), <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a2a0>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>), <OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43d80>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', 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 0x752e17e1a980>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17decae0>, kernel=<OpOverload(op='aten.special_i0', overload='default')>), <OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec360>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>), <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded8a0>, kernel=<OpOverload(op='aten.rnn_tanh_cell', 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 0x752e17e1bb00>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40220>, kernel=<OpOverload(op='aten.max_pool1d', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43600>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6d40>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b6a0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18180>, kernel=<OpOverload(op='aten.max', overload='names_dim')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19760>, kernel=<OpOverload(op='aten.special_erfinv', overload='default')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dedda0>, kernel=<OpOverload(op='aten.flipud', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a660>, 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 0x752e17e427a0>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b880>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23a60>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='aten.layer_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec400>, kernel=<OpOverload(op='aten.layer_norm', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df77e0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>), <OpOverload(op='aten.atleast_3d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded120>, kernel=<OpOverload(op='aten.atleast_3d', overload='Sequence')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0ad40>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded080>, kernel=<OpOverload(op='aten.pad', overload='default')>), <OpOverload(op='aten.instance_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e184a0>, kernel=<OpOverload(op='aten.instance_norm', overload='default')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec7c0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7e20>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.isclose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded9e0>, kernel=<OpOverload(op='aten.isclose', overload='default')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e413a0>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f08cc0>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0aca0>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>), <OpOverload(op='aten.conv1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded620>, kernel=<OpOverload(op='aten.conv1d', overload='default')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee520>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dedf80>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def100>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.less', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deede0>, kernel=<OpOverload(op='aten.less', overload='Tensor')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defce0>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.greater', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deff60>, kernel=<OpOverload(op='aten.greater', overload='Tensor')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0a5c0>, kernel=<OpOverload(op='aten.poisson_nll_loss', 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 0x752e17def920>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def880>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>), <OpOverload(op='aten.tile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df49a0>, kernel=<OpOverload(op='aten.tile', overload='default')>), <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41bc0>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22700>, kernel=<OpOverload(op='aten.conv3d', overload='default')>), <OpOverload(op='aten.special_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43920>, kernel=<OpOverload(op='aten.special_softmax', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41f80>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18720>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42d40>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7060>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19b20>, 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 0x752e17e19e40>, kernel=<OpOverload(op='aten.align_as', overload='default')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f08e00>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten.special_log_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19da0>, kernel=<OpOverload(op='aten.special_log_softmax', 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 0x752e17e1b9c0>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1bce0>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7380>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>), <OpOverload(op='aten.log_sigmoid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee200>, kernel=<OpOverload(op='aten.log_sigmoid', overload='default')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a480>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.cross_entropy_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20720>, kernel=<OpOverload(op='aten.cross_entropy_loss', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded760>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19440>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19120>, kernel=<OpOverload(op='aten.linalg_multi_dot', 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 0x752e17defe20>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e199e0>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a8e0>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6480>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec5e0>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec220>, 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 0x752e17dec040>, kernel=<OpOverload(op='aten.max_pool2d', overload='default')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded1c0>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='aten._debug_has_internal_overlap', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17decfe0>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17decb80>, 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 0x752e17decf40>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7ec0>, 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 0x752e17dede40>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded800>, 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 0x752e17dee700>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e436a0>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten.subtract', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def240>, kernel=<OpOverload(op='aten.subtract', overload='Tensor')>), <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def060>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24040>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24720>, 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 0x752e17df54e0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c240e0>, 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 0x752e17df4a40>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df47c0>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.special_sinc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4720>, 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 0x752e17df5580>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4ea0>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18900>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42700>, 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 0x752e17e407c0>, 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 0x752e17df6ca0>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6a20>, kernel=<OpOverload(op='image.decode_png', overload='default')>), <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40860>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40fe0>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>), <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43740>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42340>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40f40>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten.einsum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e434c0>, kernel=<OpOverload(op='aten.einsum', overload='default')>), <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42fc0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e23bd5120>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40a40>, 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 0x752e17e43100>, kernel=<OpOverload(op='aten.divide', overload='Scalar')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41580>, kernel=<OpOverload(op='aten.slow_conv3d', 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 0x752e17e18360>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a160>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1bc40>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1ab60>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>), <OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a7a0>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1ad40>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18e00>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18fe0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7740>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e196c0>, kernel=<OpOverload(op='aten.histogramdd', 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 0x752e17e1a840>, kernel=<OpOverload(op='aten.scaled_dot_product_attention', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20f40>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21300>, 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 0x752e17f21080>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20b80>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09940>, 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 0x752e17f09f80>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20400>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f204a0>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0ac00>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0aa20>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09bc0>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09d00>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e209465c0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', 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 0x752e17e48180>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e20946980>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e209477e0>, kernel=<OpOverload(op='aten._convolution_mode', 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 0x752e17e4b6a0>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e482c0>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e49120>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43ba0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42520>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f232e0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', 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 0x752e17df5b20>, 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 0x752e17df5da0>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23f60>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f231a0>, 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 0x752e17f22d40>, 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 0x752e17f22200>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f200e0>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22fc0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded300>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1bd80>, kernel=<OpOverload(op='aten.lstm_cell', 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 0x752e17e1a020>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', 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 0x752e17df65c0>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>), <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20e00>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6ac0>, kernel=<OpOverload(op='c10d_functional.broadcast', 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 0x752e17e1afc0>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4f40>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df42c0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee840>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18c20>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee980>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21f80>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4540>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23740>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22ac0>, 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 0x752e17f22c00>, 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 0x752e17f21940>, 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 0x752e17f21a80>, 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 0x752e17df6200>, 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 0x752e17df5f80>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defd80>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f227a0>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e422a0>, 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 0x752e17e41620>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten.embedding_bag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def7e0>, kernel=<OpOverload(op='aten.embedding_bag', overload='default')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41800>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42ca0>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df56c0>, kernel=<OpOverload(op='aten.bilinear', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e416c0>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.embedding_bag', overload='padding_idx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e49c60>, kernel=<OpOverload(op='aten.embedding_bag', overload='padding_idx')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e48720>, 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 0x752e209479c0>, 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 0x752e17df6840>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e4b240>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def1a0>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten.divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e49080>, 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 0x752e23bd7740>, 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 0x752e23bd5580>, kernel=<OpOverload(op='aten.special_psi', overload='default')>), <OpOverload(op='aten.is_nonzero', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded4e0>, kernel=<OpOverload(op='aten.is_nonzero', overload='default')>), <OpOverload(op='c10d_functional.all_reduce', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6660>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e491c0>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e4b740>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20900>, 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 0x752e17f20a40>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6160>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f225c0>, 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 0x752e17f20220>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20fe0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41940>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f218a0>, 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 0x752e17f22020>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09da0>, kernel=<OpOverload(op='aten._version', 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 0x752e17def600>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec860>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f08d60>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18f40>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b920>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>), <OpOverload(op='aten.type_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f08ae0>, kernel=<OpOverload(op='aten.type_as', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41d00>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f211c0>, 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 0x752e17f09580>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.greater_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20ea0>, kernel=<OpOverload(op='aten.greater_equal', overload='Tensor')>), <OpOverload(op='aten.atleast_1d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22e80>, kernel=<OpOverload(op='aten.atleast_1d', overload='Sequence')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e23bd4fe0>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f213a0>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0b600>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e48e00>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e23bd5b20>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09080>, kernel=<OpOverload(op='aten.linalg_tensorinv', 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 0x752e17f08c20>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dedc60>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41440>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def4c0>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e428e0>, 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 0x752e17e41a80>, 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 0x752e17decc20>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23380>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.concat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1bec0>, kernel=<OpOverload(op='aten.concat', overload='default')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df76a0>, kernel=<OpOverload(op='image.decode_webp', overload='default')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e498a0>, kernel=<OpOverload(op='aten.argwhere', 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 0x752e17e18400>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40c20>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09e40>, kernel=<OpOverload(op='aten.nanquantile', 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 0x752e17e420c0>, 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 0x752e17df6f20>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded3a0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41b20>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e191c0>, 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 0x752e17deca40>, kernel=<OpOverload(op='aten.logdet', overload='default')>), <OpOverload(op='aten.subtract', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1bba0>, 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 0x752e17dedd00>, 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 0x752e17dee160>, 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 0x752e17df4c20>, kernel=<OpOverload(op='aten.linalg_det', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee480>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1bf60>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df51c0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7420>, kernel=<OpOverload(op='image.encode_png', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40900>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18040>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43380>, 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 0x752e17deefc0>, 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 0x752e17e42e80>, 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 0x752e17deed40>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e189a0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18540>, 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 0x752e17e1b4c0>, 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 0x752e17dece00>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='aten.log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23880>, kernel=<OpOverload(op='aten.log_softmax', overload='int')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43f60>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e418a0>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4b80>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19260>, kernel=<OpOverload(op='aten.fix', 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 0x752e17df5800>, 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 0x752e17e432e0>, 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 0x752e17e19c60>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e20946520>, kernel=<OpOverload(op='aten.rnn_relu_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 0x752e17e19ee0>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a200>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42160>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4040>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', 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 0x752e17df60c0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e180e0>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20c20>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f223e0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.max_pool1d_with_indices', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41300>, kernel=<OpOverload(op='aten.max_pool1d_with_indices', overload='default')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21620>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.concatenate', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1ade0>, 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 0x752e17e19620>, kernel=<OpOverload(op='aten.true_divide', overload='Scalar')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40ae0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19080>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1aa20>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43a60>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='aten.det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded440>, kernel=<OpOverload(op='aten.det', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1af20>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1aca0>, 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 0x752e17def380>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defba0>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4fe0>, kernel=<OpOverload(op='aten.data', overload='default')>), <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec9a0>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a520>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e411c0>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec2c0>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deef20>, 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 0x752e17dedee0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>), <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42980>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>), <OpOverload(op='aten.softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec900>, kernel=<OpOverload(op='aten.softmax', overload='int')>), <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7560>, kernel=<OpOverload(op='image.decode_gif', overload='default')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1aac0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41da0>, 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 0x752e17e193a0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18d60>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a0c0>, 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 0x752e17e1ba60>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b380>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.multiply', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5120>, kernel=<OpOverload(op='aten.multiply', overload='Tensor')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17decea0>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e425c0>, kernel=<OpOverload(op='aten._backward', overload='default')>), <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b100>, kernel=<OpOverload(op='aten._reshape_from_tensor', 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 0x752e17f0a660>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df45e0>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4ae0>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>), <OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def6a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec540>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e198a0>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42de0>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def740>, kernel=<OpOverload(op='aten._test_check_tensor', 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 0x752e17e49f80>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', 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 0x752e17f21ee0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defc40>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.sym_size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deeac0>, kernel=<OpOverload(op='aten.sym_size', overload='int')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17defec0>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.less_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b7e0>, kernel=<OpOverload(op='aten.less_equal', overload='Tensor')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23420>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22b60>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e414e0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.true_divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f237e0>, kernel=<OpOverload(op='aten.true_divide', overload='Tensor')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22ca0>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22480>, kernel=<OpOverload(op='aten.diag', overload='default')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e209105e0>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x752eecb19da0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x752eecb19ee0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x752eecb1a020>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x752eecb1a160>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x752eecb1a2a0>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x752eecb1a3e0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x752eecb19e40>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x752eecb1a520>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x752eecb1a660>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x752eecb1a340>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x752eecb1a7a0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x752eecb1a7a0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x752eecb1a0c0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x752eecb1a8e0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x752eecb19940>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x752eecb19940>, <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 0x752eecb18540>, <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 0x752eecb18540>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x752eecb191c0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x752eecb191c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x752eecb1ade0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x752eecb1ade0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x752eecb1b100>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x752eecb1b100>, <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 0x752eecce7880>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <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 0x752eecb1b420>, <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 0x752eecb1b420>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x752eecce7880>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x752eecb1b740>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x752eecb1ba60>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x752eecb1be20>, <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 0x752eecb1be20>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x752eecb1ae80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x752eecb1ae80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <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 0x752eecb19d00>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x752eecb55120>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x752eecb55120>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x752eecb553a0>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x752eecb553a0>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df59e0>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x752eecb55e40>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5300>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x752eecb1ac00>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x752eecb57100>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e40b80>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x752eecb55300>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19a80>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e404a0>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x752eecb56160>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42660>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x752eecb563e0>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42200>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x752eecb56660>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43880>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x752eecb56f20>, <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43560>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x752eecb571a0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42020>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x752eecb57420>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18a40>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x752eecb576a0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19300>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x752eecb57d80>, <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a3e0>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x752eecb57f60>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18680>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x752eecb56020>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b060>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x752eecb780e0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1a5c0>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x752eecb78360>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1ac00>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x752eecb785e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1b740>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x752eecb78860>, <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e194e0>, 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 0x752e17e19800>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x752eecb79080>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x752eecb79080>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x752eecb79440>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x752eecb79440>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x752eecb7a3e0>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x752eecb7a660>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x752eecb7a660>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x752eecb7ad40>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x752eecb7ad40>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x752eecb7b1a0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x752eecb7b1a0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x752eecb7b9c0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x752eecb7b7e0>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x752eecb7b7e0>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x752eecb7b9c0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x752eecb7bd80>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x752eecb7bd80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x752eecbb4b80>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x752eecbb65c0>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x752eecbb65c0>, <OpOverload(op='aten.selu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e499e0>, kernel=<OpOverload(op='aten.selu', overload='default')>), <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x752eecbb4ea0>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x752eecbb44a0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x752eecbb44a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x752eecbb4b80>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x752eecbb4ea0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42480>, 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 0x752e17e40540>, 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 0x752e17f23240>, kernel=<OpOverload(op='aten.nll_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x752eecbb5ee0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x752eecbb62a0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x752eecbb62a0>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x752eecbb6200>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x752eecbb6200>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x752eecbb6de0>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x752eecbb6de0>, <OpOverload(op='aten.prelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22f20>, 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 0x752e17f23060>, kernel=<OpOverload(op='aten.relu6', overload='default')>), <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x752eecbb7920>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x752eecbb7920>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21760>, 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 0x752e17f23b00>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x752eecbb6ca0>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x752eecbb6ca0>, <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x752eecbcc220>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x752eecbcc220>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x752eecbcc4a0>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x752eecbcc4a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x752eecbb6f20>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x752eecbcc2c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x752eecbcc540>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x752eecbcc680>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x752eecbcc7c0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x752eecbccea0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x752eecbccea0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x752eecbcd260>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x752eecbcd260>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x752eecbcd620>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x752eecbcd620>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x752eecbcdbc0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x752eecbcdbc0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x752eecbce0c0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x752eecbce0c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x752eecbce5c0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x752eecbce5c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x752eecbce980>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x752eecbce980>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x752eecbced40>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x752eecbced40>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x752eecbce160>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x752eecbce160>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e20947b00>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x752eecbcf240>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x752eecbcf600>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x752eecbcf600>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x752eecbcfc40>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x752eecbcfc40>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x752eecc376a0>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x752eecc35260>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x752eecc35260>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x752eecc376a0>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x752eecc36520>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x752eecc36520>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x752eecc36de0>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x752eecc36de0>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x752eecc37240>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x752eecc37240>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x752eecc36020>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x752eecc36020>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x752eecc37880>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x752eecc37880>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x752eecc37ce0>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x752eecc37ce0>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x752eecc4aa20>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x752eecc4aa20>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x752eecc4ade0>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x752eecc4ade0>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def2e0>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def420>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deccc0>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17def560>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee2a0>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded260>, 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 0x752e17deda80>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded6c0>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17ded580>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dee8e0>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x752eecc77920>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x752eecc767a0>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17deee80>, 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 0x752e17deeca0>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4180>, 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 0x752e17df4220>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x752eecc760c0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x752eecc760c0>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x752eecc771a0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x752eecc771a0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x752eecc77ce0>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x752eecc77380>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x752eecc77560>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x752eecc765c0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x752eecc90540>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x752eecc779c0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x752eecc779c0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x752eecc77ec0>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x752eecc77ec0>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x752eecc75620>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x752eecc75620>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x752eecc92840>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x752eecc90900>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x752eecc90900>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x752eecc90f40>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x752eecc90f40>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x752eecc90ea0>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x752eecc911c0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x752eecc911c0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x752eecc914e0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x752eecc914e0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x752eecc91b20>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x752eecc91b20>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e1be20>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e19d00>, 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 0x752e17e19f80>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x752eecc93240>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x752eecc92a20>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x752eecc90a40>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x752eecc92d40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x752eecc92d40>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x752eeccb9440>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x752eecc93100>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x752eecc93100>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x752eecc93420>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x752eecc93600>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x752eecc93a60>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x752eecc93a60>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x752eeccb9440>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec720>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17dec0e0>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x752eecc93ce0>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x752eeccba980>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x752eeccbbf60>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x752eeccbbf60>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4400>, 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 0x752e17df44a0>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x752eeccbae80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x752eeccbae80>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x752eeccbab60>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x752eeccbab60>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42f20>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e42a20>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x752eeccbb4c0>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x752eeccbb4c0>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x752eeccbb9c0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x752eeccbb9c0>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x752eeccbbc40>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x752eeccbbc40>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x752eecce4220>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x752eecce4220>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x752eecce45e0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x752eecce45e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x752eecce49a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x752eecce49a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x752eecce4d60>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x752eecce4d60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x752eecce4e00>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x752eecce5760>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x752eecce5760>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x752eecce5c60>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x752eecce5c60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x752eecce4cc0>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x752eecce4cc0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x752eecce7ec0>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x752eecce6200>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x752eecce6200>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x752eecce6480>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x752eeccbb1a0>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x752eecce7e20>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x752eecce62a0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x752eecce7e20>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x752eecce6520>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x752eecce6660>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x752eecce6660>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x752eecce67a0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x752eecce68e0>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x752eecce7ce0>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x752eecce6a20>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x752eecb1a5c0>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x752eecce6b60>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x752eecce6ca0>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x752eecce6de0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x752eecce6f20>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x752eecce7ba0>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x752eecce7060>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x752eecce7060>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x752eecce71a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x752eecce71a0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x752eecce7240>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x752eecce72e0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x752eecb1ab60>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x752eecce7420>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x752eecce7420>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x752eecce74c0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x752eecce7560>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x752eecce7560>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x752eecce76a0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x752eecce76a0>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x752eecce7740>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x752eecb19300>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x752eecce77e0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x752eecce7740>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x752eecce7c40>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x752eecce7920>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x752eecce7920>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x752eecce7a60>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x752eecce7a60>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x752eecce79c0>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x752eecce7ba0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x752eecce79c0>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x752eecb19580>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x752eecce6d40>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x752eecce6d40>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x752eecce6ac0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x752eecb193a0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x752eecce6840>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x752eecce65c0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x752eecce7b00>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x752eecce6160>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x752eecb1aac0>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x752eecce7f60>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x752eecb180e0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x752eecce7600>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x752eecb18220>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x752eecb18220>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x752eecce7100>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x752eecb18360>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x752eecb1aac0>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x752eecb184a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x752eecb184a0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x752eecb185e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x752eecb185e0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x752eecb18720>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x752eecce6c00>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x752eecb18860>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x752eecb189a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x752eecb189a0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x752eecce6700>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x752eecb18ae0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x752eecb18ae0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x752eecb18c20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x752eecb18d60>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x752eecb1a840>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x752eecb18ea0>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x752eecb1ba60>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x752eecce63e0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x752eecce5080>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x752eecb18fe0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x752eecb19120>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x752eecce5080>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x752eecb19260>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x752eecb19080>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x752eecb18e00>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x752eecb18b80>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x752eecce6e80>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x752eecb18900>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x752eecce7380>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x752eecb18680>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x752eecb18400>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x752eecb18400>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x752eecb18180>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x752eecce7d80>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x752eecb19620>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x752eecb19760>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x752eecb19760>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x752eecb198a0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x752eecb199e0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x752eecb19b20>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x752eecb19b20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x752eecb1aa20>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x752eecb19c60>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x752eecd22340>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x752eecd207c0>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x752eecd207c0>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x752eecd22c00>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x752eecd22c00>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x752eecd236a0>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x752eecd23a60>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x752eecd23a60>, <OpOverload(op='aten.matmul', overload='default')>: <function matmul at 0x752eecd34400>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x752eecd23d80>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x752eecd23d80>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x752eecd23f60>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x752eecd23f60>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x752eecd347c0>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x752eecd347c0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x752eecd23600>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x752eecef6480>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x752eecd34c20>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd34fe0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd34fe0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x752eecd35e40>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x752eecd35800>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x752eecd35800>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x752eecd35e40>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x752eecd35c60>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x752eecd35bc0>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x752eecd35bc0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x752eecef6480>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x752eecd23600>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37a60>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x752eecd36fc0>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x752eecd36d40>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x752eecd36d40>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37ce0>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x752eecd376a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x752eecd376a0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x752eecd374c0>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x752eecd374c0>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37f60>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37740>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37880>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37ec0>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd379c0>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37b00>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37c40>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37c40>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37d80>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd377e0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd377e0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36f20>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36ca0>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36ca0>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36980>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36980>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd363e0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd363e0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36020>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd700e0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70220>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70360>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd704a0>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x752eecd704a0>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70720>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70860>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd709a0>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e41760>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x752eecdd9e40>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x752eecdd9e40>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x752eecddb880>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x752eecddb880>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x752eecdec2c0>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x752eecdec2c0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x752eecdeccc0>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x752eecd734c0>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x752eecd734c0>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x752eecdecfe0>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x752eecdecfe0>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x752eecded4e0>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x752eecded4e0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x752eecdedee0>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x752eecdedee0>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f23ce0>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x752eecdee480>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x752eecdee480>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x752eece09440>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x752eece09440>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x752eece09940>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x752eece09940>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x752eece0ad40>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x752eece0ad40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x752eecea2480>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x752eecea18a0>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x752eecea2e80>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x752eecea1a80>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37a60>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x752eecea2e80>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x752eececa700>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x752eecd20f40>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x752eecea1da0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x752eece78860>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x752eed253560>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x752eecc23ec0>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0b1a0>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x752eecd236a0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x752eecef6160>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x752eecea23e0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x752eecef6160>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x752eecea1da0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x752eecea3f60>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x752eecc23ec0>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e18860>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x752eecce63e0>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x752eece79da0>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x752eece784a0>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x752eece784a0>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x752eece78860>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x752eece78400>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x752eece78400>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x752eece79440>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x752eece79440>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x752eece7b6a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x752eece793a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x752eece793a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x752eece7b6a0>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x752eece78040>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x752eece78040>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x752eece799e0>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x752eece799e0>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x752eece79d00>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x752eece79d00>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x752eece79da0>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x752eece7a0c0>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x752eece7a0c0>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x752eece7a480>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x752eece7a480>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x752eece7aca0>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x752eece7aca0>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x752eece7ad40>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x752eece7b060>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x752eece7b100>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x752eece7b600>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x752eece7b600>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x752eecea00e0>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x752eece79580>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x752eece79580>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x752eece7bec0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x752eece78d60>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x752eece78d60>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x752eece7b9c0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x752eece7be20>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x752eece7be20>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x752eecea02c0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x752eecea04a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x752eece7bd80>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x752eece7bd80>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x752eecea0720>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x752eecea0720>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x752eecea0ae0>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x752eecea0ae0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x752eecea0e00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x752eecea13a0>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x752eecea0c20>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x752eecea0c20>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x752eecea0900>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x752eecea0900>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x752eecea18a0>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x752eecea1b20>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x752eecea1b20>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x752eecea2480>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x752eecea32e0>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x752eecea2700>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x752eecea2700>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x752eecea2840>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x752eecea2840>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x752eecea23e0>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x752eecea31a0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x752eecea31a0>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x752eecea1a80>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x752eecea3240>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x752eecea3240>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x752eecea3560>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x752eecea3ba0>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x752eecea3ba0>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x752eecea3f60>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x752eecec8220>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x752eecec84a0>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x752eecec84a0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x752eecec8a40>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x752eecec8a40>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x752eecec8720>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x752eecec8720>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x752eecec8ae0>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x752eecec8ae0>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x752eecec8c20>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x752eecec8d60>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x752eecea3ce0>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x752eecec9300>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x752eecec93a0>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x752eecec8ea0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x752eecec8ea0>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x752eecec9620>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x752eecec9260>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x752eecec9260>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x752eecec9940>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x752eecec96c0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x752eececb6a0>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x752eececa520>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x752eececa160>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x752eececa160>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x752eececa5c0>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x752eececa660>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x752eececa840>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x752eececaac0>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x752eececa660>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x752eececab60>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x752eececaca0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x752eecef6520>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x752eececb880>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x752eecec9580>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x752eecec9580>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x752eececb920>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x752eececba60>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x752eececb6a0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x752eecef51c0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x752eececbc40>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x752eececbc40>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x752eececbce0>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x752eececbce0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x752eecef4900>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x752eecef4900>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x752eecef4d60>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x752eecd222a0>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x752eecef6520>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x752eecec9b20>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x752eecec9b20>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x752eecec8180>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x752eecef7600>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x752eecef5120>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x752eecef5120>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x752eecef5da0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x752eecef5da0>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x752eecef68e0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x752eecef5f80>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x752eecef7100>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x752eececb9c0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x752eecef4ae0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x752eecef7600>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x752eecef7100>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x752eecef6ca0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x752eecef5620>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x752eececb9c0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x752eecef5620>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x752eecef4ae0>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x752eecef7d80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x752eecef7e20>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x752eecd20040>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x752eecd20220>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x752eecd20400>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x752eecef4cc0>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x752eecd20860>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x752eecd20a40>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x752eecd20d60>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x752eecd21120>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x752eecd21300>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x752eecd219e0>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x752eecd219e0>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x752eecd21f80>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x752eecd21d00>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x752eecd21800>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x752eecd21d00>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x752eecd21f80>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x752eecc356c0>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x752eecc356c0>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x752eecc35b20>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x752eecc35b20>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x752eecb18e00>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x752eecc35f80>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x752eecc35f80>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x752eecc485e0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x752eecc485e0>, <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20680>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x752eecc48a40>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x752eecc48a40>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x752eeccb9760>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x752eecce6e80>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x752eecc49760>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x752eecc49760>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f219e0>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x752eecef68e0>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x752eecd36c00>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x752eecc934c0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x752eecc934c0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x752eecc76660>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e43420>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x752eecdeccc0>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x752eecc74d60>, <OpOverload(op='aten.where', overload='ScalarOther')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f20cc0>, 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 0x752e17f21d00>, kernel=<OpOverload(op='aten.where', overload='ScalarSelf')>), <OpOverload(op='aten.where', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09620>, kernel=<OpOverload(op='aten.where', overload='Scalar')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x752eecd21800>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x752eecd36ac0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x752eecea13a0>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x752eecec8220>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x752eecd36ac0>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x752eecc747c0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x752eecc747c0>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17e49a80>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22340>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21c60>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.full', overload='default')>: <function full at 0x752eeccb9760>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f236a0>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e209119e0>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.var', overload='out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f22980>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x752eecc74d60>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x752eecc74c20>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x752eecc74c20>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x752eecef4d60>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x752eecb1b740>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x752eecc4bf60>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x752eecdd9b20>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x752eecdd9b20>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x752eecdef7e0>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x752eecdd8b80>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x752eecdda340>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x752eecdda340>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x752eece0be20>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x752eecdda840>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x752eecddb240>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x752eecdda840>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x752eecddad40>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x752eecddad40>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x752eecddb240>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x752eece78ea0>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x752eecddbd80>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x752eecddbd80>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x752eecea0e00>, <OpOverload(op='aten.isfinite', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df4d60>, kernel=<OpOverload(op='aten.isfinite', overload='default')>), <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x752eecdee3e0>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x752eecdee8e0>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x752eecdee8e0>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x752eecdeede0>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x752eecdeede0>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x752eecb18b80>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x752eecdef2e0>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x752eecdef2e0>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x752eecdef7e0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x752eecc76c00>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x752eeccba340>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x752eece08680>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x752eece08680>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x752eecdefec0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x752eecdefec0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x752eece08f40>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x752eece08f40>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x752eece09e40>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x752eece09e40>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x752eece0a340>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x752eece0a340>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x752eece0a840>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x752eece0a840>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x752eece0a3e0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x752eece0a3e0>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x752eece0b060>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x752eece0b060>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x752eecd36c00>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x752eece0b920>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x752eece0b920>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x752eece0be20>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x752eecc20360>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x752eecc20360>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x752eecc20cc0>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x752eecc20cc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x752eecc236a0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x752eecc236a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x752eecc23b00>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x752eed253560>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x752eecc23b00>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x752eecc227a0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x752eecc227a0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x752eecd70180>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x752eecd70180>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x752eecd71b20>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x752eecc4bf60>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x752eecd72a20>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x752eecd72a20>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x752eecd72f20>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x752eecd72f20>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x752eecd73420>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x752eecd73920>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x752eecd73920>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x752eecd73e20>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x752eecd73e20>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x752eecdd9120>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x752eecdd9120>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x752eecdd9620>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x752eecdd9620>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x752eecdd8860>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x752eecdd8c20>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x752eecdee3e0>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x752eecdd8360>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x752eecdd8360>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x752eecdd8860>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x752eecdd8b80>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x752eecdd8c20>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f0a020>, 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 0x752e17f20540>, kernel=<OpOverload(op='aten.conv2d', overload='default')>), <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x752eecef68e0>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x752eece7a840>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x752eecec9800>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x752eece7ad40>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x752eecd36840>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x752eece78f40>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x752eecc92a20>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x752eecc90a40>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x752eecd73420>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x752eecd21800>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x752eecd71b20>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x752eecc93e20>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x752eecc76660>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x752eecc936a0>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x752eeccba340>, <OpOverload(op='aten.slice', overload='Tensor')>: <function slice_forward at 0x752eecea1bc0>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x752eeccb91c0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x752eeccb91c0>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x752eeccba700>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x752eeccba700>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x752eeccba480>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x752eeccba480>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x752eeccb8a40>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x752eeccb8a40>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x752eeccb8cc0>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x752eeccb8cc0>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x752eeccb8e00>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x752eeccb8e00>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x752eecc92e80>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x752eecc92e80>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x752eeccb8c20>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x752eeccb8c20>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f21260>, 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 0x752e17f20ae0>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x752eecea2200>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x752eecea2200>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd372e0>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x752eecce6480>, <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x752eecef5ee0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x752eecc349a0>, <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7ce0>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c245e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7c40>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5ee0>, 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 0x752e17df68e0>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6980>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df67a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7ba0>, kernel=<OpOverload(op='sparse.qlinear_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 0x752e17df5d00>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>), <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7880>, 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 0x752e17df6b60>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24540>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c242c0>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df79c0>, 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 0x752e17df7920>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7d80>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>), <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24680>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>), <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6de0>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7100>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7240>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df71a0>, 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 0x752e17df72e0>, 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 0x752e17df6700>, 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 0x752e17df5c60>, 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 0x752e17df5a80>, 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 0x752e17df5e40>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df74c0>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6340>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df62a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24180>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24400>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df7f60>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df6fc0>, 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 0x752e17df7a60>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c247c0>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c24220>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17df5bc0>, 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.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795080>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786660>, kernel=<OpOverload(op='aten.is_inference', 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 0x752e177902c0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790680>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a04a0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793600>, kernel=<OpOverload(op='aten.inverse', overload='default')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3ec0>, kernel=<OpOverload(op='aten.is_leaf', overload='default')>), <OpOverload(op='torchvision._cuda_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aae80>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784180>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3380>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8180>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aafc0>, kernel=<OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>), <OpOverload(op='aten._cast_Float', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a16c0>, kernel=<OpOverload(op='aten._cast_Float', overload='default')>), <OpOverload(op='aten._cast_Half', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8400>, kernel=<OpOverload(op='aten._cast_Half', overload='default')>), <OpOverload(op='aten.ctc_loss', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796980>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a31a0>, kernel=<OpOverload(op='aten._sparse_softmax', 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 0x752e177a1760>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790d60>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a11c0>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794f40>, 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 0x752e17796480>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2660>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796ca0>, kernel=<OpOverload(op='aten.log_softmax', overload='Dimname')>), <OpOverload(op='image._jpeg_version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9760>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1120>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790860>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3920>, 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 0x752e17797880>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f560>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0c20>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792f20>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0680>, kernel=<OpOverload(op='aten.trapz', overload='x')>), <OpOverload(op='aten.nanmean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792ca0>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177971a0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797fba0>, kernel=<OpOverload(op='aten.cummaxmin_backward', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f2e0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177874c0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785800>, kernel=<OpOverload(op='aten.index_select_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 0x752e1797e3e0>, 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 0x752e17786fc0>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2c00>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d120>, kernel=<OpOverload(op='aten.special_round', overload='default')>), <OpOverload(op='aten.msort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786520>, kernel=<OpOverload(op='aten.msort', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e660>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9080>, kernel=<OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>), <OpOverload(op='aten._cast_Double', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177840e0>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797fd80>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8b80>, kernel=<OpOverload(op='_test.cat', 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 0x752e177939c0>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>), <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8f40>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795800>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e8e0>, 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 0x752e17784040>, kernel=<OpOverload(op='aten.to_dense_backward', overload='default')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d080>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795120>, kernel=<OpOverload(op='aten.special_digamma', 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 0x752e177860c0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177951c0>, kernel=<OpOverload(op='aten.embedding_sparse_backward', overload='default')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e700>, kernel=<OpOverload(op='aten.diagflat', overload='default')>), <OpOverload(op='aten.value_selecting_reduction_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797eb60>, 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 0x752e17790220>, 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 0x752e1797dd00>, 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 0x752e1797cfe0>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3420>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d3a0>, kernel=<OpOverload(op='aten.is_conj', overload='default')>), <OpOverload(op='aten.row_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d8a0>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa160>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784900>, kernel=<OpOverload(op='aten._weight_norm', 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 0x752e17785080>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <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 0x752e17786200>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a84a0>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785940>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177900e0>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3ba0>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f740>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785c60>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f100>, kernel=<OpOverload(op='aten.one_hot', overload='default')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177877e0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787560>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f4c0>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0b80>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0ea0>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f060>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786ac0>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786e80>, kernel=<OpOverload(op='aten.chalf', 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 0x752e17787740>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e160>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2e80>, kernel=<OpOverload(op='aten.argsort', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f420>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796fc0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793560>, 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 0x752e177936a0>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787a60>, kernel=<OpOverload(op='aten.orgqr', overload='default')>), <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177968e0>, 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 0x752e177a3d80>, 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 0x752e17790fe0>, kernel=<OpOverload(op='aten.pinverse', overload='default')>), <OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0720>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3f60>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795d00>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790400>, kernel=<OpOverload(op='aten.arccosh', overload='default')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791a80>, kernel=<OpOverload(op='aten._add_batch_dim', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a99e0>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792340>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177919e0>, kernel=<OpOverload(op='aten.linalg_eigh', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a37e0>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791580>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1580>, kernel=<OpOverload(op='aten.is_neg', overload='default')>), <OpOverload(op='aten.conv1d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0f40>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795da0>, 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 0x752e177aa660>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792c00>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177927a0>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792980>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793ce0>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a82c0>, kernel=<OpOverload(op='aten.linalg_eigvalsh', overload='default')>), <OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793a60>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>), <OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790900>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', 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 0x752e177944a0>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797600>, kernel=<OpOverload(op='aten.special_i0', overload='default')>), <OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796e80>, kernel=<OpOverload(op='aten._cufft_clear_plan_cache', overload='default')>), <OpOverload(op='aten.rnn_tanh_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0400>, kernel=<OpOverload(op='aten.rnn_tanh_cell', 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 0x752e17794a40>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177920c0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a98a0>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795300>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796340>, kernel=<OpOverload(op='aten.max', overload='names_dim')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795ee0>, kernel=<OpOverload(op='aten.special_erfinv', overload='default')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0900>, kernel=<OpOverload(op='aten.flipud', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794400>, 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 0x752e177928e0>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794cc0>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d620>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa340>, 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 0x752e177856c0>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177972e0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa980>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790720>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785bc0>, kernel=<OpOverload(op='aten.__or__', overload='Tensor')>), <OpOverload(op='aten.special_exp2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785a80>, kernel=<OpOverload(op='aten.special_exp2', overload='default')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1080>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0ae0>, kernel=<OpOverload(op='aten.linalg_cholesky', overload='default')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1c60>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785760>, kernel=<OpOverload(op='aten.poisson_nll_loss', 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 0x752e177a2480>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a23e0>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>), <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790e00>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791bc0>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792e80>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792a20>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9bc0>, kernel=<OpOverload(op='image.encode_jpeg', overload='default')>), <OpOverload(op='aten.embedding_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794180>, 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 0x752e17793740>, kernel=<OpOverload(op='aten.align_as', overload='default')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785580>, kernel=<OpOverload(op='aten.arcsin', 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 0x752e17794860>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794ae0>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9ee0>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177953a0>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a02c0>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177962a0>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177967a0>, kernel=<OpOverload(op='aten.linalg_multi_dot', 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 0x752e177a2980>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796020>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177959e0>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8fe0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797100>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796d40>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797ce0>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='aten._debug_has_internal_overlap', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797b00>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177976a0>, 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 0x752e17797a60>, kernel=<OpOverload(op='aten.linalg_inv', overload='default')>), <OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aaa20>, 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 0x752e177a09a0>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0360>, 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 0x752e177a1260>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792020>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1bc0>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='profiler._record_function_enter_new', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aab60>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177ab240>, 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 0x752e177a8040>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aac00>, 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 0x752e177a3560>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a32e0>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a80e0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a39c0>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792d40>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790c20>, 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 0x752e177909a0>, 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 0x752e177a9800>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9580>, kernel=<OpOverload(op='image.decode_png', overload='default')>), <OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177904a0>, kernel=<OpOverload(op='aten._validate_sparse_coo_tensor_args', overload='default')>), <OpOverload(op='aten.special_gammaincc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177916c0>, kernel=<OpOverload(op='aten.special_gammaincc', overload='default')>), <OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177914e0>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791080>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790ea0>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791da0>, kernel=<OpOverload(op='aten.nll_loss_nd', overload='default')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786480>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791940>, kernel=<OpOverload(op='aten.smm', overload='default')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792700>, kernel=<OpOverload(op='aten.slow_conv3d', 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 0x752e17793ba0>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793920>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794b80>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <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 0x752e17795440>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794ea0>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796520>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796660>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa2a0>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796160>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784c20>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177849a0>, 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 0x752e17784720>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784360>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784fe0>, 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 0x752e17785120>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784f40>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784e00>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786020>, kernel=<OpOverload(op='aten.matrix_exp', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785ee0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785f80>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785da0>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786700>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', 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 0x752e17787240>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786980>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786a20>, kernel=<OpOverload(op='aten._convolution_mode', 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 0x752e17787420>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787060>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787100>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790040>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='str_ord')>), <OpOverload(op='aten.to_mkldnn_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787880>, kernel=<OpOverload(op='aten.to_mkldnn_backward', overload='default')>), <OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797de40>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', 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 0x752e177a8680>, 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 0x752e177a8900>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e0c0>, kernel=<OpOverload(op='aten.stride', overload='int')>), <OpOverload(op='aten._shape_as_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797dee0>, 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 0x752e1797eac0>, 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 0x752e1797e7a0>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797ef20>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797ede0>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797e20>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177947c0>, kernel=<OpOverload(op='aten.lstm_cell', 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 0x752e177932e0>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', 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 0x752e177a9120>, kernel=<OpOverload(op='image._is_compiled_against_turbo', overload='default')>), <OpOverload(op='aten.masked_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f1a0>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9620>, kernel=<OpOverload(op='c10d_functional.broadcast', 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 0x752e17795b20>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3a60>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2de0>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a13a0>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796a20>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a14e0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797fc40>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3060>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f6a0>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f600>, 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 0x752e1797f380>, 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 0x752e1797fb00>, 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 0x752e1797f9c0>, 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 0x752e177a8d60>, 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 0x752e177a8ae0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a28e0>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797dc60>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787b00>, 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 0x752e17787920>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787e20>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787c40>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790180>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177871a0>, 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 0x752e177868e0>, 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 0x752e177a93a0>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786de0>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1d00>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786340>, 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 0x752e177862a0>, 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 0x752e177a91c0>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177872e0>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787380>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784ea0>, 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 0x752e17784d60>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8cc0>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797fe20>, 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 0x752e17784ae0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177847c0>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787f60>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797fa60>, 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 0x752e1797fce0>, kernel=<OpOverload(op='aten.matrix_exp_backward', overload='default')>), <OpOverload(op='aten._version', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785300>, kernel=<OpOverload(op='aten._version', 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 0x752e177a2160>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797380>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177854e0>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796700>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794c20>, kernel=<OpOverload(op='aten._sparse_sum', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792200>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784860>, 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 0x752e17785d00>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786160>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f240>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785620>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786ca0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177863e0>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17785b20>, kernel=<OpOverload(op='aten.linalg_tensorinv', 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 0x752e17785440>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a07c0>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791260>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2020>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177922a0>, 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 0x752e177907c0>, 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 0x752e17797740>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e520>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa200>, kernel=<OpOverload(op='image.decode_webp', overload='default')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786d40>, kernel=<OpOverload(op='aten.argwhere', 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 0x752e17793b00>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790f40>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177859e0>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9a80>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797ec0>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792660>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792fc0>, kernel=<OpOverload(op='aten.linalg_solve_ex', overload='default')>), <OpOverload(op='aten.is_distributed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0860>, 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 0x752e177a0cc0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0fe0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794d60>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3ce0>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9f80>, kernel=<OpOverload(op='image.encode_png', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790360>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177963e0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten._test_serialization_subcmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177925c0>, 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 0x752e177a1b20>, 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 0x752e177a18a0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796ac0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793380>, 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 0x752e17797920>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790ae0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787ec0>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a36a0>, kernel=<OpOverload(op='aten.special_log1p', overload='default')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177931a0>, kernel=<OpOverload(op='aten.fix', 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 0x752e177a8360>, 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 0x752e17792b60>, 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 0x752e17795c60>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177867a0>, kernel=<OpOverload(op='aten.rnn_relu_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 0x752e17793e20>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793880>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791440>, kernel=<OpOverload(op='aten.size', overload='int')>), <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8c20>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794040>, kernel=<OpOverload(op='aten.combinations', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e340>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797ed40>, 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 0x752e1797ee80>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177918a0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796840>, kernel=<OpOverload(op='aten.gather_backward', overload='default')>), <OpOverload(op='aten.native_channel_shuffle', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795580>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791620>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795760>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795620>, 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 0x752e177a1ee0>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2700>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3b00>, kernel=<OpOverload(op='aten.data', overload='default')>), <OpOverload(op='aten._is_zerotensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177974c0>, kernel=<OpOverload(op='aten._is_zerotensor', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177942c0>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791760>, kernel=<OpOverload(op='aten.cudnn_is_acceptable', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796de0>, kernel=<OpOverload(op='aten.linalg_norm', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1a80>, 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 0x752e177a0a40>, kernel=<OpOverload(op='aten._sparse_softmax', overload='int')>), <OpOverload(op='aten.result_type', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791d00>, kernel=<OpOverload(op='aten.result_type', overload='Scalar')>), <OpOverload(op='image.decode_gif', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa0c0>, kernel=<OpOverload(op='image.decode_gif', overload='default')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177954e0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792160>, 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 0x752e177940e0>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177965c0>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793240>, 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 0x752e17794900>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177958a0>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177979c0>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792840>, kernel=<OpOverload(op='aten._backward', overload='default')>), <OpOverload(op='aten._reshape_from_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177945e0>, kernel=<OpOverload(op='aten._reshape_from_tensor', 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 0x752e177853a0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3100>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3600>, kernel=<OpOverload(op='aten.linalg_norm', overload='ord_str')>), <OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2200>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797060>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795f80>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792480>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a22a0>, kernel=<OpOverload(op='aten._test_check_tensor', 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 0x752e17786c00>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', 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 0x752e17784220>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a27a0>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2a20>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e480>, kernel=<OpOverload(op='aten.float_power', overload='Scalar')>), <OpOverload(op='aten.float_power', overload='Tensor_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e2a0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787ba0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e980>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d1c0>, kernel=<OpOverload(op='aten.diag', overload='default')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17f09d00>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x752eecb19da0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x752eecb19ee0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x752eecb1a020>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x752eecb1a160>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x752eecb1a2a0>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x752eecb1a3e0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x752eecb19e40>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x752eecb1a520>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x752eecb1a660>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x752eecb1a340>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x752eecb1a7a0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x752eecb1a7a0>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x752eecb1a0c0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x752eecb1a8e0>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x752eecb19940>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x752eecb19940>, <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 0x752eecb18540>, <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 0x752eecb18540>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x752eecb191c0>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x752eecb1ade0>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x752eecb1ade0>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x752eecb1b100>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x752eecb1b100>, <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 0x752eecce7880>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x752eecb1b420>, <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 0x752eecb1b420>, <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 0x752eecb1b420>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x752eecce7880>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x752eecb1b740>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x752eecb1ba60>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x752eecb1be20>, <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 0x752eecb1be20>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x752eecb1ae80>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x752eecb1ae80>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x752eecb19d00>, <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 0x752eecb19d00>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x752eecb55120>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x752eecb553a0>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8540>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x752eecb55e40>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a3e20>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x752eecb1ac00>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x752eecb57100>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790cc0>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x752eecb55300>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793c40>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790a40>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x752eecb56160>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17790540>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x752eecb563e0>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177913a0>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x752eecb56660>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177911c0>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x752eecb56f20>, <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791ee0>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x752eecb571a0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791b20>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x752eecb57420>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793060>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x752eecb576a0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793100>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x752eecb57d80>, <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793f60>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x752eecb57f60>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177934c0>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x752eecb56020>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794680>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x752eecb780e0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794220>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x752eecb78360>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177956c0>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x752eecb785e0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17795260>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x752eecb78860>, <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796200>, 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 0x752e177960c0>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x752eecb79080>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x752eecb79440>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x752eecb7a3e0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x752eecb7a660>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x752eecb7ad40>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x752eecb7b1a0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x752eecb7b9c0>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x752eecb7b7e0>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x752eecb7b7e0>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x752eecb7bd80>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x752eecbb4b80>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x752eecbb65c0>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x752eecbb65c0>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x752eecbb4ea0>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x752eecbb44a0>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x752eecbb4b80>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x752eecbb4ea0>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787ce0>, 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 0x752e177879c0>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x752eecbb5ee0>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x752eecbb62a0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x752eecbb62a0>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x752eecbb6200>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x752eecbb6de0>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x752eecbb7920>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797efc0>, 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 0x752e1797f880>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x752eecbb6ca0>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x752eecbb6ca0>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x752eecbcc220>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x752eecbcc4a0>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x752eecbb6f20>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x752eecbcc2c0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x752eecbcc540>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x752eecbcc680>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x752eecbcc7c0>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x752eecbccea0>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x752eecbccea0>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x752eecbcd260>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x752eecbcd260>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x752eecbcd620>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x752eecbcdbc0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x752eecbcdbc0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x752eecbce0c0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x752eecbce0c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x752eecbce5c0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x752eecbce5c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x752eecbce980>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x752eecbce980>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x752eecbcf100>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x752eecbced40>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x752eecbce160>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x752eecbce160>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17786840>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x752eecbcf240>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x752eecbcf600>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x752eecbcf600>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x752eecbcfc40>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x752eecbcfc40>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x752eecbf40e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x752eecc21580>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x752eecc22020>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x752eecc376a0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x752eecc35260>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x752eecc36520>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x752eecc36520>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x752eecc36de0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x752eecc37240>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x752eecc36020>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x752eecc37880>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x752eecc37ce0>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x752eecc49e40>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x752eecc4a2a0>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x752eecc4aa20>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x752eecc4ade0>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x752eecc4b1a0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x752eecc4b420>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x752eecc4b6a0>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x752eecc74720>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1e40>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1f80>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177977e0>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x752eecc749a0>, <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a20c0>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x752eecc4b740>, <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0e00>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797d80>, 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 0x752e177a05e0>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a0220>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a00e0>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x752eecc754e0>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a1440>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x752eecc75300>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x752eecc767a0>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a19e0>, 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 0x752e177a1800>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2ca0>, 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 0x752e177a2d40>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x752eecc75c60>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x752eecc760c0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x752eecc771a0>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x752eecc765c0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x752eecc90540>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x752eecc779c0>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x752eecc77ec0>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x752eecc75620>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x752eecc75620>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x752eecc90900>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x752eecc90f40>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x752eecc90ea0>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x752eecc911c0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x752eecc911c0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x752eecc914e0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x752eecc91b20>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x752eecc927a0>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17794720>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17793ec0>, 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 0x752e17793d80>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x752eecc92160>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x752eecc91f80>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x752eecc92a20>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x752eecc90a40>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x752eecc92d40>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x752eecc92d40>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x752eecc93100>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x752eecc93100>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x752eecc93a60>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x752eecc93a60>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x752eeccb9440>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17797240>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x752eecc93b00>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17796c00>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x752eecc93ba0>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x752eeccbbf60>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x752eeccbbf60>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x752eeccb9800>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x752eeccb9a80>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x752eeccb9d00>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a2f20>, 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 0x752e177a2fc0>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x752eecc937e0>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x752eeccbae80>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x752eeccbae80>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x752eeccbad40>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x752eeccbab60>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x752eeccbab60>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791e40>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17791c60>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x752eeccbb240>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x752eeccbb4c0>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x752eeccbb4c0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x752eeccbb9c0>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x752eeccbbc40>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x752eeccb8ae0>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x752eecce4220>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x752eecce4220>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x752eecce45e0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x752eecce45e0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x752eecce49a0>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x752eecce49a0>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x752eecce4d60>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x752eecce4d60>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x752eecce4e00>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x752eecce5760>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x752eecce5c60>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x752eecce5ee0>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x752eecce4cc0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x752eecce7ec0>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x752eecce6200>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x752eecce6200>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x752eecce6480>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x752eeccbb1a0>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x752eecce7e20>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x752eecce62a0>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x752eecce7e20>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x752eecce6520>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x752eecce6660>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x752eecce6660>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x752eecce67a0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x752eecce68e0>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x752eecce7ce0>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x752eecce6a20>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x752eecb1a5c0>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x752eecce6b60>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x752eecce6ca0>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x752eecce6de0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x752eecce6f20>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x752eecce7ba0>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x752eecce7060>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x752eecce7060>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x752eecce71a0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x752eecce71a0>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x752eecce7240>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x752eecce72e0>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x752eecb1ab60>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x752eecce7420>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x752eecce7420>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x752eecce74c0>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x752eecce7560>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x752eecce7560>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x752eecce76a0>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x752eecce76a0>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x752eecce7740>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x752eecb19300>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x752eecce77e0>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x752eecce7740>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x752eecce7c40>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x752eecce7920>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x752eecce7920>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x752eecce7a60>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x752eecce7a60>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x752eecce79c0>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x752eecce7ba0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x752eecce79c0>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x752eecb19580>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x752eecce6fc0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x752eecce6d40>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x752eecce6d40>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x752eecce6ac0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x752eecb193a0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x752eecce6840>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x752eecce65c0>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x752eecce7b00>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x752eecce6160>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x752eecb1aac0>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x752eecce7f60>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x752eecb180e0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x752eecce7600>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x752eecb18220>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x752eecb18220>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x752eecce7100>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x752eecb18360>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x752eecb1aac0>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x752eecb184a0>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x752eecb184a0>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x752eecb185e0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x752eecb185e0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x752eecb18720>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x752eecce6c00>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x752eecb18860>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x752eecb189a0>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x752eecb189a0>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x752eecce6700>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x752eecb18ae0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x752eecb18ae0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x752eecb18c20>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x752eecb18d60>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x752eecb1a840>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x752eecb18ea0>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x752eecb1ba60>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x752eecce63e0>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x752eecce5080>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x752eecb18fe0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x752eecb19120>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x752eecce5080>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x752eecb19260>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x752eecb19080>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x752eecb18e00>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x752eecb18b80>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x752eecce6e80>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x752eecb18900>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x752eecce7380>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x752eecb18680>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x752eecb18400>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x752eecb18400>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x752eecb18180>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x752eecce7d80>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x752eecb19620>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x752eecb19760>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x752eecb19760>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x752eecb198a0>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x752eecb199e0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x752eecb19b20>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x752eecb19b20>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x752eecb1aa20>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x752eecb19c60>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x752eecd22340>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x752eecd207c0>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x752eecd22c00>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x752eecd22c00>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x752eecd236a0>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x752eecd23a60>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x752eecd23d80>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x752eecd23f60>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x752eecd23f60>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x752eecd347c0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x752eecd23600>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x752eecef6480>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x752eecd34540>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x752eecd34ae0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x752eecd34fe0>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd34fe0>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x752eecd351c0>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x752eecd35e40>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x752eecd35800>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x752eecd35800>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x752eecd35bc0>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x752eecd35bc0>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x752eecef6480>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x752eecd23600>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x752eecd367a0>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37a60>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x752eecd36a20>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x752eecd36d40>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x752eecd36d40>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x752eecd371a0>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37ce0>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x752eecd376a0>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x752eecd376a0>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x752eecd374c0>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x752eecd374c0>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37f60>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37740>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37880>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37ec0>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd379c0>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37b00>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37c40>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37c40>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37d80>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd377e0>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd377e0>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36f20>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36ca0>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36ca0>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36980>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36980>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x752eecd363e0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd363e0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd36020>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd700e0>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70220>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70360>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd704a0>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x752eecd704a0>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x752eecd705e0>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70720>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x752eecd70860>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd709a0>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17787d80>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x752eecdd9e40>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x752eecdd9e40>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x752eecddb880>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x752eecddb880>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x752eecdec2c0>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x752eecdeccc0>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x752eecd734c0>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x752eecdecfe0>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x752eecded4e0>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x752eecdedee0>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x752eecdedee0>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x752eecdefd80>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f7e0>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x752eecdee480>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x752eecdee480>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x752eece09440>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x752eece09940>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x752eece09940>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x752eece0ad40>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x752eecc21e40>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x752eecea2480>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x752eecea18a0>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x752eecea2e80>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x752eecea1a80>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x752eecd37a60>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x752eecea2e80>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x752eecea1da0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x752eece78860>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x752eed253560>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x752eecc23ec0>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177858a0>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x752eecd236a0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x752eecce5120>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x752eecef6160>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x752eecea23e0>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x752eecef6160>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x752eecea1da0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x752eecea3f60>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x752eecc23ec0>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792de0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x752eecce63e0>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x752eece79da0>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x752eece784a0>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x752eece784a0>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x752eece78860>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x752eece78400>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x752eece78400>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x752eece79440>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x752eece7b6a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x752eece793a0>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x752eece793a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x752eece7b6a0>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x752eece78040>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x752eece799e0>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x752eece79d00>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x752eece79d00>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x752eece79da0>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x752eece7a0c0>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x752eece7a0c0>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x752eece7a480>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x752eece7a480>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x752eece7aca0>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x752eece7ad40>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x752eece7b100>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x752eece7b600>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x752eece7b600>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x752eecea00e0>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x752eece79580>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x752eece7bec0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x752eece78d60>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x752eece78d60>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x752eece7b9c0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x752eece7be20>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x752eece7be20>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x752eecea02c0>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x752eecea04a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x752eece7bd80>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x752eece7bd80>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x752eecea0720>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x752eecea0720>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x752eecea0ae0>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x752eecea0ae0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x752eecea0e00>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x752eecea13a0>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x752eecea0c20>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x752eecea0c20>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x752eecea0900>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x752eecea0900>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x752eecea18a0>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x752eecea1b20>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x752eecea1b20>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x752eecea2480>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x752eecea32e0>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x752eecea2700>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x752eecea2700>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x752eecea2840>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x752eecea2840>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x752eecea31a0>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x752eecea1a80>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x752eecea3240>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x752eecea3240>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x752eecea3ba0>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x752eecec8220>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x752eecec84a0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x752eecec8a40>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x752eecec8a40>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x752eecec8720>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x752eecec8720>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x752eecec8ae0>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x752eecec8ae0>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x752eecec8d60>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x752eecec9300>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x752eecec93a0>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x752eecec8ea0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x752eecec8ea0>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x752eecec9620>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x752eecec9260>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x752eecec9940>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x752eecec96c0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x752eececb6a0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x752eececa160>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x752eececa660>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x752eececaac0>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x752eececa660>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x752eececab60>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x752eececaca0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x752eecef6520>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x752eececb880>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x752eecec9580>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x752eececb920>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x752eececba60>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x752eececb6a0>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x752eecef51c0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x752eececbc40>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x752eececbc40>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x752eececbce0>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x752eececbce0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x752eecef4900>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x752eecef4900>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x752eecef4d60>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x752eecd222a0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x752eecec9b20>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x752eecec9b20>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x752eecec8180>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x752eecef7600>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x752eecef4fe0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x752eecef5120>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x752eecef5120>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x752eecef56c0>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x752eecef5da0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x752eecef5da0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x752eecef5f80>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x752eececb9c0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x752eecef4ae0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x752eecef7600>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x752eecef7100>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x752eecef6ca0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x752eecef5620>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x752eecef6d40>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x752eececb9c0>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x752eecef5620>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x752eecef7e20>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x752eecd20040>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x752eecd20220>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x752eecd20400>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x752eecd20a40>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x752eecd20d60>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x752eecd21120>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x752eecd22700>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x752eecd21300>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x752eecd219e0>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x752eecd21f80>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x752eecd21d00>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x752eecc349a0>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x752eecc34e00>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x752eecc356c0>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x752eecc356c0>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x752eecc35b20>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x752eecc35b20>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x752eecb18e00>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x752eecc35f80>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x752eecc35f80>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x752eecc36980>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x752eecc48180>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x752eecc485e0>, <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177844a0>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x752eecc48a40>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x752eeccb9760>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x752eecce6e80>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x752eecc49300>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x752eecc49760>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x752eecc49760>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797f920>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x752eecc22b60>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x752eecc49bc0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x752eecc934c0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x752eecc76660>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17792520>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x752eecc4bce0>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x752eecd21800>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x752eecd36ac0>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x752eecea13a0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x752eecc747c0>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177876a0>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e200>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797d260>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x752eecc74b80>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797df80>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17c94900>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.var', overload='out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e1797e020>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x752eecc75260>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x752eecc74d60>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x752eecc74c20>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x752eecef4d60>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x752eecb1b740>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x752eecc4bf60>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x752eecdd9b20>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x752eecdd9b20>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x752eecdef7e0>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x752eecdd8b80>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x752eecdda340>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x752eece0be20>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x752eecddb240>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x752eecdda840>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x752eecddad40>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x752eecddbd80>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x752eecea0e00>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x752eecdee3e0>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x752eecdee8e0>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x752eecdeede0>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x752eecb18b80>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x752eecdef2e0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x752eecc76c00>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x752eeccba340>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x752eece08680>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x752eecdefec0>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x752eece08a40>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x752eece08f40>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x752eece09e40>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x752eece0a340>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x752eece0a340>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x752eece0a840>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x752eece0a3e0>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x752eece0b060>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x752eece0b920>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x752eecc20360>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x752eecc76ca0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x752eecc20cc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x752eecc35bc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x752eecc21120>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x752eecc219e0>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x752eecc21a80>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x752eecc222a0>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x752eecc22700>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x752eecc236a0>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x752eecc236a0>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x752eecc23b00>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x752eed253560>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x752eecc23b00>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x752eecc23f60>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x752eecc227a0>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x752eecc227a0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x752eecd70180>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x752eecd72a20>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x752eecd72f20>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x752eecd73920>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x752eecd73e20>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x752eecdd9120>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x752eecdd9620>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x752eecdd8860>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x752eecdd8c20>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x752eecdee3e0>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x752eecdd8360>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x752eecdd8c20>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x752eecec8cc0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x752eecc48ea0>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x752eece7a840>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x752eecec9800>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x752eece7ad40>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x752eecd36840>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x752eecc20860>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x752eecc92a20>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x752eecd73420>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x752eecd71b20>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x752eecc76660>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x752eecec8e00>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x752eeccb91c0>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x752eeccba700>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x752eeccba480>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x752eeccb8a40>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x752eeccb8cc0>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x752eeccb8e00>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x752eecc92e80>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x752eeccb8c20>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e17784a40>, 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 0x752e17784400>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x752eecea2200>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x752eecd372e0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x752eece08a40>, <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa840>, kernel=<OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177ab100>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa7a0>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8a40>, 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 0x752e177a9440>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a94e0>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9300>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa700>, kernel=<OpOverload(op='sparse.qlinear_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 0x752e177a8860>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>), <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa3e0>, 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 0x752e177a96c0>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177ab060>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aade0>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa520>, 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 0x752e177aa480>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa8e0>, kernel=<OpOverload(op='quantized.conv2d_transpose', overload='default')>), <OpOverload(op='quantized.conv3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177ab1a0>, kernel=<OpOverload(op='quantized.conv3d_unpack', overload='default')>), <OpOverload(op='quantized.conv_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9940>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9c60>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9da0>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9d00>, 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 0x752e177a9e40>, 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 0x752e177a9260>, 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 0x752e177a87c0>, 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 0x752e177a85e0>, 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 0x752e177a89a0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aa020>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8ea0>, kernel=<OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8e00>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aaca0>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aaf20>, kernel=<OpOverload(op='quantized.linear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aaac0>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a9b20>, 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 0x752e177aa5c0>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177ab2e0>, kernel=<OpOverload(op='quantized.conv2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177aad40>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x752e177a8720>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}