.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.00042784600009326823 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.0012301789975026622 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-TJK._add_shape_information] dynamic shapes replacements={} [GraphBuilder-TJK.optimize] start with 13 nodes [GraphBuilder-TJK.optimize] #patterns=85 [GraphBuilder-TJK.optimize] start with subgraphs [GraphBuilder-TJK.optimize] done with subgraphs [GraphBuilderPatternOptimization-TJK.optimize] start with 7 nodes, 0 initializers, 85 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-TJK.optimize] same children={'SameChildrenPattern', 'SameChildrenFromInputPattern'} [GraphBuilderPatternOptimization-TJK.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-TJK.optimize] increase priority to 1 [GraphBuilderPatternOptimization-TJK.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-TJK.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=FunctionCausalMaskPattern:0.000 [GraphBuilderPatternOptimization-TJK.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-TJK.optimize] increase priority to 3 [GraphBuilderPatternOptimization-TJK.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-TJK.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=MatMulAddPattern:0.000 [GraphBuilderPatternOptimization-TJK.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-TJK.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-TJK.optimize] done after 5 iterations with 4 nodes in 0.007 [GraphBuilder-TJK.optimize] done with 4 nodes in 0.008 [GraphBuilder-TJK.to_onnx] make_model 1 inits 0 params [GraphBuilder-TJK.time_evaluation_constants_] 0 [GraphBuilder-TJK._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-TJK._build_initializers] switch low/high order [GraphBuilder-TJK._build_initializers] done in 1.115000486606732e-06s with 1 initializers, 0 large initializers [GraphBuilder-TJK._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.009239798000635346s and 4 nodes, 1 initializers, 5 inputs, 1 outputs [onnx_custom_backend] to_onnx done in 0.011656473998300498 with 4 nodes and 0 local functions. [onnx_custom_backend] starts creating InferenceSession [onnx_custom_backend] InferenceSession done in 0.0019504310002957936 discrepancies: 5.960464477539063e-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.0006131450027169194 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.0015434179986186791 s [to_onnx] start conversion to onnx (before optimization) mask_outputs=None [GraphBuilder-CIA._add_shape_information] dynamic shapes replacements={} [GraphBuilder-CIA.optimize] start with 13 nodes [GraphBuilder-CIA.optimize] #patterns=85 [GraphBuilder-CIA.optimize] start with subgraphs [GraphBuilder-CIA.optimize] done with subgraphs [GraphBuilderPatternOptimization-CIA.optimize] start with 7 nodes, 0 initializers, 85 patterns, priorities=[0, 1, 3], max_iter=30 [GraphBuilderPatternOptimization-CIA.optimize] same children={'SameChildrenPattern', 'SameChildrenFromInputPattern'} [GraphBuilderPatternOptimization-CIA.optimize] iteration 0: 7 nodes, priority=0 [GraphBuilderPatternOptimization-CIA.optimize] increase priority to 1 [GraphBuilderPatternOptimization-CIA.optimize] iteration 1: 7 nodes, priority=1 [GraphBuilderPatternOptimization-CIA.optimize] applies 2 matches, 1*TransposeEqualReshapePattern, 1*TransposeMatMulPattern - time=0.001 | max_time=IdentityPattern:0.000 [GraphBuilderPatternOptimization-CIA.optimize] iteration 2: 6 nodes, priority=1 [GraphBuilderPatternOptimization-CIA.optimize] increase priority to 3 [GraphBuilderPatternOptimization-CIA.optimize] iteration 3: 6 nodes, priority=3 [GraphBuilderPatternOptimization-CIA.optimize] applies 2 matches, 2*MatMulAddPattern - time=0.001 | max_time=ShapeBasedEditDistanceReshapePattern:0.000 [GraphBuilderPatternOptimization-CIA.optimize] iteration 4: 4 nodes, priority=3 [GraphBuilderPatternOptimization-CIA.optimize] stops current_priority_index=3, priorities=[0, 1, 3] [GraphBuilderPatternOptimization-CIA.optimize] done after 5 iterations with 4 nodes in 0.012 [GraphBuilder-CIA.optimize] done with 4 nodes in 0.013 [GraphBuilder-CIA.to_onnx] make_model 1 inits 0 params [GraphBuilder-CIA.time_evaluation_constants_] 0 [GraphBuilder-CIA._build_initializers] start with 1 initializers, large_model=False, external_threshold=1024 [GraphBuilder-CIA._build_initializers] switch low/high order [GraphBuilder-CIA._build_initializers] done in 1.0830008250195533e-06s with 1 initializers, 0 large initializers [GraphBuilder-CIA._add_shape_information] dynamic shapes replacements={} [to_onnx] to_onnx done in 0.014119540999672608s and 4 nodes, 1 initializers, 5 inputs, 1 outputs +C init7_s2_-1_1: int64:(2,):[-1, 1] +I input0: float32:(32, 10):0.2675032317638397,-0.10546927899122238,0.1660752147436142,0.15767191350460052,-0.020796235650777817... +I input1: float32:(32,):0.24780017137527466,-0.10903133451938629,0.15580907464027405,0.3047914505004883,0.05736222863197327... +I input2: float32:(3, 10):-1.4073114395141602,-0.3455547094345093,0.8224779367446899,-0.12058211863040924,0.23375029861927032... +I input3: float32:(1, 32):0.007405896671116352,0.005556197836995125,-0.11588052660226822,-0.14415216445922852,0.031211933121085167... +I input4: float32:(1,):[-0.15288875997066498] Gemm(input2, input0, input1) -> input_1 + input_1: float32:(3, 32):-0.4920448064804077,-0.11916979402303696,0.03290171921253204,0.43793386220932007,-0.773595929145813... Sigmoid(input_1) -> input_2 + input_2: float32:(3, 32):0.3794119954109192,0.47024276852607727,0.5082246661186218,0.6077666282653809,0.31570175290107727... 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.007405896671116352,0.005556197836995125,-0.11588052660226822,-0.14415216445922852,0.031211933121085167... Gemm(input_2, l_self_modules_layers_modules_2_parameters_weight_::T10, input4) -> output_0 + output_0: float32:(3, 1):[0.29841262102127075, 0.3401516079902649, 0.25289177894592285] discrepancies: 5.960464477539063e-08
dynger_backend¶
- experimental_experiment.torch_dynamo.dynger_backend(graph_module: GraphModule, args: List[Tensor | SymInt | SymFloat], dynamic_shapes: Dict[str, Any] | Tuple[Any] | None = None, optimize: bool = True, verbose: int | Tuple[int, int] = 0) Callable[source]¶
Eager backend for dynamo.
- Parameters:
graph_module – graph to export
args – arguments
optimize – optimize or not, those optimization would be done on the graph module itself
verbose – adjust verbosity, if tuple, if gives different verbosity level to the exporter and the runtime
- Returns:
Callable
Next examples shows how to display intermediate results while executing the graph produced by torch dynamo.
<<<
import torch from experimental_experiment.torch_dynamo import dynger_backend class MLP(torch.nn.Module): def __init__(self): super().__init__() self.layers = torch.nn.Sequential( torch.nn.Linear(10, 32), torch.nn.Sigmoid(), torch.nn.Linear(32, 1), ) def forward(self, x): return self.layers(x) x = torch.randn(3, 10, dtype=torch.float32) mlp = MLP() expected = mlp(x) compiled_model = torch.compile( mlp, backend=lambda *args, **kwargs: dynger_backend(*args, verbose=10, **kwargs), dynamic=False, fullgraph=True, ) got = compiled_model(x) diff = (expected - got).max() print(f"discrepancies: {diff}")
>>>
[dynger_backend] use existing <class 'torch.fx.graph_module.GraphModule.__new__.<locals>.GraphModuleImpl'> [dynger_backend] begin execution with 9 nodes <built-in function linear>((l_x_, l_self_modules_layers_modules_0_parameters_weight_, l_self_modules_layers_modules_0_parameters_bias_)) -> input_1 + input_1: torch.float32:torch.Size([3, 32]):-0.4327884912490845,-0.09558623284101486,-0.1036977767944336,-0.8529689311981201,0.8067309856414795... <built-in method sigmoid of type object at 0x7d71e7d2db60>((input_1,)) -> input_2 + input_2: torch.float32:torch.Size([3, 32]):0.3934606611728668,0.4761216342449188,0.47409871220588684,0.2988104224205017,0.6914124488830566... <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.3906020522117615,0.2821168601512909,0.20338648557662964 [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 0x7d717b4bcce0>: <function out_dtype_decomp at 0x7d717adb4360>, <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7d717abf4fe0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7d717abf4fe0>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7d717abf56c0>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7d717abf56c0>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7d717abf5b20>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7d717abf5b20>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7d717abf6160>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7d717abf6160>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7d717abf5ee0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7d717abf5ee0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7d717abf6480>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7d717abf6480>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7d717abf7560>, <OpOverload(op='aten.selu', overload='default')>: <function selu at 0x7d717abf68e0>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7d717abf7880>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7d717abf6e80>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7d717abf6e80>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7d717abf7560>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7d717abf7880>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: <function margin_ranking_loss at 0x7d717aa28040>, <OpOverload(op='aten.hinge_embedding_loss', overload='default')>: <function hinge_embedding_loss at 0x7d717aa28360>, <OpOverload(op='aten.nll_loss', overload='default')>: <function nll_loss at 0x7d717aa28900>, <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7d717aa28900>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7d717aa28cc0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7d717aa28cc0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7d717abf7240>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7d717abf7240>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7d717aa28f40>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7d717aa28f40>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7d717aa29800>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7d717aa29800>, <OpOverload(op='aten.prelu', overload='default')>: <function prelu at 0x7d717aa29a80>, <OpOverload(op='aten.relu6', overload='default')>: <function relu6 at 0x7d717aa29d00>, <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7d717aa2a340>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7d717aa2a340>, <OpOverload(op='aten.pairwise_distance', overload='default')>: <function pairwise_distance at 0x7d717aa2a5c0>, <OpOverload(op='aten.pdist', overload='default')>: <function pdist at 0x7d717aa2a980>, <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7d717aa2ac00>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7d717aa2ac00>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7d717aa2ae80>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7d717aa2ae80>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7d717aa29940>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7d717aa2b1a0>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7d717aa2aca0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7d717aa2af20>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7d717aa28860>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7d717aa28860>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7d717aa51760>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7d717aa2b880>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7d717aa2b880>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7d717aa51760>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7d717aa2bc40>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7d717aa2bc40>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7d717aa2bf60>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7d717aa505e0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7d717aa505e0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7d717aa50ae0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7d717aa50ae0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7d717aa50fe0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7d717aa50fe0>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7d717aa513a0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7d717aa513a0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7d717aa518a0>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7d717aa518a0>, <OpOverload(op='aten.special_ndtr', overload='default')>: <function ndtr at 0x7d717aa51c60>, <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7d717aa51c60>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7d717aa52020>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7d717aa52020>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7d717aa52660>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7d717aa52660>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='dimname')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7d717ad0e200>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.meshgrid', overload='default')>: <function meshgrid at 0x7d717ab304a0>, <OpOverload(op='aten.meshgrid', overload='indexing')>: <function meshgrid at 0x7d717ab304a0>, <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7d717ab32520>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7d717ab32520>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7d717ab31940>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7d717ab31080>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7d717ab31080>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7d717ab31940>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7d717ab319e0>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7d717ab319e0>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7d717ab31ee0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7d717ab31ee0>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7d717ab32160>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7d717ab32160>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7d717ab32700>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7d717ab32ac0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7d717ab32ac0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7d717ab32e80>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7d717ab32e80>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7d717ab33240>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7d717ab33240>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7d717ab33c40>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7d717ab33c40>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7d717ab60180>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7d717ab61760>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7d717ab31e40>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7d717ab60720>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7d717ab60720>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7d717ab619e0>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7d717ab609a0>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7d717ab33740>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7d717ab61c60>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7d717ab61c60>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7d717ab607c0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7d717ab60a40>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7d717ab63a60>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7d717ab60b80>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7d717ab60b80>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7d717ab60cc0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7d717ab61ee0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7d717ab61ee0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7d717ab60e00>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7d717ab60f40>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7d717ab94f40>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7d717ab61080>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7d717ab611c0>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7d717ab62160>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7d717ab61300>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7d717ab637e0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7d717ab623e0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7d717ab61440>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7d717ab61580>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7d717ab61580>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7d717ab616c0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7d717ab616c0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7d717ab61800>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7d717ab62340>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7d717ab61940>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7d717ab61940>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7d717ab62340>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7d717ab61a80>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7d717ab61a80>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7d717ab61bc0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7d717ab63560>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7d717ab61d00>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7d717ab61e40>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7d717ab61e40>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7d717ab62200>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7d717ab61f80>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7d717ab61f80>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7d717ab620c0>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7d717ab639c0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7d717ab639c0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7d717ab61260>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7d717ab61260>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7d717ab60fe0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7d717ab60d60>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7d717ab60ae0>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7d717ab60680>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7d717ab63880>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7d717ab625c0>, <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 0x7d717ab63420>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7d717ab62700>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7d717ab62700>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7d717ab62840>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7d717ab62980>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7d717ab62980>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7d717ab62ac0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7d717ab62ac0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7d717ab62c00>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7d717ab631a0>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7d717ab62660>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7d717ab62d40>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7d717ab62e80>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7d717ab628e0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7d717ab62fc0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7d717ab62fc0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7d717ab63100>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7d717ab63240>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7d717ab62b60>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7d717ab63380>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7d717ab634c0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7d717ab62de0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7d717ab632e0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7d717ab63600>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7d717ab63060>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7d717ab63740>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7d717ab60040>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7d717ab60040>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7d717ab60c20>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7d717ab60c20>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7d717ab61120>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7d717ab62ca0>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7d717ab61620>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7d717ab61b20>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7d717ab61b20>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7d717ab627a0>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7d717ab62020>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7d717ab63b00>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7d717ab60900>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7d717ab63c40>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7d717ab63c40>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7d717ab63d80>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7d717ab63ec0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7d717ab94040>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7d717ab94040>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7d717ab613a0>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7d717ab94180>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7d717ab942c0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7d717ab94400>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7d717ab61da0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7d717ab94540>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7d717ab94680>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7d717ab947c0>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7d717ab94900>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7d717ab63ba0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7d717ab63ba0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7d717ab94a40>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7d717ab94b80>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7d717ab63ce0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7d717ab63f60>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7d717ab94cc0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7d717ab94cc0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7d717ab94e00>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7d717ab94fe0>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7d717ab94fe0>, <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 0x7d717ab949a0>, <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 0x7d717ab949a0>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7d717ab94360>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7d717ab94360>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7d717ab95300>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7d717ab95300>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7d717ab95f80>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7d717ab95620>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <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 0x7d717ab95940>, <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 0x7d717ab95940>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7d717ab95c60>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7d717ab95c60>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7d717ab95f80>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7d717ab60ea0>, <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 0x7d717ab60ea0>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7d717ab962a0>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7d717ab962a0>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <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 0x7d717ab959e0>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7d717ab977e0>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7d717ab97600>, <OpOverload(op='aten.fft_ifft', overload='default')>: <function ifft at 0x7d717ab977e0>, <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7d717ab97880>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7d717ab97880>, <OpOverload(op='aten.fft_fft', overload='default')>: <function fft at 0x7d717abcc360>, <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7d717abcc360>, <OpOverload(op='aten.fft_rfft', overload='default')>: <function rfft at 0x7d717abcc400>, <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7d717abcc400>, <OpOverload(op='aten.fft_irfft', overload='default')>: <function irfft at 0x7d717abcc680>, <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7d717abcc680>, <OpOverload(op='aten.fft_hfft', overload='default')>: <function hfft at 0x7d717abcc900>, <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7d717abcc900>, <OpOverload(op='aten.fft_ihfft', overload='default')>: <function ihfft at 0x7d717abccb80>, <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7d717abccb80>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7d717abcdb20>, <OpOverload(op='aten.fft_fftn', overload='default')>: <function fftn at 0x7d717abcd440>, <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7d717abcd440>, <OpOverload(op='aten.fft_fft2', overload='default')>: <function fft2 at 0x7d717abcdb20>, <OpOverload(op='aten.fft_ifftn', overload='default')>: <function ifftn at 0x7d717abcd6c0>, <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7d717abcd6c0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: <function rfftn at 0x7d717abcd940>, <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7d717abcd940>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: <function ihfftn at 0x7d717abcdbc0>, <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7d717abcdbc0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: <function irfftn at 0x7d717abce2a0>, <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7d717abce2a0>, <OpOverload(op='aten.fft_hfftn', overload='default')>: <function hfftn at 0x7d717abce520>, <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7d717abce520>, <OpOverload(op='aten.fft_ifft2', overload='default')>: <function ifft2 at 0x7d717abccae0>, <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7d717abccae0>, <OpOverload(op='aten.fft_rfft2', overload='default')>: <function rfft2 at 0x7d717abce5c0>, <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7d717abce5c0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: <function irfft2 at 0x7d717abce840>, <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7d717abce840>, <OpOverload(op='aten.fft_hfft2', overload='default')>: <function hfft2 at 0x7d717abceac0>, <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7d717abceac0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: <function ihfft2 at 0x7d717abced40>, <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7d717abced40>, <OpOverload(op='aten.fft_fftshift', overload='default')>: <function fftshift at 0x7d717abcede0>, <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7d717abcf560>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7d717abcf560>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7d717abcf920>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7d717abcf920>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7d717aa2bf60>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7d717abf4d60>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.is_complex', overload='default')>: <function is_complex at 0x7d717ac1a8e0>, <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7d717ac4c2c0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7d717ac4c2c0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7d717aca5d00>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7d717ac4dd00>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7d717ac4dd00>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7d717ab32700>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7d717ac4e700>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7d717ac4e700>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7d717ac4f100>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7d717ac4f100>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7d717ac4d760>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7d717ac4d760>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7d717ac4f420>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7d717ac4f420>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7d717ac4f920>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7d717ac4f920>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7d717ac64360>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7d717ac64360>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7d717ac791c0>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.logsumexp', overload='names')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7d717ac65800>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7d717ac65800>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7d717aa2b060>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7d717ac67880>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7d717ac67880>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7d717ab62e80>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7d717ac67d80>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7d717ac791c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7d717ab332e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7d717ac936a0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7d717ac936a0>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7d717ad0f9c0>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7d717ab97600>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7d717aca49a0>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7d717aca49a0>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7d717aca5260>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7d717aca5260>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7d717aca56c0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7d717aca56c0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7d717aca5760>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7d717aca5d00>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7d717aca6160>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7d717aca6160>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7d717ace8220>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7d717ab62480>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7d717acc0ea0>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7d717acc0ea0>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7d717acc1260>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7d717acc1260>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dimname')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.std_mean', overload='correction_names')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std_mean', overload='names_dim')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std_mean', overload='default')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std_mean', overload='dim')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std', overload='default')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='dim')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='names_dim')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.var_mean', overload='default')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='dim')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='correction_names')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='names_dim')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7d717ace85e0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7d717ace85e0>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7d717acc36a0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7d717acc36a0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7d717ace96c0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7d717ace96c0>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7d717acea200>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7d717ace98a0>, <OpOverload(op='aten.index_select', overload='dimname')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7d717ace9ee0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7d717ace9ee0>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7d717ace8220>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7d717aceade0>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7d717aceade0>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7d717aceb420>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7d717aceb420>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7d717aceb6a0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7d717aceb9c0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7d717aceb9c0>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7d717aceb380>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7d717ad0c040>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7d717ad0c040>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7d717ad0c5e0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7d717ad0d580>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7d717ad0d760>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7d717ad0cf40>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7d717ad0c180>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7d717ad0d260>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7d717ad0d620>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7d717ad0d620>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7d717ad0d940>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7d717ad0db20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7d717ad0f9c0>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7d717ad0df80>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.cumsum', overload='dimname')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7d717ad216c0>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7d717ad22020>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7d717ad223e0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7d717ad223e0>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7d717ad22660>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7d717ad22660>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7d717ad228e0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7d717ad22e80>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7d717ad22e80>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7d717ad22b60>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7d717ad22b60>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7d717ad22f20>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7d717ad23060>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7d717ad231a0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7d717ad23740>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7d717ad23f60>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7d717ad23f60>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7d717ad237e0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7d717ad21440>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7d717ad236a0>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7d717ad489a0>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7d717ad48cc0>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7d717ad485e0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7d717ad485e0>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7d717ad48a40>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7d717ad48b80>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7d717ad48f40>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7d717ad49120>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7d717ad48fe0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7d717ad49b20>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7d717ad49d00>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7d717ad48c20>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7d717ad49ee0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7d717ad49da0>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7d717ad49b20>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7d717ad4a0c0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7d717ad4a0c0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7d717ad4ad40>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7d717ad4b1a0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7d717ad4b1a0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7d717ad6cea0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7d717ad49440>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7d717ad49440>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7d717ad49f80>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.index_add', overload='dimname')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7d717ad6cea0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7d717ad4b560>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7d717ad6cd60>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7d717ad4bc40>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7d717ad6c400>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7d717ad6d580>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7d717ad49c60>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7d717ad6cd60>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7d717ad6da80>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7d717ad6d580>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7d717ad6d760>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7d717ad6d760>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7d717ad6da80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7d717ad6e2a0>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7d717ad6e480>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7d717ad6e660>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7d717ad6e840>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7d717ad6e7a0>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7d717ad6f380>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7d717ad6f1a0>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7d717ad6f560>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7d717ad6ede0>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7d717ad6f740>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7d717ad6fe20>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7d717ad8c400>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7d717ad6fe20>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7d717ad6f4c0>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7d717ad8c180>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7d717ad8da80>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7d717ad6c720>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7d717ad6c720>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7d717ad8d080>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7d717ad8d080>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7d717ad8db20>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7d717ad8db20>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7d717ad8dee0>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7d717ad8e200>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7d717ad8e200>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7d717ad8e480>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7d717ad8e480>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7d717ad6e5c0>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7d717ad6e5c0>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7d717ad8ede0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7d717adb4540>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7d717ad8f2e0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f420>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f420>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7d717adb42c0>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7d717ad8fc40>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7d717ad8fc40>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7d717adb42c0>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7d717adb40e0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7d717ad8ff60>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7d717ad8ff60>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7d717ad8f2e0>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb62a0>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb62a0>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7d717adb5440>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7d717ac67d80>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7d717adb51c0>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7d717adb51c0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7d717adb5b20>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6480>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7d717adb5940>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7d717adb5940>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5800>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5e40>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb60c0>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb60c0>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6340>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6200>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6020>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6020>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5580>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5580>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5260>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5260>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4fe0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4fe0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4b80>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4220>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6660>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb68e0>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb68e0>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6ca0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7d717ab31e40>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7d717ad4b560>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7d717ad4a660>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.where', overload='ScalarOther')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.where', overload='Scalar')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.where', overload='ScalarSelf')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7d717ad23a60>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7d717ad0d260>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7d717adb4f40>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7d717acc2ca0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7d717acc2ca0>, <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7d717ad236a0>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.var', overload='dim')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5d00>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7d717aeeff60>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='default')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction_names')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5f80>, <OpOverload(op='aten.var', overload='names_dim')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7d717acc3240>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7d717ad4a660>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7d717ab63420>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7d717acc3240>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7d717aecb7e0>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7d717acc3100>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7d717acc3100>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7d717ad6eca0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7d717ab636a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7d717aecb1a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7d717ad0efc0>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7d717ad0efc0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7d717aecb7e0>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7d717ad8c180>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7d717ad8dee0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7d717ad6ee80>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7d717ad48c20>, <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7d717aecaca0>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7d717aeefce0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7d717b313ce0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7d717ad6c5e0>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7d717ad21440>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7d717ad6fc40>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7d717ad4ad40>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7d717aeed580>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7d717ad228e0>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7d717ac92980>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7d717adb5b20>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7d717ace9e40>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7d717ad8ede0>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7d717ad8c400>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7d717ab60180>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7d717ad6c540>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7d717ad6c540>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7d717aeef7e0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7d717ad4bc40>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7d717aeca8e0>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7d717aeca8e0>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7d717aeca840>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7d717aeca840>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7d717aecb880>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7d717aecb880>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7d717aeca660>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7d717aeca660>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7d717aecbe20>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7d717aecbe20>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7d717aeec180>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7d717aeec180>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7d717aeec220>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7d717aeec220>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7d717aeec540>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7d717aeec540>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7d717aeec900>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7d717aeec900>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7d717aeed120>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7d717aeed120>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7d717aeed1c0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7d717aeed4e0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7d717aeeda80>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7d717aeeda80>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7d717aeee3e0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7d717aecb1a0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7d717ad21620>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7d717aeedb20>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7d717aeedb20>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7d717aeee7a0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7d717aeedbc0>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7d717ad21620>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7d717aeee5c0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7d717aeedee0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7d717aeee340>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7d717aeee340>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7d717aeee980>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7d717aeeec00>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7d717aeeec00>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7d717aeeefc0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7d717aeef2e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7d717aeef2e0>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7d717aeef880>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7d717aeef880>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7d717aeedda0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7d717aeed9e0>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7d717ad21580>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7d717ad21580>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7d717aeefce0>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7d717aeeff60>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7d717ad20220>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7d717ad20900>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7d717ad20b80>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7d717ad20b80>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7d717ad21760>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7d717ad20cc0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7d717ad20cc0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7d717ad21300>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7d717ad21300>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7d717ad20860>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7d717ad20860>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7d717ad232e0>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7d717ad216c0>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7d717ac78cc0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7d717ac782c0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7d717ac782c0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7d717ac787c0>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7d717ac787c0>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7d717ac91f80>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7d717ac78cc0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7d717ac66b60>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7d717ac66b60>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7d717ac794e0>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7d717ac794e0>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7d717aca6e80>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7d717ac7a2a0>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7d717ac7a7a0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7d717ab61bc0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7d717ac7b100>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7d717ac7b100>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7d717aeedbc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7d717ac91b20>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7d717ac91b20>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7d717adb5080>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7d717aca5760>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7d717ac92020>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7d717ac92020>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6b60>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7d717ac93b00>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7d717ac93b00>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5a80>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7d717ab95620>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7d717aca5b20>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7d717ac93f60>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7d717ac93f60>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7d717ab30ea0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7d717aca4400>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7d717aca4400>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7d717ad4b600>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7d717aca6a20>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7d717aca6a20>, <OpOverload(op='aten.index_copy', overload='dimname')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb67a0>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7d717aca6e80>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7d717aca5b20>, <OpOverload(op='aten.std', overload='correction_names')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7d717aca7ba0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7d717aca7ba0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7d717ac92980>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7d717ad49c60>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7d717ad8c7c0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7d717ad219e0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7d717ad22f20>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7d717ab613a0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7d717aeef7e0>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7d717adb5080>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7d717aceb600>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7d717ad0d9e0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7d717ad0d9e0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7d717aecaca0>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7d717ace8b80>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7d717aeeefc0>, <OpOverload(op='aten.cat', overload='names')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7d717ad22020>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7d717ab30900>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7d717ac193a0>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7d717ac193a0>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7d717ac198a0>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7d717ac198a0>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7d717aca6ac0>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7d717ac19da0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7d717ac19da0>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7d717ac1a2a0>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7d717ad23d80>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7d717ac1b560>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7d717ac91f80>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7d717ac1ba60>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7d717ac1ba60>, <OpOverload(op='aten.std', overload='names_dim')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7d717ac1a7a0>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7d717ac1b060>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7d717ad0df80>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7d717ac1a7a0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7d717ac1aca0>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7d717ac1aca0>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7d717ac79da0>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7d717ac4d6c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7d717ac1b060>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7d717ac1bf60>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7d717ac19940>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7d717ac19940>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7d717ad6c5e0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7d717ac4c7c0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7d717ac4c7c0>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6de0>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7d717ac4ccc0>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7d717ac4ccc0>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7d717ac64d60>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7d717ac7a2a0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7d717ac4d1c0>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7d717ac4d1c0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7d717ac4d6c0>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7d717aecb2e0>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7d717ac65c60>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7d717aecb380>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7d717ac4e200>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7d717ac4e200>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7d717ac64860>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7d717b313ce0>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7d717ac64860>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7d717ac64d60>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7d717ac65260>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7d717ac65260>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7d717ac66ac0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7d717ac7a7a0>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7d717ac65760>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7d717aeedda0>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7d717ac65c60>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7d717ac66ac0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7d717ac66480>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7d717ac66480>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: <function ifftshift at 0x7d717abcee80>, <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7d717ab620c0>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7d717ac67380>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7d717ac67380>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5bc0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7d717adb63e0>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7d717adb63e0>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7d717ac1bf60>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7d717adb7f60>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7d717adb7f60>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7d717ac1a2a0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7d717adb4f40>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7d717ad23b00>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7d717ac18ea0>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7d717ac18ea0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7d717ac1b560>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7d717ac79da0>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7d717ac65760>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7d717aca6ac0>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7d717ad6cd60>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7d717aeed1c0>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x7d717ad23c40>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7d717aeeccc0>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7d717adb4cc0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7d717aeed9e0>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7d717ad20680>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7d717ad20680>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5da0>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7d717ad0cf40>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7d717ad0c180>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7d717ab609a0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7d717ad6fc40>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7d717ace8b80>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7d717ad0e3e0>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7d717ad0e340>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7d717ab30900>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7d717ad0f740>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7d717ad0f740>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7d717ab30c20>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7d717ab30c20>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7d717ab309a0>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7d717ab309a0>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7d717ad0ef20>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7d717ad0ef20>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7d717ace9620>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7d717ace9620>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7d717ad0f380>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7d717ad0f380>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7d717ad0dd00>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7d717ad0dd00>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7d717ad0c7c0>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7d717ad0c7c0>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <OpOverload(op='aten.item', overload='default')>: <function item at 0x7d717acc1bc0>, <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7d717ad6f2e0>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7d717aceb6a0>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7d717ab63880>}
- 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 0x7d717ad22b60>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7d717ad23a60>}
- 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.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96900e0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9708860>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cce00>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.is_nonzero', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ede0>, kernel=<OpOverload(op='aten.is_nonzero', overload='default')>), <OpOverload(op='aten.conv1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ef20>, kernel=<OpOverload(op='aten.conv1d', overload='default')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec9a0>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690680>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e160>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee5c0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef9c0>, kernel=<OpOverload(op='aten._test_string_default', overload='default')>), <OpOverload(op='aten.special_erfc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692020>, kernel=<OpOverload(op='aten.special_erfc', 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 0x7d70b96919e0>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.tile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d300>, kernel=<OpOverload(op='aten.tile', overload='default')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ccc0>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ca40>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten.divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692f20>, kernel=<OpOverload(op='aten.divide', overload='Scalar')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c7c0>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c680>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.einsum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fb240>, kernel=<OpOverload(op='aten.einsum', overload='default')>), <OpOverload(op='aten.max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c2c0>, kernel=<OpOverload(op='aten.max_pool1d', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b5d00>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='aten.special_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693920>, kernel=<OpOverload(op='aten.special_softmax', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd620>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98edda0>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709440>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692160>, kernel=<OpOverload(op='aten.fake_quantize_per_tensor_affine', overload='default')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fac00>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eede0>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f9da0>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten.scaled_dot_product_attention', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691620>, kernel=<OpOverload(op='aten.scaled_dot_product_attention', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b97096c0>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f8f40>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c132f420>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691760>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f91c0>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969fc40>, 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 0x7d70b970bec0>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed800>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970bba0>, kernel=<OpOverload(op='aten.align_as', overload='default')>), <OpOverload(op='aten.special_log_softmax', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970bce0>, 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 0x7d70b970b1a0>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b4c0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690860>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9708e00>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98efba0>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd120>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709ee0>, kernel=<OpOverload(op='aten.max', overload='names_dim')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6520>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a480>, kernel=<OpOverload(op='aten.special_erfinv', 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 0x7d70b9709e40>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e660>, 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 0x7d70b969e840>, kernel=<OpOverload(op='aten.special_exp2', 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 0x7d70be1b54e0>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ea20>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19b1a0>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19aca0>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19bd80>, 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 0x7d70be19b920>, 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 0x7d70b969f1a0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12d42c0>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d800>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6d40>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690cc0>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b60c0>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e2a0>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6f20>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ce00>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4720>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten.instance_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692840>, kernel=<OpOverload(op='aten.instance_norm', overload='default')>), <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7ce0>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7060>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690400>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.max_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6480>, kernel=<OpOverload(op='aten.max_pool3d', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b67a0>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc7c0>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>), <OpOverload(op='aten.embedding_bag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b47c0>, kernel=<OpOverload(op='aten.embedding_bag', overload='default')>), <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc360>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee980>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec2c0>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc0e0>, 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 0x7d70b98efb00>, 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 0x7d70b98efce0>, 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 0x7d70b98ed120>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96931a0>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>), <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fade0>, kernel=<OpOverload(op='aten.nll_loss_nd', 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 0x7d70b96cd8a0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693240>, kernel=<OpOverload(op='aten.smm', overload='default')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4540>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692d40>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691b20>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692980>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692520>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7d70b96920c0>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691e40>, kernel=<OpOverload(op='aten.embedding_sparse_backward', 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 0x7d70b96cd080>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691440>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96913a0>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691260>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e980>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690d60>, kernel=<OpOverload(op='aten.gradient', overload='tensorarray')>), <OpOverload(op='aten.cross_entropy_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690c20>, kernel=<OpOverload(op='aten.cross_entropy_loss', overload='default')>), <OpOverload(op='aten.sspaddmm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690a40>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690180>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6980>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969efc0>, 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 0x7d70b969eca0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969eb60>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e340>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969dd00>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d940>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>), <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d6c0>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.affine_grid_generator_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d4e0>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d440>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', 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 0x7d70b969cea0>, 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 0x7d70b969cc20>, kernel=<OpOverload(op='aten.to_mkldnn_backward', 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 0x7d70b96cd6c0>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c180>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693c40>, 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 0x7d70b9693a60>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>), <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693600>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693100>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692fc0>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970bc40>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <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 0x7d70b970aac0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eed40>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec040>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d260>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f420>, kernel=<OpOverload(op='c10d_functional.broadcast', overload='default')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b77e0>, kernel=<OpOverload(op='aten._test_check_tensor', overload='default')>), <OpOverload(op='aten.special_sinc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98efe20>, kernel=<OpOverload(op='aten.special_sinc', 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 0x7d70be199120>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970be20>, 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 0x7d70be19a2a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693880>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c132ed40>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten.less_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fafc0>, kernel=<OpOverload(op='aten.less_equal', overload='Tensor')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eea20>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed1c0>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b5120>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fa980>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c9a0>, kernel=<OpOverload(op='aten.special_log1p', 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 0x7d70b970b240>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692a20>, kernel=<OpOverload(op='aten.inverse', overload='default')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b97099e0>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7560>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b97089a0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709120>, 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 0x7d70be1b51c0>, 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 0x7d70c12d7ba0>, 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 0x7d70be1b6fc0>, 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 0x7d70b96cc5e0>, 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 0x7d70b969f740>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b97093a0>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d717bbfd260>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709580>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709260>, 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 0x7d70b96fa3e0>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <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 0x7d70b98ed580>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970ae80>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.true_divide', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b7e0>, kernel=<OpOverload(op='aten.true_divide', overload='Scalar')>), <OpOverload(op='aten.stft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969dee0>, kernel=<OpOverload(op='aten.stft', overload='default')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec720>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee7a0>, 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 0x7d70b9708d60>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12b59e0>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969fd80>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten.embedding_bag', overload='padding_idx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef100>, kernel=<OpOverload(op='aten.embedding_bag', overload='padding_idx')>), <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef7e0>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef560>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709c60>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eff60>, 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 0x7d70b970ba60>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>), <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d1c0>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b880>, 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 0x7d70b969f380>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef2e0>, kernel=<OpOverload(op='aten.cummaxmin_backward', 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 0x7d70b97098a0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef380>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692700>, kernel=<OpOverload(op='aten.size', overload='int')>), <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 0x7d70b98ed3a0>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19bf60>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed6c0>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970af20>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b600>, 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 0x7d70be1b6a20>, 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 0x7d70b970ad40>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef420>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed620>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7740>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b740>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e020>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96902c0>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.type_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed8a0>, kernel=<OpOverload(op='aten.type_as', overload='default')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970aa20>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be199300>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.greater_equal', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f8fe0>, kernel=<OpOverload(op='aten.greater_equal', overload='Tensor')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed940>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec4a0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98edd00>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6ca0>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b53a0>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eeca0>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96936a0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96937e0>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7380>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b65c0>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.greater', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d760>, kernel=<OpOverload(op='aten.greater', overload='Tensor')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98efa60>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef920>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='aten.grid_sampler', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec220>, kernel=<OpOverload(op='aten.grid_sampler', overload='default')>), <OpOverload(op='aten.concat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee8e0>, 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 0x7d70b969ff60>, kernel=<OpOverload(op='image.decode_webp', 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 0x7d70b98edf80>, 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 0x7d70b98ee840>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b63e0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <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 0x7d70b98ec680>, 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 0x7d70b96cc2c0>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19a020>, 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 0x7d70b98ec900>, 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 0x7d70b98ef6a0>, 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 0x7d70be1b49a0>, 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 0x7d70be1b5620>, 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 0x7d70be1b4360>, kernel=<OpOverload(op='aten.linalg_det', overload='default')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969cd60>, kernel=<OpOverload(op='aten.orgqr', overload='default')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6340>, kernel=<OpOverload(op='aten.rms_norm', overload='default')>), <OpOverload(op='aten.atleast_2d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98edbc0>, kernel=<OpOverload(op='aten.atleast_2d', overload='Sequence')>), <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 0x7d70b98ecd60>, 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 0x7d70b9708b80>, kernel=<OpOverload(op='aten._test_serialization_subcmul', 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 0x7d70be19a340>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709bc0>, 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 0x7d70b96f8c20>, kernel=<OpOverload(op='aten.cross', overload='default')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7f60>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7ba0>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7600>, kernel=<OpOverload(op='aten.avg_pool1d', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b5c60>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b380>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693560>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten.less', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e0c0>, kernel=<OpOverload(op='aten.less', overload='Tensor')>), <OpOverload(op='aten.bilinear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a2a0>, kernel=<OpOverload(op='aten.bilinear', overload='default')>), <OpOverload(op='aten.conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96932e0>, kernel=<OpOverload(op='aten.conv3d', 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 0x7d70b9709760>, 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 0x7d70be1b7e20>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be198fe0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98edb20>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a980>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <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 0x7d70b98ecae0>, 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 0x7d70b98ed300>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12d5120>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6c00>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693f60>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd300>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970afc0>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten.layer_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96909a0>, kernel=<OpOverload(op='aten.layer_norm', overload='default')>), <OpOverload(op='aten.max_pool1d_with_indices', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b5f80>, kernel=<OpOverload(op='aten.max_pool1d_with_indices', overload='default')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef4c0>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.concatenate', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b6a0>, kernel=<OpOverload(op='aten.concatenate', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eeb60>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a700>, 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 0x7d70b9693ba0>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e480>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eec00>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693380>, 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 0x7d70b969dbc0>, kernel=<OpOverload(op='aten._is_zerotensor', 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 0x7d70b9691c60>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ed40>, 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 0x7d70b969c360>, 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 0x7d70b969c720>, 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 0x7d70b9693420>, 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 0x7d70b9690360>, 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 0x7d70b96ccea0>, kernel=<OpOverload(op='image.decode_gif', 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 0x7d70b98ecfe0>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693e20>, kernel=<OpOverload(op='aten.arccosh', 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 0x7d70b969cae0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693d80>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969cf40>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7880>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969cb80>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>), <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690e00>, 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 0x7d70b9690ea0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc4a0>, kernel=<OpOverload(op='image.read_file', overload='default')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c0e0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.multiply', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c220>, kernel=<OpOverload(op='aten.multiply', overload='Tensor')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96904a0>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <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 0x7d70b9693ec0>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c040>, 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 0x7d70b9693740>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969dda0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c900>, 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 0x7d70b96cdda0>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f600>, 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 0x7d70b98ef1a0>, 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 0x7d70b9709b20>, 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 0x7d70b98efec0>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4860>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ecc20>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd4e0>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed760>, 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 0x7d70b96cc400>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970aca0>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.det', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fad40>, kernel=<OpOverload(op='aten.det', overload='default')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4680>, kernel=<OpOverload(op='aten.one_hot', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee660>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fa200>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <OpOverload(op='aten.special_erf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ece00>, kernel=<OpOverload(op='aten.special_erf', 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 0x7d70b969f880>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b76a0>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f8680>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed9e0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec360>, 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 0x7d70b98ecb80>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98efd80>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='aten.repeat_interleave', overload='self_int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970ade0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_int')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc220>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b79c0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed260>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f100>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7420>, kernel=<OpOverload(op='aten.special_i0', 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 0x7d70b98eee80>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4b80>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970ac00>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9708ea0>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a5c0>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b2e0>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12d7880>, 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 0x7d70b96f9440>, kernel=<OpOverload(op='aten.msort', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f9bc0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96f9580>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709940>, kernel=<OpOverload(op='aten._add_batch_dim', 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 0x7d70b9709800>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc720>, 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 0x7d70b96fa480>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d580>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96916c0>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten.softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1985e0>, kernel=<OpOverload(op='aten.softmax', overload='int')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19ba60>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b59e0>, kernel=<OpOverload(op='aten.index_select_backward', 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 0x7d70b96f82c0>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969fe20>, kernel=<OpOverload(op='_test.cat', overload='default')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed080>, kernel=<OpOverload(op='aten.linalg_eigh', 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 0x7d70b970b9c0>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a7a0>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19aa20>, kernel=<OpOverload(op='aten.kron', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691300>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>), <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19bec0>, kernel=<OpOverload(op='aten.to_dense_backward', 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 0x7d70b9708cc0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19bc40>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19ae80>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709da0>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691f80>, kernel=<OpOverload(op='aten.flipud', overload='default')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709620>, 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 0x7d70b97094e0>, 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 0x7d70b96fb060>, kernel=<OpOverload(op='aten.to_dense', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c5e0>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9708ae0>, kernel=<OpOverload(op='aten.cov', 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 0x7d70b96cc9a0>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='aten.max_pool2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9708900>, kernel=<OpOverload(op='aten.max_pool2d', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e520>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='aten.isclose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b68e0>, kernel=<OpOverload(op='aten.isclose', overload='default')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19b880>, kernel=<OpOverload(op='aten.argwhere', 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 0x7d70b969d8a0>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19b380>, kernel=<OpOverload(op='aten.is_inference', overload='default')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6e80>, 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 0x7d70b970b560>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969eac0>, 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 0x7d70b969e700>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e8e0>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b71a0>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6700>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>), <OpOverload(op='aten.group_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98efc40>, 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 0x7d70b98ec400>, kernel=<OpOverload(op='aten.pinverse', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970bd80>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee020>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed4e0>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.special_expm1', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a520>, kernel=<OpOverload(op='aten.special_expm1', overload='default')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969dc60>, kernel=<OpOverload(op='aten.argsort', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693b00>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a0c0>, kernel=<OpOverload(op='aten.chalf', overload='default')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b100>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a020>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96faf20>, kernel=<OpOverload(op='aten.log_softmax', overload='int')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6ac0>, 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 0x7d70be1b56c0>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692b60>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.subtract', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7d80>, kernel=<OpOverload(op='aten.subtract', overload='Tensor')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12ad080>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9709300>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691bc0>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691940>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12b4860>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691d00>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f920>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692e80>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eda80>, 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 0x7d70b96cc180>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b420>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a160>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970a660>, kernel=<OpOverload(op='aten.linalg_eigvalsh', 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 0x7d70c12afa60>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fb560>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec5e0>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', 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 0x7d70b970a200>, 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 0x7d70b96f99e0>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e7a0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969de40>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96927a0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691580>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9708a40>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c1288680>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692340>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d620>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692ca0>, kernel=<OpOverload(op='aten.combinations', 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 0x7d70b969d3a0>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690f40>, kernel=<OpOverload(op='aten._nnpack_available', overload='default')>), <OpOverload(op='aten.log_sigmoid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b72e0>, kernel=<OpOverload(op='aten.log_sigmoid', 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 0x7d70b96ccf40>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690900>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.pad', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690540>, kernel=<OpOverload(op='aten.pad', overload='default')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19ade0>, kernel=<OpOverload(op='aten.fix', overload='default')>), <OpOverload(op='aten.atleast_3d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690720>, kernel=<OpOverload(op='aten.atleast_3d', overload='Sequence')>), <OpOverload(op='image._is_compiled_against_turbo', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f4c0>, 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 0x7d70b970bf60>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970ab60>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693060>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6160>, 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 0x7d70b98eef20>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cca40>, kernel=<OpOverload(op='image.encode_png', 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 0x7d70be19ac00>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec860>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e3e0>, kernel=<OpOverload(op='aten.linalg_cholesky', 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 0x7d70b98eefc0>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eeac0>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970b060>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c400>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7240>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='aten.atleast_1d', overload='Sequence')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6b60>, kernel=<OpOverload(op='aten.atleast_1d', overload='Sequence')>), <OpOverload(op='aten.sym_size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b5760>, 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 0x7d70b9709d00>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef880>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690040>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b62a0>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6840>, kernel=<OpOverload(op='aten.diag', overload='default')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4ae0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b5080>, 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 0x7d70c15e8400>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.true_divide', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef740>, 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 0x7d70b98ec540>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4f40>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten.celu', overload='default')>: <function celu at 0x7d717abf4fe0>, <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7d717abf4fe0>, <OpOverload(op='aten.elu', overload='default')>: <function elu at 0x7d717abf56c0>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7d717abf56c0>, <OpOverload(op='aten.relu', overload='default')>: <function relu at 0x7d717abf5b20>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7d717abf5b20>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7d717abf6160>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7d717abf6160>, <OpOverload(op='aten.leaky_relu', overload='default')>: <function leaky_relu at 0x7d717abf5ee0>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7d717abf5ee0>, <OpOverload(op='aten.mish', overload='default')>: <function mish at 0x7d717abf6480>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7d717abf6480>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7d717abf7560>, <OpOverload(op='aten.selu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be19a520>, kernel=<OpOverload(op='aten.selu', overload='default')>), <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7d717abf7880>, <OpOverload(op='aten.softplus', overload='default')>: <function softplus at 0x7d717abf6e80>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7d717abf6e80>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7d717abf7560>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7d717abf7880>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7ec0>, 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 0x7d70be1b7920>, 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 0x7d70be1b7a60>, kernel=<OpOverload(op='aten.nll_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7d717aa28900>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7d717aa28cc0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7d717aa28cc0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7d717abf7240>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7d717abf7240>, <OpOverload(op='aten.hardtanh', overload='default')>: <function hardtanh at 0x7d717aa28f40>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7d717aa28f40>, <OpOverload(op='aten.gelu', overload='default')>: <function gelu at 0x7d717aa29800>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7d717aa29800>, <OpOverload(op='aten.prelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec180>, 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 0x7d70b98eccc0>, kernel=<OpOverload(op='aten.relu6', overload='default')>), <OpOverload(op='aten.glu', overload='default')>: <function glu at 0x7d717aa2a340>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7d717aa2a340>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ecea0>, 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 0x7d70b98edee0>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.pixel_shuffle', overload='default')>: <function pixel_shuffle at 0x7d717aa2ac00>, <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7d717aa2ac00>, <OpOverload(op='aten.pixel_unshuffle', overload='default')>: <function pixel_unshuffle at 0x7d717aa2ae80>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7d717aa2ae80>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7d717aa29940>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7d717aa2b1a0>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7d717aa2aca0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7d717aa2af20>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7d717aa28860>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7d717aa28860>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7d717aa51760>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7d717aa2b880>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7d717aa2b880>, <OpOverload(op='aten.logit', overload='default')>: <function logit at 0x7d717aa51760>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7d717aa2bc40>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7d717aa2bc40>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7d717aa2bf60>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7d717aa505e0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7d717aa505e0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7d717aa50ae0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7d717aa50ae0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7d717aa50fe0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7d717aa50fe0>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7d717aa513a0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7d717aa513a0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7d717aa518a0>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7d717aa518a0>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96911c0>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7d717aa51c60>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7d717aa52020>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7d717aa52020>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7d717aa52660>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7d717aa52660>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692660>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.unsqueeze', overload='default')>: <function unsqueeze at 0x7d717ad0e200>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.lerp', overload='Tensor')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='default')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969ec00>, 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 0x7d70b969ee80>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7d717ab32520>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7d717ab32520>, <OpOverload(op='aten.masked_fill', overload='Scalar')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill', overload='Tensor')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7d717ab31940>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7d717ab31080>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7d717ab31080>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7d717ab31940>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d120>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d080>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7d717ab319e0>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7d717ab319e0>, <OpOverload(op='aten.triu', overload='default')>: <function triu at 0x7d717ab31ee0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7d717ab31ee0>, <OpOverload(op='aten.tril', overload='default')>: <function tril at 0x7d717ab32160>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7d717ab32160>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7d717ab32700>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7d717ab32ac0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7d717ab32ac0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7d717ab32e80>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7d717ab32e80>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7d717ab33240>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7d717ab33240>, <OpOverload(op='aten.rad2deg', overload='default')>: <function rad2deg at 0x7d717ab33c40>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7d717ab33c40>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7d717ab60180>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7d717ab61760>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7d717ab31e40>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7d717ab60720>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7d717ab60720>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7d717ab619e0>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7d717ab609a0>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7d717ab33740>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7d717ab61c60>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7d717ab61c60>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7d717ab607c0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7d717ab60a40>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7d717ab63a60>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7d717ab60b80>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7d717ab60b80>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7d717ab60cc0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7d717ab61ee0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7d717ab61ee0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7d717ab60e00>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7d717ab60f40>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7d717ab94f40>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7d717ab61080>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7d717ab611c0>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7d717ab62160>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7d717ab61300>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7d717ab637e0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7d717ab623e0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7d717ab61440>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7d717ab61580>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7d717ab61580>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7d717ab616c0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7d717ab616c0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7d717ab61800>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7d717ab62340>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7d717ab61940>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7d717ab61940>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7d717ab62340>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7d717ab61a80>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7d717ab61a80>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7d717ab61bc0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7d717ab63560>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7d717ab61d00>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7d717ab61e40>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7d717ab61e40>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7d717ab62200>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7d717ab61f80>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7d717ab61f80>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7d717ab620c0>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7d717ab639c0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7d717ab639c0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7d717ab61260>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7d717ab61260>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7d717ab60fe0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7d717ab60d60>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7d717ab60ae0>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7d717ab60680>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7d717ab63880>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7d717ab625c0>, <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 0x7d717ab63420>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7d717ab62700>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7d717ab62700>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7d717ab62840>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7d717ab62980>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7d717ab62980>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7d717ab62ac0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7d717ab62ac0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7d717ab62c00>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7d717ab631a0>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7d717ab62660>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7d717ab62d40>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7d717ab62e80>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7d717ab628e0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7d717ab62fc0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7d717ab62fc0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7d717ab63100>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7d717ab63240>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7d717ab62b60>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7d717ab63380>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7d717ab634c0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7d717ab62de0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7d717ab632e0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7d717ab63600>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7d717ab63060>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7d717ab63740>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7d717ab60040>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7d717ab60040>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7d717ab60c20>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7d717ab60c20>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7d717ab61120>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7d717ab62ca0>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7d717ab61620>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7d717ab61b20>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7d717ab61b20>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7d717ab627a0>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7d717ab62020>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7d717ab63b00>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7d717ab60900>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7d717ab63c40>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7d717ab63c40>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7d717ab63d80>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7d717ab63ec0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7d717ab94040>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7d717ab94040>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7d717ab613a0>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7d717ab94180>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7d717ab942c0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7d717ab94400>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7d717ab61da0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7d717ab94540>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7d717ab94680>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7d717ab947c0>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7d717ab94900>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7d717ab63ba0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7d717ab63ba0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7d717ab94a40>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7d717ab94b80>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7d717ab63ce0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7d717ab63f60>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7d717ab94cc0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7d717ab94cc0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7d717ab94e00>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7d717ab94fe0>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7d717ab94fe0>, <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 0x7d717ab949a0>, <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 0x7d717ab949a0>, <OpOverload(op='aten.diagonal_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7d717ab94360>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7d717ab94360>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7d717ab95300>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7d717ab95300>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7d717ab95f80>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7d717ab95620>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <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 0x7d717ab95940>, <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 0x7d717ab95940>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7d717ab95c60>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7d717ab95c60>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7d717ab95f80>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7d717ab60ea0>, <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 0x7d717ab60ea0>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7d717ab962a0>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7d717ab962a0>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <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 0x7d717ab959e0>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7d717ab977e0>, <OpOverload(op='aten.complex', overload='default')>: <function complex at 0x7d717ab97600>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7b00>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.polar', overload='default')>: <function polar at 0x7d717ab97880>, <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7d717ab97880>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6de0>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7d717abcc360>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7c40>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7d717abcc400>, <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b74c0>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7d717abcc680>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b7100>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7d717abcc900>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b6660>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7d717abccb80>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7d717abcdb20>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70be1b4d60>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7d717abcd440>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ed440>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef060>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7d717abcd6c0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef240>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7d717abcd940>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ee700>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7d717abcdbc0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ef600>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7d717abce2a0>, <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98eca40>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7d717abce520>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ede40>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7d717abccae0>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98edc60>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7d717abce5c0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ec0e0>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7d717abce840>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fab60>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7d717abceac0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fb2e0>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7d717abced40>, <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fa660>, kernel=<OpOverload(op='aten.fft_fftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='default')>: <function cross at 0x7d717abcf560>, <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7d717abcf560>, <OpOverload(op='aten.linalg_vector_norm', overload='default')>: <function vector_norm at 0x7d717abcf920>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7d717abcf920>, <OpOverload(op='aten.special_erfcx', overload='default')>: <function erfcx at 0x7d717aa2bf60>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7d717abf4d60>, <OpOverload(op='aten.lerp', overload='Scalar')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e5c0>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7d717ac4c2c0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7d717ac4c2c0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.logical_or', overload='default')>: <function logical_or at 0x7d717aca5d00>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7d717ac4dd00>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7d717ac4dd00>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7d717ab32700>, <OpOverload(op='aten.frac', overload='default')>: <function frac at 0x7d717ac4e700>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7d717ac4e700>, <OpOverload(op='aten.isinf', overload='default')>: <function isinf at 0x7d717ac4f100>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7d717ac4f100>, <OpOverload(op='aten.isposinf', overload='default')>: <function isposinf at 0x7d717ac4d760>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7d717ac4d760>, <OpOverload(op='aten.isneginf', overload='default')>: <function isneginf at 0x7d717ac4f420>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7d717ac4f420>, <OpOverload(op='aten.isnan', overload='default')>: <function isnan at 0x7d717ac4f920>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7d717ac4f920>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7d717ac64360>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7d717ac64360>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7d717ac791c0>, <OpOverload(op='aten.logsumexp', overload='default')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9693ce0>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7d717ac65800>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7d717ac65800>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7d717aa2b060>, <OpOverload(op='aten.sigmoid', overload='default')>: <function sigmoid at 0x7d717ac67880>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7d717ac67880>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7d717ab62e80>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7d717ac67d80>, <OpOverload(op='aten.sinc', overload='default')>: <function sinc at 0x7d717ac791c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7d717ab332e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.heaviside', overload='default')>: <function heaviside at 0x7d717ac936a0>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7d717ac936a0>, <OpOverload(op='aten.arange', overload='start_step')>: <function arange at 0x7d717ad0f9c0>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7d717ab97600>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7d717aca49a0>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7d717aca49a0>, <OpOverload(op='aten.logaddexp', overload='default')>: <function logaddexp at 0x7d717aca5260>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7d717aca5260>, <OpOverload(op='aten.logaddexp2', overload='default')>: <function logaddexp2 at 0x7d717aca56c0>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7d717aca56c0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7d717aca5760>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7d717aca5d00>, <OpOverload(op='aten.logical_xor', overload='default')>: <function logical_xor at 0x7d717aca6160>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7d717aca6160>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7d717ace8220>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7d717ab62480>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.xlogy', overload='Tensor')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='Scalar_Other')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='Scalar_Self')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.addcdiv', overload='default')>: <function addcdiv at 0x7d717acc0ea0>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7d717acc0ea0>, <OpOverload(op='aten.addcmul', overload='default')>: <function addcmul at 0x7d717acc1260>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7d717acc1260>, <OpOverload(op='aten.clamp', overload='default')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='Tensor')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp_min', overload='default')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='Tensor')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_max', overload='default')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='Tensor')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.all', overload='dim')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dims')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.any', overload='default')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dim')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dims')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692200>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691080>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691120>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96914e0>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690fe0>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96922a0>, 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 0x7d70b9691da0>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.mean', overload='default')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='dim')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691800>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690b80>, 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 0x7d70b9690ae0>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96905e0>, 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 0x7d70b96907c0>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.addr', overload='default')>: <function addr at 0x7d717ace85e0>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7d717ace85e0>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7d717acc36a0>, <OpOverload(op='aten.native_layer_norm', overload='default')>: <function native_layer_norm at 0x7d717acc36a0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.constant_pad_nd', overload='default')>: <function constant_pad_nd at 0x7d717ace96c0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7d717ace96c0>, <OpOverload(op='aten.index_select', overload='default')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.native_group_norm', overload='default')>: <function native_group_norm at 0x7d717acea200>, <OpOverload(op='aten.expand', overload='default')>: <function expand at 0x7d717ace98a0>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7d717ace9120>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96939c0>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.flip', overload='default')>: <function flip at 0x7d717ace9ee0>, <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7d717ace9ee0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7d717aceaa20>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7d717ace8220>, <OpOverload(op='aten.repeat', overload='default')>: <function repeat at 0x7d717aceade0>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7d717aceade0>, <OpOverload(op='aten.roll', overload='default')>: <function roll at 0x7d717aceb420>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7d717aceb420>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7d717aceb6a0>, <OpOverload(op='aten.stack', overload='default')>: <function stack at 0x7d717aceb9c0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7d717aceb9c0>, <OpOverload(op='aten.split_with_sizes', overload='default')>: <function split_with_sizes at 0x7d717aceb380>, <OpOverload(op='aten.unbind', overload='int')>: <function unbind at 0x7d717ad0c040>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7d717ad0c040>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c540>, 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 0x7d70b969c4a0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7d717ad0c5e0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.ones', overload='default')>: <function ones at 0x7d717ad0d580>, <OpOverload(op='aten.t', overload='default')>: <function t at 0x7d717ad0d760>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7d717ad0cf40>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7d717ad0c180>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7d717ad0d260>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7d717ad0d620>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7d717ad0d620>, <OpOverload(op='aten.alias', overload='default')>: <function alias at 0x7d717ad0d940>, <OpOverload(op='aten.unfold', overload='default')>: <function unfold at 0x7d717ad0db20>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7d717ad0f9c0>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7d717ad0df80>, <OpOverload(op='aten.cumsum', overload='default')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692c00>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7d717ad216c0>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7d717ad22020>, <OpOverload(op='aten._softmax', overload='default')>: <function _softmax at 0x7d717ad223e0>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7d717ad223e0>, <OpOverload(op='aten._log_softmax', overload='default')>: <function _log_softmax at 0x7d717ad22660>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7d717ad22660>, <OpOverload(op='aten.embedding', overload='default')>: <function embedding at 0x7d717ad228e0>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7d717ad22e80>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7d717ad22e80>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7d717ad22b60>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7d717ad22b60>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7d717ad22f20>, <OpOverload(op='aten.unsafe_split', overload='Tensor')>: <function unsafe_split at 0x7d717ad23060>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7d717ad231a0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7d717ad23740>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7d717ad23f60>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7d717ad23f60>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='default')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7d717ad237e0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7d717ad21440>, <OpOverload(op='aten.addmv', overload='default')>: <function addmv at 0x7d717ad236a0>, <OpOverload(op='aten._native_batch_norm_legit_no_training', overload='default')>: <function _native_batch_norm_legit_no_training at 0x7d717ad489a0>, <OpOverload(op='aten._native_batch_norm_legit_functional', overload='default')>: <function _native_batch_norm_legit_functional at 0x7d717ad48cc0>, <OpOverload(op='aten.native_batch_norm', overload='default')>: <function native_batch_norm at 0x7d717ad485e0>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7d717ad485e0>, <OpOverload(op='aten._native_batch_norm_legit', overload='default')>: <function _native_batch_norm_legit at 0x7d717ad48a40>, <OpOverload(op='aten._native_batch_norm_legit', overload='no_stats')>: <function _native_batch_norm_legit_no_stats at 0x7d717ad48b80>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7d717ad48f40>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7d717ad49120>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7d717ad48fe0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7d717ad49b20>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7d717ad49d00>, <OpOverload(op='aten._to_copy', overload='default')>: <function _to_copy at 0x7d717ad48c20>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7d717ad49ee0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7d717ad49da0>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7d717ad49b20>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7d717ad4a0c0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7d717ad4a0c0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7d717ad4ad40>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7d717ad4b1a0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7d717ad4b1a0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7d717ad6cea0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7d717ad49440>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7d717ad49440>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7d717ad49f80>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969cfe0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.upsample_nearest2d', overload='default')>: <function upsample_nearest2d at 0x7d717ad6cea0>, <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7d717ad4b560>, <OpOverload(op='aten.upsample_nearest1d', overload='vec')>: <function _upsample_nearest_vec at 0x7d717ad6cd60>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7d717ad4bc40>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7d717ad6c400>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten.upsample_nearest1d', overload='default')>: <function upsample_nearest1d at 0x7d717ad6d580>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7d717ad49c60>, <OpOverload(op='aten.upsample_nearest3d', overload='vec')>: <function _upsample_nearest_vec at 0x7d717ad6cd60>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7d717ad6da80>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7d717ad6d580>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten.upsample_nearest3d', overload='default')>: <function upsample_nearest3d at 0x7d717ad6d760>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7d717ad6d760>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7d717ad6da80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7d717ad6e2a0>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7d717ad6e480>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7d717ad6e660>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7d717ad6e840>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7d717ad6e7a0>, <OpOverload(op='aten.gru', overload='input')>: <function gru_impl at 0x7d717ad6f380>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7d717ad6f1a0>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7d717ad6f560>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7d717ad6ede0>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7d717ad6f740>, <OpOverload(op='aten.upsample_linear1d', overload='default')>: <function upsample_linear1d at 0x7d717ad6fe20>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7d717ad8c400>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7d717ad6fe20>, <OpOverload(op='aten._unsafe_index', overload='Tensor')>: <function _unsafe_index at 0x7d717ad6f4c0>, <OpOverload(op='aten.upsample_bilinear2d', overload='default')>: <function upsample_bilinear2d at 0x7d717ad8c180>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7d717ad6fc40>, <OpOverload(op='aten.upsample_bicubic2d', overload='vec')>: <function upsample_bicubic2d_vec at 0x7d717ad8da80>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7d717ad6c720>, <OpOverload(op='aten.nll_loss_forward', overload='default')>: <function nll_loss_forward at 0x7d717ad6c720>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7d717ad8d080>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7d717ad8d080>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7d717ad8db20>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7d717ad8db20>, <OpOverload(op='aten.grid_sampler_2d', overload='default')>: <function grid_sampler_2d at 0x7d717ad8dee0>, <OpOverload(op='aten.mv', overload='default')>: <function mv at 0x7d717ad8e200>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7d717ad8e200>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7d717ad8e480>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7d717ad8e480>, <OpOverload(op='aten.upsample_bicubic2d', overload='default')>: <function upsample_bicubic2d_default at 0x7d717ad6e5c0>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7d717ad6e5c0>, <OpOverload(op='aten.reflection_pad1d', overload='default')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad2d', overload='default')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad3d', overload='default')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.replication_pad1d', overload='default')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7d717ad8ede0>, <OpOverload(op='aten.replication_pad2d', overload='default')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad3d', overload='default')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten._scaled_dot_product_flash_attention_for_cpu', overload='default')>: <function scaled_dot_product_flash_attention_for_cpu at 0x7d717adb4540>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7d717ad8f2e0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f420>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f420>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7d717adb42c0>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7d717ad8fc40>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7d717ad8fc40>, <OpOverload(op='aten.arange', overload='default')>: <function arange_default at 0x7d717adb42c0>, <OpOverload(op='aten.arange', overload='start')>: <function arange_start at 0x7d717adb40e0>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7d717ad8ff60>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7d717ad8ff60>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7d717ad8f2e0>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='default')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb62a0>, <OpOverload(op='aten.floor_divide', overload='default')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb62a0>, <OpOverload(op='aten.bernoulli', overload='default')>: <function bernoulli at 0x7d717adb5440>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7d717ac67d80>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7d717adb51c0>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7d717adb51c0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7d717adb5b20>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6480>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7d717adb5940>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7d717adb5940>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5800>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5e40>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb60c0>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb60c0>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6340>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6200>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6020>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6020>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5580>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5580>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5260>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5260>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4fe0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4fe0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4b80>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4220>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6660>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb68e0>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb68e0>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6ca0>, <OpOverload(op='aten.dot', overload='default')>: <function dot at 0x7d717ab31e40>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7d717ad4b560>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7d717ad4a660>, <OpOverload(op='aten.where', overload='self')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.where', overload='ScalarOther')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96925c0>, kernel=<OpOverload(op='aten.where', overload='ScalarOther')>), <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.where', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96923e0>, kernel=<OpOverload(op='aten.where', overload='Scalar')>), <OpOverload(op='aten.where', overload='ScalarSelf')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96928e0>, kernel=<OpOverload(op='aten.where', overload='ScalarSelf')>), <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692480>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.normal', overload='float_float')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7d717ad23a60>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7d717ad0d260>, <OpOverload(op='aten.sum', overload='default')>: <function sum_default at 0x7d717adb4f40>, <OpOverload(op='aten.sum', overload='dim_IntList')>: <function sum at 0x7d717acc2ca0>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7d717acc2ca0>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9690220>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7d717ad236a0>, <OpOverload(op='aten.prod', overload='dim_int')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='default')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969da80>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5d00>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7d717aeeff60>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969e200>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969df80>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969d9e0>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5f80>, <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969db20>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.var', overload='out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.amax', overload='default')>: <function amax at 0x7d717acc3240>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7d717ad4a660>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7d717ab63420>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7d717acc3240>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7d717aecb7e0>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7d717acc3100>, <OpOverload(op='aten.amin', overload='default')>: <function amin at 0x7d717acc3100>, <OpOverload(op='aten.lstm', overload='input')>: <function lstm_impl at 0x7d717ad6eca0>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7d717ab636a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7d717aecb1a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7d717ad0efc0>, <OpOverload(op='aten.full', overload='default')>: <function full at 0x7d717ad0efc0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7d717aecb7e0>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7d717ad8c180>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7d717ad8dee0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7d717ad6ee80>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7d717ad48c20>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691ee0>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7d717aecaca0>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7d717aeefce0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='Tensor_float')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7d717b313ce0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7d717ad6c5e0>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7d717ad21440>, <OpOverload(op='aten.upsample_trilinear3d', overload='vec')>: <function _upsample_linear_vec at 0x7d717ad6fc40>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7d717ad4ad40>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7d717aeed580>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7d717ad228e0>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7d717ac92980>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7d717adb5b20>, <OpOverload(op='aten.permute', overload='default')>: <function permute at 0x7d717ace9e40>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7d717ace8cc0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7d717ad8ede0>, <OpOverload(op='aten.upsample_trilinear3d', overload='default')>: <function upsample_trilinear3d at 0x7d717ad8c400>, <OpOverload(op='aten.deg2rad', overload='default')>: <function deg2rad at 0x7d717ab60180>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7d717ad6c540>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7d717ad6c540>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7d717aeef7e0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7d717ad4bc40>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7d717aeca8e0>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7d717aeca8e0>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7d717aeca840>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7d717aeca840>, <OpOverload(op='aten.hardsigmoid', overload='default')>: <function hardsigmoid at 0x7d717aecb880>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7d717aecb880>, <OpOverload(op='aten.hardtanh_backward', overload='default')>: <function hardtanh_backward at 0x7d717aeca660>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7d717aeca660>, <OpOverload(op='aten.hardswish', overload='default')>: <function hardswish at 0x7d717aecbe20>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7d717aecbe20>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7d717aeec180>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7d717aeec180>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7d717aeec220>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7d717aeec220>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7d717aeec540>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7d717aeec540>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7d717aeec900>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7d717aeec900>, <OpOverload(op='aten.silu', overload='default')>: <function silu at 0x7d717aeed120>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7d717aeed120>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7d717aeed1c0>, <OpOverload(op='aten._prelu_kernel', overload='default')>: <function _prelu_kernel at 0x7d717aeed4e0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7d717aeeda80>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7d717aeeda80>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7d717aeee3e0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7d717aecb1a0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7d717ad21620>, <OpOverload(op='aten.mse_loss', overload='default')>: <function mse_loss at 0x7d717aeedb20>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7d717aeedb20>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7d717aeee7a0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7d717aeedbc0>, <OpOverload(op='aten.im2col', overload='default')>: <function im2col at 0x7d717ad21620>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7d717aeee5c0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7d717aeedee0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7d717aeee340>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7d717aeee340>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7d717aeee980>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7d717aeeec00>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7d717aeeec00>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7d717aeeefc0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7d717aeef2e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7d717aeef2e0>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7d717aeef880>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7d717aeef880>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7d717aeedda0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7d717aeed9e0>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7d717ad21580>, <OpOverload(op='aten.col2im', overload='default')>: <function col2im at 0x7d717ad21580>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7d717aeefce0>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7d717aeeff60>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7d717ad20220>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7d717ad20900>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7d717ad20900>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7d717ad20b80>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7d717ad20b80>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7d717ad21760>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7d717ad20cc0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7d717ad20cc0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7d717ad21300>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7d717ad21300>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7d717ad20860>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7d717ad20860>, <OpOverload(op='aten.split', overload='Tensor')>: <function split at 0x7d717ad232e0>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7d717ad216c0>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7d717ac78cc0>, <OpOverload(op='aten.sign', overload='default')>: <function sign at 0x7d717ac782c0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7d717ac782c0>, <OpOverload(op='aten.bitwise_xor', overload='Tensor')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7d717ac787c0>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7d717ac787c0>, <OpOverload(op='aten.pow', overload='Scalar')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7d717ac91f80>, <OpOverload(op='aten.sin', overload='default')>: <function sin at 0x7d717ac78cc0>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7d717ac66b60>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.sinh', overload='default')>: <function sinh at 0x7d717ac66b60>, <OpOverload(op='aten.sqrt', overload='default')>: <function sqrt at 0x7d717ac794e0>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7d717ac794e0>, <OpOverload(op='aten.bitwise_and', overload='Scalar')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7d717aca6e80>, <OpOverload(op='aten.tanh', overload='default')>: <function tanh at 0x7d717ac7a2a0>, <OpOverload(op='aten.trunc', overload='default')>: <function trunc at 0x7d717ac7a7a0>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7d717ab61bc0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.add', overload='Scalar')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7d717ac7b100>, <OpOverload(op='aten.atan2', overload='default')>: <function atan2 at 0x7d717ac7b100>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_and', overload='Tensor')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7d717aeedbc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_or', overload='Scalar')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Tensor')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.div', overload='Tensor_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.eq', overload='Scalar')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7d717ac91b20>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7d717ac91b20>, <OpOverload(op='aten.fmod', overload='Scalar')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.fmod', overload='Tensor')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.squeeze', overload='dim')>: <function squeeze_default at 0x7d717adb5080>, <OpOverload(op='aten.logical_not', overload='default')>: <function logical_not at 0x7d717aca5760>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7d717ac92020>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7d717ac92020>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6b60>, <OpOverload(op='aten.ge', overload='Scalar')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Tensor')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.gt', overload='Scalar')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Tensor')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7d717ac93b00>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7d717ac93b00>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5a80>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7d717ab95620>, <OpOverload(op='aten.logical_and', overload='default')>: <function logical_and at 0x7d717aca5b20>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7d717ac93f60>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7d717ac93f60>, <OpOverload(op='aten.randn', overload='default')>: <function randn at 0x7d717ab30ea0>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7d717aca4400>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7d717aca4400>, <OpOverload(op='aten.le', overload='Scalar')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.le', overload='Tensor')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7d717ad4b600>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.lt', overload='Tensor')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7d717aca6a20>, <OpOverload(op='aten.maximum', overload='default')>: <function maximum at 0x7d717aca6a20>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969c860>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb67a0>, <OpOverload(op='aten.minimum', overload='default')>: <function minimum at 0x7d717aca6e80>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7d717aca5b20>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96918a0>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.ne', overload='Tensor')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7d717aca7ba0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7d717aca7ba0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7d717ac92980>, <OpOverload(op='aten.remainder', overload='Tensor')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.remainder', overload='Scalar')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7d717ad49c60>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.sub', overload='Scalar')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.sub', overload='Tensor')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7d717ad20220>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7d717ad8c7c0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.dropout', overload='default')>: <function dropout at 0x7d717ad219e0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7d717ad22f20>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7d717ab613a0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7d717aeef7e0>, <OpOverload(op='aten.squeeze', overload='default')>: <function squeeze_default at 0x7d717adb5080>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7d717aceb600>, <OpOverload(op='aten.transpose', overload='int')>: <function transpose at 0x7d717ad0d9e0>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7d717ad0d9e0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7d717aecaca0>, <OpOverload(op='aten.all', overload='default')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7d717ace8b80>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7d717aeeefc0>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96934c0>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.native_dropout', overload='default')>: <function native_dropout at 0x7d717ad22020>, <OpOverload(op='aten.cat', overload='default')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7d717ab30900>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7d717ac193a0>, <OpOverload(op='aten.asin', overload='default')>: <function asin at 0x7d717ac193a0>, <OpOverload(op='aten.asinh', overload='default')>: <function asinh at 0x7d717ac198a0>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7d717ac198a0>, <OpOverload(op='aten.clone', overload='default')>: <function clone at 0x7d717aca6ac0>, <OpOverload(op='aten.atan', overload='default')>: <function atan at 0x7d717ac19da0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7d717ac19da0>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7d717ac1a2a0>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7d717ad23d80>, <OpOverload(op='aten.cos', overload='default')>: <function cos at 0x7d717ac1b560>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7d717ac91f80>, <OpOverload(op='aten.cosh', overload='default')>: <function cosh at 0x7d717ac1ba60>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7d717ac1ba60>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9691a80>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.bitwise_not', overload='default')>: <function bitwise_not at 0x7d717ac1a7a0>, <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7d717ac1b060>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7d717ad0df80>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7d717ac1a7a0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7d717ac1aca0>, <OpOverload(op='aten.ceil', overload='default')>: <function ceil at 0x7d717ac1aca0>, <OpOverload(op='aten.tan', overload='default')>: <function tan at 0x7d717ac79da0>, <OpOverload(op='aten.exp2', overload='default')>: <function exp2 at 0x7d717ac4d6c0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7d717ac1b060>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7d717ac1bf60>, <OpOverload(op='aten.erf', overload='default')>: <function erf at 0x7d717ac19940>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7d717ac19940>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7d717ad6c5e0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7d717ac4c7c0>, <OpOverload(op='aten.erfc', overload='default')>: <function erfc at 0x7d717ac4c7c0>, <OpOverload(op='aten.normal', overload='float_Tensor')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6de0>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7d717ac4ccc0>, <OpOverload(op='aten.exp', overload='default')>: <function exp at 0x7d717ac4ccc0>, <OpOverload(op='aten.log', overload='default')>: <function log at 0x7d717ac64d60>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7d717ac7a2a0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7d717ac4d1c0>, <OpOverload(op='aten.expm1', overload='default')>: <function expm1 at 0x7d717ac4d1c0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7d717ac4d6c0>, <OpOverload(op='aten.fill', overload='Scalar')>: <function fill_scalar at 0x7d717aecb2e0>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7d717ac65c60>, <OpOverload(op='aten.matmul', overload='default')>: <function matmul at 0x7d717ad8e840>, <OpOverload(op='aten.fill', overload='Tensor')>: <function fill_tensor at 0x7d717aecb380>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7d717ac4e200>, <OpOverload(op='aten.floor', overload='default')>: <function floor at 0x7d717ac4e200>, <OpOverload(op='aten.isfinite', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70c12d7a60>, kernel=<OpOverload(op='aten.isfinite', overload='default')>), <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7d717ac64860>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7d717b313ce0>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7d717ac64860>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7d717ac64d60>, <OpOverload(op='aten.bitwise_xor', overload='Scalar')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.pow', overload='Tensor_Scalar')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.log1p', overload='default')>: <function log1p at 0x7d717ac65260>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7d717ac65260>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7d717ac66ac0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7d717ac7a7a0>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7d717ac65760>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7d717aeedda0>, <OpOverload(op='aten.log10', overload='default')>: <function log10 at 0x7d717ac65c60>, <OpOverload(op='aten.round', overload='decimals')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.reciprocal', overload='default')>: <function reciprocal at 0x7d717ac66ac0>, <OpOverload(op='aten.neg', overload='default')>: <function neg at 0x7d717ac66480>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7d717ac66480>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96fa020>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.ne', overload='Scalar')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.lt', overload='Scalar')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.round', overload='default')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7d717ab620c0>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7d717ac67380>, <OpOverload(op='aten.rsqrt', overload='default')>: <function rsqrt at 0x7d717ac67380>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5bc0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7d717adb63e0>, <OpOverload(op='aten.abs', overload='default')>: <function abs at 0x7d717adb63e0>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7d717ac1bf60>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7d717adb7f60>, <OpOverload(op='aten.acos', overload='default')>: <function acos at 0x7d717adb7f60>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.atanh', overload='default')>: <function atanh at 0x7d717ac1a2a0>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7d717adb4f40>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7d717ad23b00>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7d717ac18ea0>, <OpOverload(op='aten.acosh', overload='default')>: <function acosh at 0x7d717ac18ea0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7d717ac1b560>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7d717ac79da0>, <OpOverload(op='aten.log2', overload='default')>: <function log2 at 0x7d717ac65760>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7d717aca6ac0>, <OpOverload(op='aten.eq', overload='Tensor')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b970bb00>, 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 0x7d70b970b920>, kernel=<OpOverload(op='aten.conv2d', overload='default')>), <OpOverload(op='aten.mul', overload='Tensor')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.div', overload='Tensor')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.upsample_nearest2d', overload='vec')>: <function _upsample_nearest_vec at 0x7d717ad6cd60>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7d717aeed1c0>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x7d717ad23c40>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7d717aeeccc0>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7d717adb4cc0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7d717aeed9e0>, <OpOverload(op='aten.slice_scatter', overload='default')>: <function slice_scatter at 0x7d717ad20680>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7d717ad20680>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5da0>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7d717ad0cf40>, <OpOverload(op='aten.diagonal', overload='default')>: <function diagonal at 0x7d717ad0c180>, <OpOverload(op='aten.select_scatter', overload='default')>: <function select_scatter at 0x7d717ab609a0>, <OpOverload(op='aten.upsample_bilinear2d', overload='vec')>: <function _upsample_linear_vec at 0x7d717ad6fc40>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7d717ace8b80>, <OpOverload(op='aten.detach', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.zeros', overload='default')>: <function zeros at 0x7d717ad0e3e0>, <OpOverload(op='aten._unsafe_view', overload='default')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.view', overload='default')>: <function view at 0x7d717ad0e340>, <OpOverload(op='aten.empty_strided', overload='default')>: <function empty_strided at 0x7d717ab30900>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.empty_like', overload='default')>: <function empty_like at 0x7d717ad0f740>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7d717ad0f740>, <OpOverload(op='aten.ones_like', overload='default')>: <function ones_like at 0x7d717ab30c20>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7d717ab30c20>, <OpOverload(op='aten.zeros_like', overload='default')>: <function zeros_like at 0x7d717ab309a0>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7d717ab309a0>, <OpOverload(op='aten.new_empty', overload='default')>: <function new_empty at 0x7d717ad0ef20>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7d717ad0ef20>, <OpOverload(op='aten.new_empty_strided', overload='default')>: <function new_empty_strided at 0x7d717ace9620>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7d717ace9620>, <OpOverload(op='aten.new_full', overload='default')>: <function new_full at 0x7d717ad0f380>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7d717ad0f380>, <OpOverload(op='aten.new_zeros', overload='default')>: <function new_zeros at 0x7d717ad0dd00>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7d717ad0dd00>, <OpOverload(op='aten.new_ones', overload='default')>: <function new_ones at 0x7d717ad0c7c0>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7d717ad0c7c0>, <OpOverload(op='aten.view_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9692de0>, 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 0x7d70b9692ac0>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten._unsafe_index_put', overload='default')>: <function _unsafe_index_put at 0x7d717ad6f2e0>, <OpOverload(op='aten.add', overload='Tensor')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.slice', overload='Tensor')>: <function slice_forward at 0x7d717ad20040>, <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7d717aceb6a0>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7d717ab63880>, <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96ccd60>, 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 0x7d70b96cda80>, 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 0x7d70b96cd580>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96ccc20>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f7e0>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969fb00>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f060>, 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 0x7d70b96cc540>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd3a0>, 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 0x7d70b96cd260>, 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 0x7d70b96cdc60>, 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 0x7d70b96cd940>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd760>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd800>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cccc0>, 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 0x7d70b969fce0>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f560>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f2e0>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f6a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc680>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969fa60>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc900>, 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 0x7d70b969f9c0>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd9e0>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96ccfe0>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc860>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96ccb80>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cdbc0>, kernel=<OpOverload(op='quantized.linear_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 0x7d70b96ccae0>, 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 0x7d70b969fba0>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969f240>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cdb20>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cc040>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd1c0>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b969fec0>, kernel=<OpOverload(op='quantized.linear_unpack_fp16', overload='default')>), <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cd440>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b96cdd00>, 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.frobenius_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046de0>, kernel=<OpOverload(op='aten.frobenius_norm', overload='dim')>), <OpOverload(op='aten.divide', overload='Tensor_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030f40>, kernel=<OpOverload(op='aten.divide', overload='Tensor_mode')>), <OpOverload(op='image.decode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057b00>, kernel=<OpOverload(op='image.decode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.logcumsumexp', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903dee0>, kernel=<OpOverload(op='aten.logcumsumexp', overload='dimname')>), <OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047380>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='tol_tensor')>), <OpOverload(op='aten.special_expit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054ea0>, kernel=<OpOverload(op='aten.special_expit', overload='default')>), <OpOverload(op='aten.can_cast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f2e0>, kernel=<OpOverload(op='aten.can_cast', overload='default')>), <OpOverload(op='aten._test_string_default', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e520>, kernel=<OpOverload(op='aten._test_string_default', 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 0x7d70b904c720>, kernel=<OpOverload(op='aten.quantized_rnn_relu_cell', overload='default')>), <OpOverload(op='aten.output_nr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f9c0>, kernel=<OpOverload(op='aten.output_nr', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f740>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Scalar')>), <OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f4c0>, kernel=<OpOverload(op='aten._thnn_differentiable_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.cosine_embedding_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f380>, kernel=<OpOverload(op='aten.cosine_embedding_loss', overload='default')>), <OpOverload(op='aten.linalg_svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044e00>, kernel=<OpOverload(op='aten.linalg_svd', overload='default')>), <OpOverload(op='image.write_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070360>, kernel=<OpOverload(op='image.write_file', overload='default')>), <OpOverload(op='aten.cosine_similarity', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c900>, kernel=<OpOverload(op='aten.cosine_similarity', overload='default')>), <OpOverload(op='aten._pad_enum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030360>, kernel=<OpOverload(op='aten._pad_enum', overload='default')>), <OpOverload(op='aten.to_sparse_csc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033c40>, kernel=<OpOverload(op='aten.to_sparse_csc', overload='default')>), <OpOverload(op='aten.linalg_matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e3e0>, kernel=<OpOverload(op='aten.linalg_matrix_power', overload='default')>), <OpOverload(op='aten.scatter', overload='dimname_src')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033880>, kernel=<OpOverload(op='aten.scatter', overload='dimname_src')>), <OpOverload(op='aten.gradient', overload='scalarint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90305e0>, kernel=<OpOverload(op='aten.gradient', overload='scalarint')>), <OpOverload(op='aten.divide', overload='Scalar_mode')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90339c0>, kernel=<OpOverload(op='aten.divide', overload='Scalar_mode')>), <OpOverload(op='aten.adaptive_max_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b952e160>, kernel=<OpOverload(op='aten.adaptive_max_pool1d', overload='default')>), <OpOverload(op='aten.histogramdd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c4a0>, kernel=<OpOverload(op='aten.histogramdd', overload='default')>), <OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033600>, kernel=<OpOverload(op='aten._choose_qparams_per_tensor', overload='default')>), <OpOverload(op='image.encode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056980>, 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 0x7d70b90331a0>, kernel=<OpOverload(op='aten.embedding_backward', overload='default')>), <OpOverload(op='aten.concatenate', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c400>, kernel=<OpOverload(op='aten.concatenate', overload='names')>), <OpOverload(op='aten.align_as', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032e80>, kernel=<OpOverload(op='aten.align_as', overload='default')>), <OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032480>, kernel=<OpOverload(op='aten._validate_sparse_bsc_tensor_args', overload='default')>), <OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90327a0>, kernel=<OpOverload(op='aten.sparse_csr_tensor', overload='crow_col_value')>), <OpOverload(op='aten.take_along_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047560>, kernel=<OpOverload(op='aten.take_along_dim', overload='default')>), <OpOverload(op='aten._cast_Char', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90318a0>, kernel=<OpOverload(op='aten._cast_Char', overload='default')>), <OpOverload(op='aten.__and__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903fc40>, kernel=<OpOverload(op='aten.__and__', overload='Scalar')>), <OpOverload(op='profiler._record_function_enter', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057e20>, kernel=<OpOverload(op='profiler._record_function_enter', overload='default')>), <OpOverload(op='aten.max', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90311c0>, kernel=<OpOverload(op='aten.max', overload='names_dim')>), <OpOverload(op='aten.arcsin', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046ac0>, kernel=<OpOverload(op='aten.arcsin', overload='default')>), <OpOverload(op='aten.special_erfinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030d60>, kernel=<OpOverload(op='aten.special_erfinv', 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 0x7d70b9031120>, kernel=<OpOverload(op='aten._validate_sparse_bsr_tensor_args', overload='default')>), <OpOverload(op='aten.__or__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90553a0>, 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 0x7d70b9055580>, kernel=<OpOverload(op='aten.special_exp2', 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 0x7d70b9044c20>, kernel=<OpOverload(op='aten._cufft_set_plan_cache_max_size', overload='default')>), <OpOverload(op='aten.trapz', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055760>, kernel=<OpOverload(op='aten.trapz', overload='dx')>), <OpOverload(op='aten.special_logsumexp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030040>, kernel=<OpOverload(op='aten.special_logsumexp', overload='default')>), <OpOverload(op='aten.column_stack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bce0>, kernel=<OpOverload(op='aten.column_stack', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921b7e0>, 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 0x7d70b921b6a0>, 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 0x7d70b9055ee0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor', overload='default')>), <OpOverload(op='aten.flatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921b100>, kernel=<OpOverload(op='aten.flatten_dense_tensors', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='b')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054540>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='b')>), <OpOverload(op='aten.size', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90465c0>, kernel=<OpOverload(op='aten.size', overload='Dimname')>), <OpOverload(op='aten.conv_transpose1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90479c0>, kernel=<OpOverload(op='aten.conv_transpose1d', overload='default')>), <OpOverload(op='aten.linalg_ldl_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046980>, kernel=<OpOverload(op='aten.linalg_ldl_factor', overload='default')>), <OpOverload(op='aten.not_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054fe0>, kernel=<OpOverload(op='aten.not_equal', overload='Scalar')>), <OpOverload(op='aten.chain_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045ee0>, kernel=<OpOverload(op='aten.chain_matmul', overload='default')>), <OpOverload(op='aten.sort', overload='dimname_stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fb00>, kernel=<OpOverload(op='aten.sort', overload='dimname_stable')>), <OpOverload(op='aten.argsort', overload='stable')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044b80>, kernel=<OpOverload(op='aten.argsort', overload='stable')>), <OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90456c0>, kernel=<OpOverload(op='aten._thnn_differentiable_gru_cell_backward', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045300>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_float')>), <OpOverload(op='aten.corrcoef', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047100>, kernel=<OpOverload(op='aten.corrcoef', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044860>, kernel=<OpOverload(op='aten.nuclear_norm', overload='dim')>), <OpOverload(op='inductor._alloc_from_pool', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90574c0>, kernel=<OpOverload(op='inductor._alloc_from_pool', overload='default')>), <OpOverload(op='c10d_functional.wait_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057060>, kernel=<OpOverload(op='c10d_functional.wait_tensor', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c4a0>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='int')>), <OpOverload(op='aten.less_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903dd00>, kernel=<OpOverload(op='aten.less_equal', overload='Scalar')>), <OpOverload(op='image.decode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056de0>, 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 0x7d70b903d440>, 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 0x7d70b903d6c0>, 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 0x7d70b903d080>, kernel=<OpOverload(op='aten._validate_sparse_csr_tensor_args', overload='default')>), <OpOverload(op='aten.gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904dee0>, kernel=<OpOverload(op='aten.gru_cell', overload='default')>), <OpOverload(op='aten.nll_loss_nd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c180>, kernel=<OpOverload(op='aten.nll_loss_nd', 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 0x7d70b90705e0>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor_coalesced', overload='default')>), <OpOverload(op='aten.smm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904df80>, kernel=<OpOverload(op='aten.smm', overload='default')>), <OpOverload(op='aten.trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046b60>, kernel=<OpOverload(op='aten.trapezoid', overload='dx')>), <OpOverload(op='aten.slow_conv3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904da80>, kernel=<OpOverload(op='aten.slow_conv3d', overload='default')>), <OpOverload(op='aten.promote_types', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c860>, kernel=<OpOverload(op='aten.promote_types', overload='default')>), <OpOverload(op='aten.special_multigammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d6c0>, kernel=<OpOverload(op='aten.special_multigammaln', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d260>, kernel=<OpOverload(op='aten._sparse_sum', 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 0x7d70b904ce00>, kernel=<OpOverload(op='aten.fake_quantize_per_channel_affine_cachemask_backward', overload='default')>), <OpOverload(op='aten.embedding_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904cb80>, kernel=<OpOverload(op='aten.embedding_sparse_backward', 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 0x7d70b9057d80>, kernel=<OpOverload(op='c10d_functional.all_reduce_coalesced', overload='default')>), <OpOverload(op='aten.kl_div', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c180>, kernel=<OpOverload(op='aten.kl_div', overload='default')>), <OpOverload(op='aten._weight_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c0e0>, kernel=<OpOverload(op='aten._weight_norm', overload='default')>), <OpOverload(op='aten._gather_sparse_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047f60>, kernel=<OpOverload(op='aten._gather_sparse_backward', overload='default')>), <OpOverload(op='aten.to_sparse_bsr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90556c0>, kernel=<OpOverload(op='aten.to_sparse_bsr', overload='default')>), <OpOverload(op='aten.gradient', overload='tensorarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047a60>, 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 0x7d70b9047740>, kernel=<OpOverload(op='aten.sspaddmm', overload='default')>), <OpOverload(op='aten.nanquantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046e80>, kernel=<OpOverload(op='aten.nanquantile', overload='default')>), <OpOverload(op='aten.poisson_nll_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046c00>, kernel=<OpOverload(op='aten.poisson_nll_loss', overload='default')>), <OpOverload(op='aten.matrix_exp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055d00>, 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 0x7d70b90559e0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_size', overload='default')>), <OpOverload(op='aten.histogramdd', overload='TensorList_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90558a0>, kernel=<OpOverload(op='aten.histogramdd', overload='TensorList_bins')>), <OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055080>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight', overload='default')>), <OpOverload(op='aten.lu_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054a40>, kernel=<OpOverload(op='aten.lu_solve', overload='default')>), <OpOverload(op='aten._convolution_mode', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054680>, kernel=<OpOverload(op='aten._convolution_mode', overload='default')>), <OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054400>, kernel=<OpOverload(op='aten._thnn_fused_lstm_cell_backward', overload='default')>), <OpOverload(op='aten.affine_grid_generator_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054220>, kernel=<OpOverload(op='aten.affine_grid_generator_backward', overload='default')>), <OpOverload(op='aten._sparse_csc_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054180>, kernel=<OpOverload(op='aten._sparse_csc_tensor_unsafe', 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 0x7d70b904fba0>, 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 0x7d70b904f920>, kernel=<OpOverload(op='aten.to_mkldnn_backward', 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 0x7d70b9070400>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_conv2d', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='self_scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ee80>, kernel=<OpOverload(op='aten.special_xlogy', overload='self_scalar')>), <OpOverload(op='aten.stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e980>, 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 0x7d70b904e7a0>, kernel=<OpOverload(op='aten._shape_as_tensor', overload='default')>), <OpOverload(op='aten.norm_except_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e340>, kernel=<OpOverload(op='aten.norm_except_dim', overload='default')>), <OpOverload(op='aten.quantized_lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904de40>, kernel=<OpOverload(op='aten.quantized_lstm_cell', overload='default')>), <OpOverload(op='aten.special_logit', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904dd00>, kernel=<OpOverload(op='aten.special_logit', overload='default')>), <OpOverload(op='aten.multiply', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032f20>, kernel=<OpOverload(op='aten.multiply', overload='Scalar')>), <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 0x7d70b9031da0>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value_size')>), <OpOverload(op='aten.linalg_svdvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e200>, kernel=<OpOverload(op='aten.linalg_svdvals', overload='default')>), <OpOverload(op='aten.unflatten_dense_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e5c0>, kernel=<OpOverload(op='aten.unflatten_dense_tensors', overload='default')>), <OpOverload(op='aten.sum_to_size', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ff60>, kernel=<OpOverload(op='aten.sum_to_size', overload='default')>), <OpOverload(op='c10d_functional.broadcast', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056160>, kernel=<OpOverload(op='c10d_functional.broadcast', overload='default')>), <OpOverload(op='aten._test_check_tensor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045580>, kernel=<OpOverload(op='aten._test_check_tensor', 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 0x7d70b921b740>, kernel=<OpOverload(op='aten._weight_norm_differentiable_backward', overload='default')>), <OpOverload(op='aten.linalg_norm', overload='ord_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033100>, 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 0x7d70b921b9c0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight', overload='default')>), <OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e5c0>, kernel=<OpOverload(op='aten._pack_padded_sequence_backward', overload='default')>), <OpOverload(op='aten.linalg_cond', overload='p_str')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b98ecc20>, kernel=<OpOverload(op='aten.linalg_cond', overload='p_str')>), <OpOverload(op='aten._sparse_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ede0>, kernel=<OpOverload(op='aten._sparse_softmax', overload='Dimname')>), <OpOverload(op='aten.tensordot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ca40>, kernel=<OpOverload(op='aten.tensordot', overload='default')>), <OpOverload(op='aten.stride', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046a20>, kernel=<OpOverload(op='aten.stride', overload='Dimname')>), <OpOverload(op='aten.linalg_vecdot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033ec0>, kernel=<OpOverload(op='aten.linalg_vecdot', overload='default')>), <OpOverload(op='aten.special_log1p', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f6a0>, kernel=<OpOverload(op='aten.special_log1p', 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 0x7d70b9032520>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='atol_rtol_tensor')>), <OpOverload(op='aten.inverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d760>, kernel=<OpOverload(op='aten.inverse', overload='default')>), <OpOverload(op='aten.conv_transpose2d', overload='input')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90307c0>, kernel=<OpOverload(op='aten.conv_transpose2d', overload='input')>), <OpOverload(op='aten.isreal', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90460c0>, kernel=<OpOverload(op='aten.isreal', overload='default')>), <OpOverload(op='aten.__xor__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90304a0>, kernel=<OpOverload(op='aten.__xor__', overload='Tensor')>), <OpOverload(op='aten.rrelu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030680>, 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 0x7d70b9046700>, 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 0x7d70b921ac00>, 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 0x7d70b90462a0>, 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 0x7d70b90572e0>, 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 0x7d70b9056480>, kernel=<OpOverload(op='c10d_functional.reduce_scatter_tensor', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031800>, kernel=<OpOverload(op='aten._sparse_mm', overload='default')>), <OpOverload(op='aten.less', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ff60>, kernel=<OpOverload(op='aten.less', overload='Scalar')>), <OpOverload(op='aten.linalg_pinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90302c0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='default')>), <OpOverload(op='aten._remove_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90319e0>, 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 0x7d70b90334c0>, kernel=<OpOverload(op='aten.special_xlogy', overload='other_scalar')>), <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 0x7d70b903c9a0>, kernel=<OpOverload(op='aten.fbgemm_linear_fp16_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.inner', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032160>, kernel=<OpOverload(op='aten.inner', overload='default')>), <OpOverload(op='aten.qr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903d3a0>, kernel=<OpOverload(op='aten.qr', overload='default')>), <OpOverload(op='aten.absolute', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f060>, 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 0x7d70b9031300>, kernel=<OpOverload(op='aten.nll_loss2d', overload='default')>), <OpOverload(op='aten.sort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903fec0>, kernel=<OpOverload(op='aten.sort', overload='dimname')>), <OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056ac0>, kernel=<OpOverload(op='prepacked.unpack_prepacked_sizes_linear', overload='default')>), <OpOverload(op='aten.is_vulkan_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903fce0>, kernel=<OpOverload(op='aten.is_vulkan_available', overload='default')>), <OpOverload(op='aten._convolution_double_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f4c0>, kernel=<OpOverload(op='aten._convolution_double_backward', overload='default')>), <OpOverload(op='aten.linalg_eigvals', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031580>, kernel=<OpOverload(op='aten.linalg_eigvals', overload='default')>), <OpOverload(op='aten.get_gradients', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032d40>, kernel=<OpOverload(op='aten.get_gradients', overload='default')>), <OpOverload(op='aten.matrix_power', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fec0>, kernel=<OpOverload(op='aten.matrix_power', overload='default')>), <OpOverload(op='aten.special_psi', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032b60>, 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 0x7d70b90560c0>, kernel=<OpOverload(op='c10d_functional.all_reduce', overload='default')>), <OpOverload(op='aten.cummaxmin_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f7e0>, kernel=<OpOverload(op='aten.cummaxmin_backward', 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 0x7d70b9031440>, kernel=<OpOverload(op='aten.sparse_bsr_tensor', overload='crow_col_value')>), <OpOverload(op='aten._propagate_xla_data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f420>, kernel=<OpOverload(op='aten._propagate_xla_data', overload='default')>), <OpOverload(op='aten.size', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d440>, kernel=<OpOverload(op='aten.size', overload='int')>), <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 0x7d70b903d760>, kernel=<OpOverload(op='aten.fbgemm_linear_int8_weight_fp32_activation', overload='default')>), <OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921b880>, kernel=<OpOverload(op='aten.linalg_pinv', overload='rcond_tensor')>), <OpOverload(op='aten._wrapped_linear_prepack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903cae0>, kernel=<OpOverload(op='aten._wrapped_linear_prepack', overload='default')>), <OpOverload(op='aten.linalg_slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032200>, kernel=<OpOverload(op='aten.linalg_slogdet', overload='default')>), <OpOverload(op='aten.conv2d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90328e0>, 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 0x7d70b90463e0>, 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 0x7d70b9032020>, kernel=<OpOverload(op='aten._version', overload='default')>), <OpOverload(op='aten.min', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f380>, kernel=<OpOverload(op='aten.min', overload='other')>), <OpOverload(op='aten.quantized_gru_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903cf40>, kernel=<OpOverload(op='aten.quantized_gru_cell', overload='default')>), <OpOverload(op='aten.sym_is_contiguous', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045bc0>, kernel=<OpOverload(op='aten.sym_is_contiguous', overload='default')>), <OpOverload(op='aten._convolution', overload='deprecated')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032a20>, kernel=<OpOverload(op='aten._convolution', overload='deprecated')>), <OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054d60>, kernel=<OpOverload(op='aten._has_compatible_shallow_copy_type', overload='default')>), <OpOverload(op='aten.triplet_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046fc0>, kernel=<OpOverload(op='aten.triplet_margin_loss', overload='default')>), <OpOverload(op='aten.is_signed', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031d00>, kernel=<OpOverload(op='aten.is_signed', overload='default')>), <OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921a840>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten.hstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903dc60>, kernel=<OpOverload(op='aten.hstack', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e7a0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='x')>), <OpOverload(op='aten.softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c680>, kernel=<OpOverload(op='aten.softmax', overload='Dimname')>), <OpOverload(op='aten.gather', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046660>, kernel=<OpOverload(op='aten.gather', overload='dimname')>), <OpOverload(op='aten.linalg_tensorinv', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045e40>, kernel=<OpOverload(op='aten.linalg_tensorinv', overload='default')>), <OpOverload(op='aten.linalg_solve', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e2a0>, kernel=<OpOverload(op='aten.linalg_solve', overload='default')>), <OpOverload(op='aten.gradient', overload='scalarrayarray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e3e0>, kernel=<OpOverload(op='aten.gradient', overload='scalarrayarray')>), <OpOverload(op='aten._lu_with_info', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e520>, kernel=<OpOverload(op='aten._lu_with_info', overload='default')>), <OpOverload(op='aten.vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045440>, kernel=<OpOverload(op='aten.vander', overload='default')>), <OpOverload(op='aten.concat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90445e0>, kernel=<OpOverload(op='aten.concat', overload='names')>), <OpOverload(op='aten.align_tensors', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e480>, kernel=<OpOverload(op='aten.align_tensors', overload='default')>), <OpOverload(op='aten.thnn_conv2d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903eb60>, kernel=<OpOverload(op='aten.thnn_conv2d', overload='default')>), <OpOverload(op='image.decode_webp', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056ca0>, kernel=<OpOverload(op='image.decode_webp', 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 0x7d70b903c7c0>, 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 0x7d70b903d1c0>, kernel=<OpOverload(op='aten._use_cudnn_rnn_flatten_weight', overload='default')>), <OpOverload(op='aten.__and__', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90467a0>, kernel=<OpOverload(op='aten.__and__', overload='Tensor')>), <OpOverload(op='image.decode_image', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056fc0>, kernel=<OpOverload(op='image.decode_image', overload='default')>), <OpOverload(op='aten.linalg_solve_ex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030180>, 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 0x7d70b9044400>, 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 0x7d70b90449a0>, kernel=<OpOverload(op='aten.retains_grad', overload='default')>), <OpOverload(op='aten.orgqr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fa60>, kernel=<OpOverload(op='aten.orgqr', overload='default')>), <OpOverload(op='aten.rms_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044ea0>, kernel=<OpOverload(op='aten.rms_norm', 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 0x7d70b903fba0>, 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 0x7d70b9031a80>, kernel=<OpOverload(op='aten._test_serialization_subcmul', 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 0x7d70b921ba60>, kernel=<OpOverload(op='aten._debug_has_internal_overlap', overload='default')>), <OpOverload(op='aten.clip', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031620>, kernel=<OpOverload(op='aten.clip', overload='Tensor')>), <OpOverload(op='aten._cast_Long', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045d00>, kernel=<OpOverload(op='aten._cast_Long', overload='default')>), <OpOverload(op='aten.conv_tbc_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045940>, kernel=<OpOverload(op='aten.conv_tbc_backward', overload='default')>), <OpOverload(op='aten.cumulative_trapezoid', overload='dx')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044cc0>, kernel=<OpOverload(op='aten.cumulative_trapezoid', overload='dx')>), <OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032660>, kernel=<OpOverload(op='aten._sparse_coo_tensor_unsafe', overload='default')>), <OpOverload(op='aten.sparse_coo_tensor', overload='indices')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e2a0>, kernel=<OpOverload(op='aten.sparse_coo_tensor', overload='indices')>), <OpOverload(op='aten._validate_sparse_compressed_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030860>, 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 0x7d70b9045800>, kernel=<OpOverload(op='aten._cast_Int', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bc40>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='KN')>), <OpOverload(op='aten.max', overload='other')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ccc0>, kernel=<OpOverload(op='aten.max', overload='other')>), <OpOverload(op='aten.cummax', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031c60>, kernel=<OpOverload(op='aten.cummax', overload='dimname')>), <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 0x7d70b903e020>, 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 0x7d70b903d260>, kernel=<OpOverload(op='aten.adaptive_avg_pool2d', overload='default')>), <OpOverload(op='aten.linalg_matmul', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921b560>, kernel=<OpOverload(op='aten.linalg_matmul', overload='default')>), <OpOverload(op='aten.to_sparse_bsc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904eca0>, kernel=<OpOverload(op='aten.to_sparse_bsc', overload='default')>), <OpOverload(op='image.decode_jpeg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070040>, kernel=<OpOverload(op='image.decode_jpeg', overload='default')>), <OpOverload(op='aten.special_polygamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90322a0>, kernel=<OpOverload(op='aten.special_polygamma', overload='default')>), <OpOverload(op='aten._sobol_engine_draw', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f560>, kernel=<OpOverload(op='aten._sobol_engine_draw', overload='default')>), <OpOverload(op='aten.gradient', overload='array')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903eac0>, kernel=<OpOverload(op='aten.gradient', overload='array')>), <OpOverload(op='aten.gather_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031bc0>, 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 0x7d70b904e8e0>, kernel=<OpOverload(op='aten.native_channel_shuffle', overload='default')>), <OpOverload(op='aten.ger', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90551c0>, kernel=<OpOverload(op='aten.ger', overload='default')>), <OpOverload(op='aten.histogramdd', overload='int_bins')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ea20>, kernel=<OpOverload(op='aten.histogramdd', overload='int_bins')>), <OpOverload(op='aten.data', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e0c0>, 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 0x7d70b9054900>, kernel=<OpOverload(op='aten._is_zerotensor', 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 0x7d70b904c9a0>, kernel=<OpOverload(op='aten._test_autograd_multiple_dispatch', overload='ntonly')>), <OpOverload(op='aten.cudnn_is_acceptable', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055a80>, 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 0x7d70b904f060>, 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 0x7d70b904f420>, 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 0x7d70b904e160>, 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 0x7d70b9047060>, 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 0x7d70b9057ba0>, kernel=<OpOverload(op='image.decode_gif', 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 0x7d70b903d580>, kernel=<OpOverload(op='aten.quantized_rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.arccosh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904eb60>, kernel=<OpOverload(op='aten.arccosh', 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 0x7d70b904f7e0>, kernel=<OpOverload(op='aten._cufft_get_plan_cache_max_size', overload='default')>), <OpOverload(op='aten._cast_Byte', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904eac0>, kernel=<OpOverload(op='aten._cast_Byte', overload='default')>), <OpOverload(op='aten.to_sparse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fc40>, kernel=<OpOverload(op='aten.to_sparse', overload='default')>), <OpOverload(op='aten._pad_circular', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045b20>, kernel=<OpOverload(op='aten._pad_circular', overload='default')>), <OpOverload(op='aten._test_ambiguous_defaults', overload='a')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f880>, kernel=<OpOverload(op='aten._test_ambiguous_defaults', overload='a')>), <OpOverload(op='aten.special_gammainc', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047b00>, 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 0x7d70b9047ba0>, kernel=<OpOverload(op='aten.repeat_interleave', overload='self_Tensor')>), <OpOverload(op='image.read_file', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90571a0>, kernel=<OpOverload(op='image.read_file', overload='default')>), <OpOverload(op='aten.kthvalue', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ede0>, kernel=<OpOverload(op='aten.kthvalue', overload='dimname')>), <OpOverload(op='aten.median', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90471a0>, kernel=<OpOverload(op='aten.median', overload='names_dim')>), <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 0x7d70b904ec00>, kernel=<OpOverload(op='aten.sparse_bsc_tensor', overload='ccol_row_value_size')>), <OpOverload(op='aten._backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ed40>, 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 0x7d70b904e480>, kernel=<OpOverload(op='aten._reshape_from_tensor', overload='default')>), <OpOverload(op='aten.linalg_matrix_rank', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054ae0>, kernel=<OpOverload(op='aten.linalg_matrix_rank', overload='default')>), <OpOverload(op='aten.is_leaf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f600>, 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 0x7d70b9070ae0>, kernel=<OpOverload(op='torchvision._cuda_version', overload='default')>), <OpOverload(op='quantized.conv2d_unpack_sizes', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056340>, 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 0x7d70b903d8a0>, 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 0x7d70b9030900>, 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 0x7d70b903f6a0>, kernel=<OpOverload(op='aten.ctc_loss', overload='Tensor')>), <OpOverload(op='aten.trapezoid', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044040>, kernel=<OpOverload(op='aten.trapezoid', overload='x')>), <OpOverload(op='aten.ldexp', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e160>, kernel=<OpOverload(op='aten.ldexp', overload='Tensor')>), <OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070220>, kernel=<OpOverload(op='mkldnn._is_mkldnn_acl_supported', overload='default')>), <OpOverload(op='aten.log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903cb80>, 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 0x7d70b9057100>, kernel=<OpOverload(op='image._jpeg_version', overload='default')>), <OpOverload(op='aten.cdist', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031f80>, kernel=<OpOverload(op='aten.cdist', overload='default')>), <OpOverload(op='aten.one_hot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90451c0>, kernel=<OpOverload(op='aten.one_hot', overload='default')>), <OpOverload(op='aten.arctan', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f240>, kernel=<OpOverload(op='aten.arctan', overload='default')>), <OpOverload(op='aten._sparse_mm', overload='reduce')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033d80>, kernel=<OpOverload(op='aten._sparse_mm', overload='reduce')>), <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 0x7d70b90565c0>, kernel=<OpOverload(op='c10d_functional.all_gather_into_tensor_coalesced', overload='default')>), <OpOverload(op='aten.slogdet', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045c60>, kernel=<OpOverload(op='aten.slogdet', overload='default')>), <OpOverload(op='aten.arctanh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033380>, kernel=<OpOverload(op='aten.arctanh', overload='default')>), <OpOverload(op='aten.nanmedian', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903dbc0>, kernel=<OpOverload(op='aten.nanmedian', overload='names_dim')>), <OpOverload(op='aten.trapz', overload='x')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e840>, 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 0x7d70b903df80>, kernel=<OpOverload(op='aten.nanmean', overload='default')>), <OpOverload(op='aten.arctan2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f9c0>, kernel=<OpOverload(op='aten.arctan2', overload='default')>), <OpOverload(op='_test.leaky_relu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056f20>, kernel=<OpOverload(op='_test.leaky_relu', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045a80>, kernel=<OpOverload(op='aten._sparse_sum', overload='dim_dtype')>), <OpOverload(op='aten.linalg_cond', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c540>, kernel=<OpOverload(op='aten.linalg_cond', overload='default')>), <OpOverload(op='image.encode_jpegs_cuda', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055e40>, kernel=<OpOverload(op='image.encode_jpegs_cuda', overload='default')>), <OpOverload(op='aten.special_i0', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044fe0>, kernel=<OpOverload(op='aten.special_i0', 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 0x7d70b903e340>, kernel=<OpOverload(op='aten._embedding_bag_sparse_backward', overload='default')>), <OpOverload(op='aten.cumprod_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90442c0>, kernel=<OpOverload(op='aten.cumprod_backward', overload='default')>), <OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031ee0>, kernel=<OpOverload(op='aten._batch_norm_impl_index_backward', overload='default')>), <OpOverload(op='aten.linalg_multi_dot', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90313a0>, kernel=<OpOverload(op='aten.linalg_multi_dot', overload='default')>), <OpOverload(op='aten.nuclear_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030ea0>, kernel=<OpOverload(op='aten.nuclear_norm', overload='default')>), <OpOverload(op='aten.vstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90325c0>, kernel=<OpOverload(op='aten.vstack', overload='default')>), <OpOverload(op='aten.special_round', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921b240>, 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 0x7d70b9033ce0>, kernel=<OpOverload(op='aten.msort', overload='default')>), <OpOverload(op='aten.linalg_lu_factor', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033ba0>, kernel=<OpOverload(op='aten.linalg_lu_factor', overload='default')>), <OpOverload(op='aten.argsort', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90337e0>, kernel=<OpOverload(op='aten.argsort', overload='dimname')>), <OpOverload(op='aten._add_batch_dim', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030720>, kernel=<OpOverload(op='aten._add_batch_dim', 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 0x7d70b9031080>, kernel=<OpOverload(op='aten._grid_sampler_2d_cpu_fallback_backward', overload='default')>), <OpOverload(op='mkldnn._is_mkldnn_bf16_supported', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057420>, 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 0x7d70b9033a60>, kernel=<OpOverload(op='aten._cast_Double', overload='default')>), <OpOverload(op='aten.is_floating_point', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90542c0>, kernel=<OpOverload(op='aten.is_floating_point', overload='default')>), <OpOverload(op='aten.greater_equal', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c400>, kernel=<OpOverload(op='aten.greater_equal', overload='Scalar')>), <OpOverload(op='aten._cast_Short', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921aac0>, kernel=<OpOverload(op='aten._cast_Short', overload='default')>), <OpOverload(op='aten.index_select_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90468e0>, kernel=<OpOverload(op='aten.index_select_backward', 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 0x7d70b903c220>, kernel=<OpOverload(op='aten._sparse_compressed_tensor_unsafe', overload='default')>), <OpOverload(op='_test.cat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056b60>, kernel=<OpOverload(op='_test.cat', overload='default')>), <OpOverload(op='aten.linalg_eigh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903cea0>, kernel=<OpOverload(op='aten.linalg_eigh', 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 0x7d70b9032ca0>, kernel=<OpOverload(op='aten._scaled_dot_product_attention_math', overload='default')>), <OpOverload(op='aten.min', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031b20>, kernel=<OpOverload(op='aten.min', overload='names_dim')>), <OpOverload(op='aten.kron', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921aa20>, kernel=<OpOverload(op='aten.kron', overload='default')>), <OpOverload(op='aten.special_xlogy', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c040>, kernel=<OpOverload(op='aten.special_xlogy', overload='default')>), <OpOverload(op='aten.to_dense_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bf60>, kernel=<OpOverload(op='aten.to_dense_backward', 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 0x7d70b90314e0>, kernel=<OpOverload(op='aten.sparse_csc_tensor', overload='ccol_row_value')>), <OpOverload(op='aten.mode', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90300e0>, kernel=<OpOverload(op='aten.mode', overload='dimname')>), <OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bec0>, kernel=<OpOverload(op='aten.linalg_pinv', overload='atol_rtol_float')>), <OpOverload(op='aten.dstack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90316c0>, kernel=<OpOverload(op='aten.dstack', overload='default')>), <OpOverload(op='aten.flipud', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ccc0>, kernel=<OpOverload(op='aten.flipud', overload='default')>), <OpOverload(op='aten.diagflat', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030540>, 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 0x7d70b9030400>, 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 0x7d70b9033420>, kernel=<OpOverload(op='aten.to_dense', overload='default')>), <OpOverload(op='aten.quantile', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f2e0>, kernel=<OpOverload(op='aten.quantile', overload='default')>), <OpOverload(op='aten.cov', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031940>, kernel=<OpOverload(op='aten.cov', 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 0x7d70b90576a0>, kernel=<OpOverload(op='profiler._record_function_enter_new', overload='default')>), <OpOverload(op='aten.diff', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055260>, kernel=<OpOverload(op='aten.diff', overload='default')>), <OpOverload(op='aten.argwhere', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bd80>, kernel=<OpOverload(op='aten.argwhere', 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 0x7d70b90545e0>, kernel=<OpOverload(op='aten.nested_to_padded_tensor', overload='default')>), <OpOverload(op='aten.is_inference', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bba0>, kernel=<OpOverload(op='aten.is_inference', overload='default')>), <OpOverload(op='aten.is_conj', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046160>, 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 0x7d70b9032840>, kernel=<OpOverload(op='aten.row_stack', overload='default')>), <OpOverload(op='aten.is_neg', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055800>, 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 0x7d70b9055440>, kernel=<OpOverload(op='aten.conv1d', overload='padding')>), <OpOverload(op='aten.gradient', overload='scalararray')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055620>, kernel=<OpOverload(op='aten.gradient', overload='scalararray')>), <OpOverload(op='aten.rnn_relu_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90453a0>, kernel=<OpOverload(op='aten.rnn_relu_cell', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044680>, kernel=<OpOverload(op='aten.adaptive_avg_pool1d', overload='default')>), <OpOverload(op='aten.pinverse', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903dda0>, kernel=<OpOverload(op='aten.pinverse', overload='default')>), <OpOverload(op='aten._to_cpu', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033060>, kernel=<OpOverload(op='aten._to_cpu', overload='default')>), <OpOverload(op='aten.fliplr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f740>, kernel=<OpOverload(op='aten.fliplr', overload='default')>), <OpOverload(op='aten.quantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903da80>, kernel=<OpOverload(op='aten.quantile', overload='scalar')>), <OpOverload(op='aten.argsort', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90549a0>, kernel=<OpOverload(op='aten.argsort', overload='default')>), <OpOverload(op='aten.multilabel_margin_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e840>, kernel=<OpOverload(op='aten.multilabel_margin_loss', overload='default')>), <OpOverload(op='aten.chalf', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030fe0>, kernel=<OpOverload(op='aten.chalf', overload='default')>), <OpOverload(op='aten.special_gammaln', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90323e0>, kernel=<OpOverload(op='aten.special_gammaln', overload='default')>), <OpOverload(op='aten.to_sparse_csr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030ae0>, kernel=<OpOverload(op='aten.to_sparse_csr', overload='default')>), <OpOverload(op='aten.linalg_vander', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046340>, 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 0x7d70b9046840>, kernel=<OpOverload(op='aten.ctc_loss', overload='IntList')>), <OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d8a0>, kernel=<OpOverload(op='aten._sparse_csr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.greater', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921ad40>, kernel=<OpOverload(op='aten.greater', overload='Scalar')>), <OpOverload(op='aten._pad_packed_sequence', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031260>, kernel=<OpOverload(op='aten._pad_packed_sequence', overload='default')>), <OpOverload(op='aten.negative', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c900>, kernel=<OpOverload(op='aten.negative', overload='default')>), <OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c680>, kernel=<OpOverload(op='aten._sparse_log_softmax', overload='Dimname')>), <OpOverload(op='aten.gradient', overload='tensorarrayint')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903fe20>, kernel=<OpOverload(op='aten.gradient', overload='tensorarrayint')>), <OpOverload(op='aten.special_digamma', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ca40>, kernel=<OpOverload(op='aten.special_digamma', overload='default')>), <OpOverload(op='c10d_functional.all_to_all_single', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056660>, kernel=<OpOverload(op='c10d_functional.all_to_all_single', overload='default')>), <OpOverload(op='aten.trace_backward', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904dbc0>, kernel=<OpOverload(op='aten.trace_backward', overload='default')>), <OpOverload(op='aten._validate_sparse_csc_tensor_args', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903cc20>, 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 0x7d70b9056e80>, kernel=<OpOverload(op='profiler._record_function_exit', overload='default')>), <OpOverload(op='aten.nanquantile', overload='scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032700>, kernel=<OpOverload(op='aten.nanquantile', overload='scalar')>), <OpOverload(op='aten.scatter_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030a40>, kernel=<OpOverload(op='aten.scatter_add', overload='dimname')>), <OpOverload(op='aten.linalg_eigvalsh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9030e00>, kernel=<OpOverload(op='aten.linalg_eigvalsh', 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 0x7d70b921a980>, kernel=<OpOverload(op='aten._sparse_bsc_tensor_unsafe', overload='default')>), <OpOverload(op='aten.linalg_matrix_norm', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033f60>, kernel=<OpOverload(op='aten.linalg_matrix_norm', overload='default')>), <OpOverload(op='aten._wrapped_quantized_linear_prepacked', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903eca0>, kernel=<OpOverload(op='aten._wrapped_quantized_linear_prepacked', 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 0x7d70b9030c20>, 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 0x7d70b9033920>, kernel=<OpOverload(op='aten.rnn_tanh_cell', overload='default')>), <OpOverload(op='aten.result_type', overload='Scalar_Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90554e0>, kernel=<OpOverload(op='aten.result_type', overload='Scalar_Tensor')>), <OpOverload(op='aten.result_type', overload='Tensor')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054b80>, kernel=<OpOverload(op='aten.result_type', overload='Tensor')>), <OpOverload(op='aten.l1_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d4e0>, kernel=<OpOverload(op='aten.l1_loss', overload='default')>), <OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c2c0>, kernel=<OpOverload(op='aten.fbgemm_pack_quantized_matrix', overload='default')>), <OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90309a0>, kernel=<OpOverload(op='aten.adaptive_avg_pool3d', overload='default')>), <OpOverload(op='aten.__xor__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921a8e0>, kernel=<OpOverload(op='aten.__xor__', overload='Scalar')>), <OpOverload(op='aten.conv3d', overload='padding')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d080>, kernel=<OpOverload(op='aten.conv3d', overload='padding')>), <OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054360>, kernel=<OpOverload(op='aten.fbgemm_pack_gemm_matrix_fp16', overload='default')>), <OpOverload(op='aten.combinations', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d9e0>, kernel=<OpOverload(op='aten.combinations', 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 0x7d70b90540e0>, kernel=<OpOverload(op='aten.infinitely_differentiable_gelu_backward', overload='default')>), <OpOverload(op='aten._nnpack_available', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047c40>, kernel=<OpOverload(op='aten._nnpack_available', 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 0x7d70b9057c40>, kernel=<OpOverload(op='mkldnn._is_mkldnn_fp16_supported', overload='default')>), <OpOverload(op='aten._rowwise_prune', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047600>, kernel=<OpOverload(op='aten._rowwise_prune', overload='default')>), <OpOverload(op='aten.fix', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b921bb00>, kernel=<OpOverload(op='aten.fix', 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 0x7d70b9056200>, 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 0x7d70b9033240>, kernel=<OpOverload(op='aten.masked_select_backward', overload='default')>), <OpOverload(op='aten.lstm_cell', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031e40>, kernel=<OpOverload(op='aten.lstm_cell', overload='default')>), <OpOverload(op='aten._sparse_sum', overload='dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904dda0>, kernel=<OpOverload(op='aten._sparse_sum', overload='dtype')>), <OpOverload(op='aten.choose_qparams_optimized', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044f40>, 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 0x7d70b903fa60>, kernel=<OpOverload(op='aten.conv_transpose3d', overload='input')>), <OpOverload(op='image.encode_png', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057740>, kernel=<OpOverload(op='image.encode_png', 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 0x7d70b921b920>, kernel=<OpOverload(op='aten._saturate_weight_to_fp16', overload='default')>), <OpOverload(op='aten.to_sparse', overload='sparse_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903e980>, kernel=<OpOverload(op='aten.to_sparse', overload='sparse_dim')>), <OpOverload(op='aten.linalg_cholesky', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055120>, kernel=<OpOverload(op='aten.linalg_cholesky', 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 0x7d70b903d120>, kernel=<OpOverload(op='aten.fbgemm_linear_quantize_weight', overload='default')>), <OpOverload(op='aten.__or__', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903d800>, kernel=<OpOverload(op='aten.__or__', overload='Scalar')>), <OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9032340>, kernel=<OpOverload(op='aten._sparse_bsr_tensor_unsafe', overload='default')>), <OpOverload(op='aten.arccos', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f100>, kernel=<OpOverload(op='aten.arccos', overload='default')>), <OpOverload(op='aten.arcsinh', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045120>, kernel=<OpOverload(op='aten.arcsinh', overload='default')>), <OpOverload(op='aten.sym_stride', overload='int')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9031760>, kernel=<OpOverload(op='aten.sym_stride', overload='int')>), <OpOverload(op='aten.scatter', overload='dimname_value')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ec00>, kernel=<OpOverload(op='aten.scatter', overload='dimname_value')>), <OpOverload(op='aten.cummin', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046d40>, kernel=<OpOverload(op='aten.cummin', overload='dimname')>), <OpOverload(op='aten.outer', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044900>, kernel=<OpOverload(op='aten.outer', overload='default')>), <OpOverload(op='aten.diag', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90447c0>, kernel=<OpOverload(op='aten.diag', overload='default')>), <OpOverload(op='aten.float_power', overload='Tensor_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90444a0>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Scalar')>), <OpOverload(op='aten.float_power', overload='Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044180>, 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 0x7d70b903fd80>, kernel=<OpOverload(op='aten.float_power', overload='Tensor_Tensor')>), <OpOverload(op='aten.square', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ed40>, kernel=<OpOverload(op='aten.square', overload='default')>), <OpOverload(op='aten.clip', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044220>, kernel=<OpOverload(op='aten.clip', overload='default')>), <OpOverload(op='aten.celu', overload='out')>: <function celu at 0x7d717abf4fe0>, <OpOverload(op='aten.elu', overload='out')>: <function elu at 0x7d717abf56c0>, <OpOverload(op='aten.relu', overload='out')>: <function relu at 0x7d717abf5b20>, <OpOverload(op='aten.channel_shuffle', overload='default')>: <function channel_shuffle at 0x7d717abf6160>, <OpOverload(op='aten.channel_shuffle', overload='out')>: <function channel_shuffle at 0x7d717abf6160>, <OpOverload(op='aten.leaky_relu', overload='out')>: <function leaky_relu at 0x7d717abf5ee0>, <OpOverload(op='aten.mish', overload='out')>: <function mish at 0x7d717abf6480>, <OpOverload(op='aten.hardshrink', overload='default')>: <function hardshrink at 0x7d717abf7560>, <OpOverload(op='aten.softshrink', overload='default')>: <function softshrink at 0x7d717abf7880>, <OpOverload(op='aten.softplus', overload='out')>: <function softplus at 0x7d717abf6e80>, <OpOverload(op='aten.hardshrink', overload='out')>: <function hardshrink at 0x7d717abf7560>, <OpOverload(op='aten.softshrink', overload='out')>: <function softshrink at 0x7d717abf7880>, <OpOverload(op='aten.margin_ranking_loss', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045da0>, 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 0x7d70b9045620>, kernel=<OpOverload(op='aten.hinge_embedding_loss', overload='default')>), <OpOverload(op='aten.nll_loss', overload='out')>: <function nll_loss at 0x7d717aa28900>, <OpOverload(op='aten.huber_loss', overload='default')>: <function huber_loss at 0x7d717aa28cc0>, <OpOverload(op='aten.huber_loss', overload='out')>: <function huber_loss at 0x7d717aa28cc0>, <OpOverload(op='aten.threshold', overload='default')>: <function threshold at 0x7d717abf7240>, <OpOverload(op='aten.threshold', overload='out')>: <function threshold at 0x7d717abf7240>, <OpOverload(op='aten.hardtanh', overload='out')>: <function hardtanh at 0x7d717aa28f40>, <OpOverload(op='aten.gelu', overload='out')>: <function gelu at 0x7d717aa29800>, <OpOverload(op='aten.glu', overload='out')>: <function glu at 0x7d717aa2a340>, <OpOverload(op='aten.pairwise_distance', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903d940>, 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 0x7d70b903cfe0>, kernel=<OpOverload(op='aten.pdist', overload='default')>), <OpOverload(op='aten.pixel_shuffle', overload='out')>: <function pixel_shuffle at 0x7d717aa2ac00>, <OpOverload(op='aten.pixel_unshuffle', overload='out')>: <function pixel_unshuffle at 0x7d717aa2ae80>, <OpOverload(op='aten.celu_', overload='default')>: <function celu at 0x7d717aa29940>, <OpOverload(op='aten.threshold_', overload='default')>: <function threshold at 0x7d717aa2b1a0>, <OpOverload(op='aten.elu_', overload='default')>: <function elu at 0x7d717aa2aca0>, <OpOverload(op='aten.mish_', overload='default')>: <function mish at 0x7d717aa2af20>, <OpOverload(op='aten.special_bessel_j0', overload='default')>: <function bessel_j0 at 0x7d717aa28860>, <OpOverload(op='aten.special_bessel_j0', overload='out')>: <function bessel_j0 at 0x7d717aa28860>, <OpOverload(op='aten.logit', overload='out')>: <function logit at 0x7d717aa51760>, <OpOverload(op='aten.special_bessel_j1', overload='default')>: <function bessel_j1 at 0x7d717aa2b880>, <OpOverload(op='aten.special_bessel_j1', overload='out')>: <function bessel_j1 at 0x7d717aa2b880>, <OpOverload(op='aten.special_entr', overload='default')>: <function entr at 0x7d717aa2bc40>, <OpOverload(op='aten.special_entr', overload='out')>: <function entr at 0x7d717aa2bc40>, <OpOverload(op='aten.special_erfcx', overload='out')>: <function erfcx at 0x7d717aa2bf60>, <OpOverload(op='aten.special_i0e', overload='default')>: <function i0e at 0x7d717aa505e0>, <OpOverload(op='aten.special_i0e', overload='out')>: <function i0e at 0x7d717aa505e0>, <OpOverload(op='aten.special_i1', overload='default')>: <function i1 at 0x7d717aa50ae0>, <OpOverload(op='aten.special_i1', overload='out')>: <function i1 at 0x7d717aa50ae0>, <OpOverload(op='aten.special_i1e', overload='default')>: <function i1e at 0x7d717aa50fe0>, <OpOverload(op='aten.special_i1e', overload='out')>: <function i1e at 0x7d717aa50fe0>, <OpOverload(op='aten.special_log_ndtr', overload='default')>: <function log_ndtr at 0x7d717aa513a0>, <OpOverload(op='aten.special_log_ndtr', overload='out')>: <function log_ndtr at 0x7d717aa513a0>, <OpOverload(op='aten.special_xlog1py', overload='default')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='self_scalar_out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.special_xlog1py', overload='other_scalar_out')>: <function xlog1py at 0x7d717aa516c0>, <OpOverload(op='aten.mvlgamma', overload='default')>: <function multigammaln at 0x7d717aa518a0>, <OpOverload(op='aten.mvlgamma', overload='out')>: <function multigammaln at 0x7d717aa518a0>, <OpOverload(op='aten.special_ndtr', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047ec0>, kernel=<OpOverload(op='aten.special_ndtr', overload='default')>), <OpOverload(op='aten.special_ndtr', overload='out')>: <function ndtr at 0x7d717aa51c60>, <OpOverload(op='aten.special_ndtri', overload='default')>: <function ndtri at 0x7d717aa52020>, <OpOverload(op='aten.special_ndtri', overload='out')>: <function ndtri at 0x7d717aa52020>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='default')>: <function spherical_bessel_j0 at 0x7d717aa52660>, <OpOverload(op='aten.special_spherical_bessel_j0', overload='out')>: <function spherical_bessel_j0 at 0x7d717aa52660>, <OpOverload(op='aten.special_zeta', overload='default')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='other_scalar')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='self_scalar')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='self_scalar_out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.special_zeta', overload='other_scalar_out')>: <function zeta at 0x7d717aa52ac0>, <OpOverload(op='aten.cumsum', overload='out')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.cumprod', overload='default')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d3a0>, kernel=<OpOverload(op='aten.cumprod', overload='dimname')>), <OpOverload(op='aten.cumprod', overload='dimname_out')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.cumprod', overload='out')>: <function cumprod at 0x7d717ad0e0c0>, <OpOverload(op='aten.eye', overload='m_out')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.eye', overload='default')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.lerp', overload='Scalar_out')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.lerp', overload='Tensor_out')>: <function lerp at 0x7d717ad0fd80>, <OpOverload(op='aten.eye', overload='out')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Tensor_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Scalar_Tensor_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.linspace', overload='Tensor_Scalar_out')>: <function linspace at 0x7d717ad0ff60>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='default')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Tensor_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Tensor_Scalar_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.logspace', overload='Scalar_Tensor_out')>: <function logspace at 0x7d717ab302c0>, <OpOverload(op='aten.meshgrid', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055940>, 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 0x7d70b9055bc0>, kernel=<OpOverload(op='aten.meshgrid', overload='indexing')>), <OpOverload(op='aten.tril_indices', overload='default')>: <function tril_indices at 0x7d717ab32520>, <OpOverload(op='aten.tril_indices', overload='out')>: <function tril_indices at 0x7d717ab32520>, <OpOverload(op='aten.masked_fill', overload='Scalar_out')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.triu_indices', overload='out')>: <function triu_indices at 0x7d717ab31940>, <OpOverload(op='aten.masked_fill', overload='Tensor_out')>: <function masked_fill at 0x7d717ab31260>, <OpOverload(op='aten.masked_fill_', overload='Scalar')>: <function masked_fill_ at 0x7d717ab31080>, <OpOverload(op='aten.masked_fill_', overload='Tensor')>: <function masked_fill_ at 0x7d717ab31080>, <OpOverload(op='aten.triu_indices', overload='default')>: <function triu_indices at 0x7d717ab31940>, <OpOverload(op='aten.norm', overload='Scalar')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fe20>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim')>), <OpOverload(op='aten.norm', overload='out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='ScalarOpt_dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='Scalar_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fd80>, kernel=<OpOverload(op='aten.norm', overload='names_ScalarOpt_dim_dtype')>), <OpOverload(op='aten.norm', overload='names_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten.trace', overload='default')>: <function trace at 0x7d717ab319e0>, <OpOverload(op='aten.trace', overload='out')>: <function trace at 0x7d717ab319e0>, <OpOverload(op='aten.triu', overload='out')>: <function triu at 0x7d717ab31ee0>, <OpOverload(op='aten.tril', overload='out')>: <function tril at 0x7d717ab32160>, <OpOverload(op='aten.bucketize', overload='Tensor')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Scalar')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Tensor_out')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.bucketize', overload='Scalar_out')>: <function bucketize at 0x7d717ab30e00>, <OpOverload(op='aten.cauchy', overload='default')>: <function cauchy at 0x7d717ab32700>, <OpOverload(op='aten.exponential', overload='default')>: <function exponential at 0x7d717ab32ac0>, <OpOverload(op='aten.exponential', overload='out')>: <function exponential at 0x7d717ab32ac0>, <OpOverload(op='aten.geometric', overload='default')>: <function geometric at 0x7d717ab32e80>, <OpOverload(op='aten.geometric', overload='out')>: <function geometric at 0x7d717ab32e80>, <OpOverload(op='aten.log_normal', overload='default')>: <function log_normal at 0x7d717ab33240>, <OpOverload(op='aten.log_normal', overload='out')>: <function log_normal at 0x7d717ab33240>, <OpOverload(op='aten.rad2deg', overload='out')>: <function rad2deg at 0x7d717ab33c40>, <OpOverload(op='aten.deg2rad', overload='out')>: <function deg2rad at 0x7d717ab60180>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.count_nonzero', overload='dim_IntList_out')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.count_nonzero', overload='default')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.digamma_', overload='default')>: <function digamma at 0x7d717ab61760>, <OpOverload(op='aten.dot', overload='out')>: <function dot at 0x7d717ab31e40>, <OpOverload(op='aten.vdot', overload='default')>: <function vdot at 0x7d717ab60720>, <OpOverload(op='aten.vdot', overload='out')>: <function vdot at 0x7d717ab60720>, <OpOverload(op='aten.deg2rad_', overload='default')>: <function deg2rad at 0x7d717ab619e0>, <OpOverload(op='aten.select_scatter', overload='out')>: <function select_scatter at 0x7d717ab609a0>, <OpOverload(op='aten.abs_', overload='default')>: <function abs at 0x7d717ab33740>, <OpOverload(op='aten.cumprod_', overload='default')>: <function cumprod at 0x7d717ab61c60>, <OpOverload(op='aten.cumprod_', overload='dimname')>: <function cumprod at 0x7d717ab61c60>, <OpOverload(op='aten.acos_', overload='default')>: <function acos at 0x7d717ab607c0>, <OpOverload(op='aten.acosh_', overload='default')>: <function acosh at 0x7d717ab60a40>, <OpOverload(op='aten.lgamma_', overload='default')>: <function lgamma at 0x7d717ab63a60>, <OpOverload(op='aten.add_', overload='Tensor')>: <function add at 0x7d717ab60b80>, <OpOverload(op='aten.add_', overload='Scalar')>: <function add at 0x7d717ab60b80>, <OpOverload(op='aten.addcmul_', overload='default')>: <function addcmul at 0x7d717ab60cc0>, <OpOverload(op='aten.cumsum_', overload='default')>: <function cumsum at 0x7d717ab61ee0>, <OpOverload(op='aten.cumsum_', overload='dimname')>: <function cumsum at 0x7d717ab61ee0>, <OpOverload(op='aten.addcdiv_', overload='default')>: <function addcdiv at 0x7d717ab60e00>, <OpOverload(op='aten.asin_', overload='default')>: <function asin at 0x7d717ab60f40>, <OpOverload(op='aten.tanh_', overload='default')>: <function tanh at 0x7d717ab94f40>, <OpOverload(op='aten.asinh_', overload='default')>: <function asinh at 0x7d717ab61080>, <OpOverload(op='aten.atan_', overload='default')>: <function atan at 0x7d717ab611c0>, <OpOverload(op='aten.cosh_', overload='default')>: <function cosh at 0x7d717ab62160>, <OpOverload(op='aten.atanh_', overload='default')>: <function atanh at 0x7d717ab61300>, <OpOverload(op='aten.log10_', overload='default')>: <function log10 at 0x7d717ab637e0>, <OpOverload(op='aten.cos_', overload='default')>: <function cos at 0x7d717ab623e0>, <OpOverload(op='aten.atan2_', overload='default')>: <function atan2 at 0x7d717ab61440>, <OpOverload(op='aten.bitwise_and_', overload='Tensor')>: <function bitwise_and at 0x7d717ab61580>, <OpOverload(op='aten.bitwise_and_', overload='Scalar')>: <function bitwise_and at 0x7d717ab61580>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor_Scalar')>: <function bitwise_left_shift at 0x7d717ab616c0>, <OpOverload(op='aten.bitwise_left_shift_', overload='Tensor')>: <function bitwise_left_shift at 0x7d717ab616c0>, <OpOverload(op='aten.bitwise_not_', overload='default')>: <function bitwise_not at 0x7d717ab61800>, <OpOverload(op='aten.copysign_', overload='Tensor')>: <function copysign at 0x7d717ab62340>, <OpOverload(op='aten.bitwise_or_', overload='Tensor')>: <function bitwise_or at 0x7d717ab61940>, <OpOverload(op='aten.bitwise_or_', overload='Scalar')>: <function bitwise_or at 0x7d717ab61940>, <OpOverload(op='aten.copysign_', overload='Scalar')>: <function copysign at 0x7d717ab62340>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor_Scalar')>: <function bitwise_right_shift at 0x7d717ab61a80>, <OpOverload(op='aten.bitwise_right_shift_', overload='Tensor')>: <function bitwise_right_shift at 0x7d717ab61a80>, <OpOverload(op='aten.bitwise_xor_', overload='Scalar')>: <function bitwise_xor at 0x7d717ab61bc0>, <OpOverload(op='aten.log1p_', overload='default')>: <function log1p at 0x7d717ab63560>, <OpOverload(op='aten.ceil_', overload='default')>: <function ceil at 0x7d717ab61d00>, <OpOverload(op='aten.clamp_', overload='default')>: <function clamp at 0x7d717ab61e40>, <OpOverload(op='aten.clamp_', overload='Tensor')>: <function clamp at 0x7d717ab61e40>, <OpOverload(op='aten.conj_physical_', overload='default')>: <function conj_physical at 0x7d717ab62200>, <OpOverload(op='aten.clamp_min_', overload='default')>: <function clamp_min at 0x7d717ab61f80>, <OpOverload(op='aten.clamp_min_', overload='Tensor')>: <function clamp_min at 0x7d717ab61f80>, <OpOverload(op='aten.clamp_max_', overload='Tensor')>: <function clamp_max at 0x7d717ab620c0>, <OpOverload(op='aten.div_', overload='Tensor')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Tensor_mode')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Scalar')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.div_', overload='Scalar_mode')>: <function div at 0x7d717ab614e0>, <OpOverload(op='aten.lerp_', overload='Scalar')>: <function lerp at 0x7d717ab639c0>, <OpOverload(op='aten.lerp_', overload='Tensor')>: <function lerp at 0x7d717ab639c0>, <OpOverload(op='aten.eq_', overload='Scalar')>: <function eq at 0x7d717ab61260>, <OpOverload(op='aten.eq_', overload='Tensor')>: <function eq at 0x7d717ab61260>, <OpOverload(op='aten.erf_', overload='default')>: <function erf at 0x7d717ab60fe0>, <OpOverload(op='aten.erfc_', overload='default')>: <function erfc at 0x7d717ab60d60>, <OpOverload(op='aten.erfinv_', overload='default')>: <function erfinv at 0x7d717ab60ae0>, <OpOverload(op='aten.exp_', overload='default')>: <function exp at 0x7d717ab60680>, <OpOverload(op='aten.le_', overload='Scalar')>: <function le at 0x7d717ab63880>, <OpOverload(op='aten.expm1_', overload='default')>: <function expm1 at 0x7d717ab625c0>, <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 0x7d717ab63420>, <OpOverload(op='aten.float_power_', overload='Tensor')>: <function float_power at 0x7d717ab62700>, <OpOverload(op='aten.float_power_', overload='Scalar')>: <function float_power at 0x7d717ab62700>, <OpOverload(op='aten.floor_', overload='default')>: <function floor at 0x7d717ab62840>, <OpOverload(op='aten.floor_divide_', overload='Scalar')>: <function floor_divide at 0x7d717ab62980>, <OpOverload(op='aten.floor_divide_', overload='Tensor')>: <function floor_divide at 0x7d717ab62980>, <OpOverload(op='aten.fmod_', overload='Tensor')>: <function fmod at 0x7d717ab62ac0>, <OpOverload(op='aten.fmod_', overload='Scalar')>: <function fmod at 0x7d717ab62ac0>, <OpOverload(op='aten.frac_', overload='default')>: <function frac at 0x7d717ab62c00>, <OpOverload(op='aten.log_normal_', overload='default')>: <function log_normal at 0x7d717ab631a0>, <OpOverload(op='aten.logical_xor_', overload='default')>: <function logical_xor at 0x7d717ab62660>, <OpOverload(op='aten.gcd_', overload='default')>: <function gcd at 0x7d717ab62d40>, <OpOverload(op='aten.ge_', overload='Tensor')>: <function ge at 0x7d717ab62e80>, <OpOverload(op='aten.logical_or_', overload='default')>: <function logical_or at 0x7d717ab628e0>, <OpOverload(op='aten.gt_', overload='Scalar')>: <function gt at 0x7d717ab62fc0>, <OpOverload(op='aten.gt_', overload='Tensor')>: <function gt at 0x7d717ab62fc0>, <OpOverload(op='aten.heaviside_', overload='default')>: <function heaviside at 0x7d717ab63100>, <OpOverload(op='aten.hypot_', overload='default')>: <function hypot at 0x7d717ab63240>, <OpOverload(op='aten.logical_not_', overload='default')>: <function logical_not at 0x7d717ab62b60>, <OpOverload(op='aten.igamma_', overload='default')>: <function igamma at 0x7d717ab63380>, <OpOverload(op='aten.igammac_', overload='default')>: <function igammac at 0x7d717ab634c0>, <OpOverload(op='aten.logical_and_', overload='default')>: <function logical_and at 0x7d717ab62de0>, <OpOverload(op='aten.log2_', overload='default')>: <function log2 at 0x7d717ab632e0>, <OpOverload(op='aten.i0_', overload='default')>: <function i0 at 0x7d717ab63600>, <OpOverload(op='aten.log_', overload='default')>: <function log at 0x7d717ab63060>, <OpOverload(op='aten.lcm_', overload='default')>: <function lcm at 0x7d717ab63740>, <OpOverload(op='aten.lt_', overload='Scalar')>: <function lt at 0x7d717ab60040>, <OpOverload(op='aten.lt_', overload='Tensor')>: <function lt at 0x7d717ab60040>, <OpOverload(op='aten.mul_', overload='Tensor')>: <function mul at 0x7d717ab60c20>, <OpOverload(op='aten.mul_', overload='Scalar')>: <function mul at 0x7d717ab60c20>, <OpOverload(op='aten.mvlgamma_', overload='default')>: <function _make_alias.<locals>._fn at 0x7d717ab61120>, <OpOverload(op='aten.geometric_', overload='default')>: <function geometric at 0x7d717ab62ca0>, <OpOverload(op='aten.nan_to_num_', overload='default')>: <function nan_to_num at 0x7d717ab61620>, <OpOverload(op='aten.ne_', overload='Scalar')>: <function ne at 0x7d717ab61b20>, <OpOverload(op='aten.ne_', overload='Tensor')>: <function ne at 0x7d717ab61b20>, <OpOverload(op='aten.exponential_', overload='default')>: <function exponential at 0x7d717ab627a0>, <OpOverload(op='aten.neg_', overload='default')>: <function neg at 0x7d717ab62020>, <OpOverload(op='aten.nextafter_', overload='default')>: <function nextafter at 0x7d717ab63b00>, <OpOverload(op='aten.cauchy_', overload='default')>: <function cauchy at 0x7d717ab60900>, <OpOverload(op='aten.pow_', overload='Scalar')>: <function pow at 0x7d717ab63c40>, <OpOverload(op='aten.pow_', overload='Tensor')>: <function pow at 0x7d717ab63c40>, <OpOverload(op='aten.rad2deg_', overload='default')>: <function rad2deg at 0x7d717ab63d80>, <OpOverload(op='aten.reciprocal_', overload='default')>: <function reciprocal at 0x7d717ab63ec0>, <OpOverload(op='aten.remainder_', overload='Tensor')>: <function remainder at 0x7d717ab94040>, <OpOverload(op='aten.remainder_', overload='Scalar')>: <function remainder at 0x7d717ab94040>, <OpOverload(op='aten.xlogy_', overload='Tensor')>: <function xlogy at 0x7d717ab613a0>, <OpOverload(op='aten.rsqrt_', overload='default')>: <function rsqrt at 0x7d717ab94180>, <OpOverload(op='aten.sgn_', overload='default')>: <function sgn at 0x7d717ab942c0>, <OpOverload(op='aten.sigmoid_', overload='default')>: <function sigmoid at 0x7d717ab94400>, <OpOverload(op='aten.trunc_', overload='default')>: <function trunc at 0x7d717ab61da0>, <OpOverload(op='aten.sign_', overload='default')>: <function sign at 0x7d717ab94540>, <OpOverload(op='aten.sin_', overload='default')>: <function sin at 0x7d717ab94680>, <OpOverload(op='aten.sinc_', overload='default')>: <function sinc at 0x7d717ab947c0>, <OpOverload(op='aten.sinh_', overload='default')>: <function sinh at 0x7d717ab94900>, <OpOverload(op='aten.true_divide_', overload='Tensor')>: <function true_divide at 0x7d717ab63ba0>, <OpOverload(op='aten.true_divide_', overload='Scalar')>: <function true_divide at 0x7d717ab63ba0>, <OpOverload(op='aten.sqrt_', overload='default')>: <function sqrt at 0x7d717ab94a40>, <OpOverload(op='aten.square_', overload='default')>: <function square at 0x7d717ab94b80>, <OpOverload(op='aten.triu_', overload='default')>: <function triu at 0x7d717ab63ce0>, <OpOverload(op='aten.tril_', overload='default')>: <function tril at 0x7d717ab63f60>, <OpOverload(op='aten.sub_', overload='Tensor')>: <function sub at 0x7d717ab94cc0>, <OpOverload(op='aten.sub_', overload='Scalar')>: <function sub at 0x7d717ab94cc0>, <OpOverload(op='aten.tan_', overload='default')>: <function tan at 0x7d717ab94e00>, <OpOverload(op='aten.alias_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7d717ab94fe0>, <OpOverload(op='aten.alias_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.alias at 0x7d717ab94fe0>, <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 0x7d717ab949a0>, <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 0x7d717ab949a0>, <OpOverload(op='aten.diagonal_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.diagonal at 0x7d717ab94360>, <OpOverload(op='aten.expand_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7d717ab95300>, <OpOverload(op='aten.expand_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.expand at 0x7d717ab95300>, <OpOverload(op='aten.t_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7d717ab95f80>, <OpOverload(op='aten.narrow_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7d717ab95620>, <OpOverload(op='aten.squeeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='dim')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='dims')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <OpOverload(op='aten.squeeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.squeeze at 0x7d717ab95940>, <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 0x7d717ab95940>, <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 0x7d717ab95940>, <OpOverload(op='aten.permute_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7d717ab95c60>, <OpOverload(op='aten.permute_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.permute at 0x7d717ab95c60>, <OpOverload(op='aten.t_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.t at 0x7d717ab95f80>, <OpOverload(op='aten.unbind_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unbind at 0x7d717ab60ea0>, <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 0x7d717ab60ea0>, <OpOverload(op='aten.unsqueeze_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7d717ab962a0>, <OpOverload(op='aten.unsqueeze_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.unsqueeze at 0x7d717ab962a0>, <OpOverload(op='aten.view_copy', overload='dtype')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <OpOverload(op='aten.view_copy', overload='out')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.view at 0x7d717ab959e0>, <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 0x7d717ab959e0>, <OpOverload(op='aten.fft_ifft', overload='out')>: <function ifft at 0x7d717ab977e0>, <OpOverload(op='aten.fft_ifft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90459e0>, kernel=<OpOverload(op='aten.fft_ifft', overload='default')>), <OpOverload(op='aten.polar', overload='out')>: <function polar at 0x7d717ab97880>, <OpOverload(op='aten.fft_fft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046200>, kernel=<OpOverload(op='aten.fft_fft', overload='default')>), <OpOverload(op='aten.fft_fft', overload='out')>: <function fft at 0x7d717abcc360>, <OpOverload(op='aten.fft_rfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045760>, kernel=<OpOverload(op='aten.fft_rfft', overload='default')>), <OpOverload(op='aten.fft_rfft', overload='out')>: <function rfft at 0x7d717abcc400>, <OpOverload(op='aten.fft_irfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90454e0>, kernel=<OpOverload(op='aten.fft_irfft', overload='default')>), <OpOverload(op='aten.fft_irfft', overload='out')>: <function irfft at 0x7d717abcc680>, <OpOverload(op='aten.fft_hfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9045260>, kernel=<OpOverload(op='aten.fft_hfft', overload='default')>), <OpOverload(op='aten.fft_hfft', overload='out')>: <function hfft at 0x7d717abcc900>, <OpOverload(op='aten.fft_ihfft', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044540>, kernel=<OpOverload(op='aten.fft_ihfft', overload='default')>), <OpOverload(op='aten.fft_ihfft', overload='out')>: <function ihfft at 0x7d717abccb80>, <OpOverload(op='aten.fft_fft2', overload='out')>: <function fft2 at 0x7d717abcdb20>, <OpOverload(op='aten.fft_fftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9044360>, kernel=<OpOverload(op='aten.fft_fftn', overload='default')>), <OpOverload(op='aten.fft_fftn', overload='out')>: <function fftn at 0x7d717abcd440>, <OpOverload(op='aten.fft_fft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903db20>, kernel=<OpOverload(op='aten.fft_fft2', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903fb00>, kernel=<OpOverload(op='aten.fft_ifftn', overload='default')>), <OpOverload(op='aten.fft_ifftn', overload='out')>: <function ifftn at 0x7d717abcd6c0>, <OpOverload(op='aten.fft_rfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f880>, kernel=<OpOverload(op='aten.fft_rfftn', overload='default')>), <OpOverload(op='aten.fft_rfftn', overload='out')>: <function rfftn at 0x7d717abcd940>, <OpOverload(op='aten.fft_ihfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f100>, kernel=<OpOverload(op='aten.fft_ihfftn', overload='default')>), <OpOverload(op='aten.fft_ihfftn', overload='out')>: <function ihfftn at 0x7d717abcdbc0>, <OpOverload(op='aten.fft_irfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903f1a0>, kernel=<OpOverload(op='aten.fft_irfftn', overload='default')>), <OpOverload(op='aten.fft_irfftn', overload='out')>: <function irfftn at 0x7d717abce2a0>, <OpOverload(op='aten.fft_hfftn', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903de40>, kernel=<OpOverload(op='aten.fft_hfftn', overload='default')>), <OpOverload(op='aten.fft_hfftn', overload='out')>: <function hfftn at 0x7d717abce520>, <OpOverload(op='aten.fft_ifft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903d620>, kernel=<OpOverload(op='aten.fft_ifft2', overload='default')>), <OpOverload(op='aten.fft_ifft2', overload='out')>: <function ifft2 at 0x7d717abccae0>, <OpOverload(op='aten.fft_rfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903ce00>, kernel=<OpOverload(op='aten.fft_rfft2', overload='default')>), <OpOverload(op='aten.fft_rfft2', overload='out')>: <function rfft2 at 0x7d717abce5c0>, <OpOverload(op='aten.fft_irfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c720>, kernel=<OpOverload(op='aten.fft_irfft2', overload='default')>), <OpOverload(op='aten.fft_irfft2', overload='out')>: <function irfft2 at 0x7d717abce840>, <OpOverload(op='aten.fft_hfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b903c360>, kernel=<OpOverload(op='aten.fft_hfft2', overload='default')>), <OpOverload(op='aten.fft_hfft2', overload='out')>: <function hfft2 at 0x7d717abceac0>, <OpOverload(op='aten.fft_ihfft2', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033e20>, kernel=<OpOverload(op='aten.fft_ihfft2', overload='default')>), <OpOverload(op='aten.fft_ihfft2', overload='out')>: <function ihfft2 at 0x7d717abced40>, <OpOverload(op='aten.fft_fftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033b00>, kernel=<OpOverload(op='aten.fft_fftshift', overload='default')>), <OpOverload(op='aten.linalg_cross', overload='out')>: <function cross at 0x7d717abcf560>, <OpOverload(op='aten.linalg_vector_norm', overload='out')>: <function vector_norm at 0x7d717abcf920>, <OpOverload(op='aten.alpha_dropout', overload='default')>: <function alpha_dropout at 0x7d717abf4d60>, <OpOverload(op='aten.is_complex', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055300>, kernel=<OpOverload(op='aten.is_complex', overload='default')>), <OpOverload(op='aten.erfinv', overload='default')>: <function erfinv at 0x7d717ac4c2c0>, <OpOverload(op='aten.erfinv', overload='out')>: <function erfinv at 0x7d717ac4c2c0>, <OpOverload(op='aten.logsumexp', overload='names_out')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.zero', overload='default')>: <function zero at 0x7d717ac4dd00>, <OpOverload(op='aten.zero', overload='out')>: <function zero at 0x7d717ac4dd00>, <OpOverload(op='aten.cauchy', overload='out')>: <function cauchy at 0x7d717ab32700>, <OpOverload(op='aten.frac', overload='out')>: <function frac at 0x7d717ac4e700>, <OpOverload(op='aten.isinf', overload='out')>: <function isinf at 0x7d717ac4f100>, <OpOverload(op='aten.isposinf', overload='out')>: <function isposinf at 0x7d717ac4d760>, <OpOverload(op='aten.isneginf', overload='out')>: <function isneginf at 0x7d717ac4f420>, <OpOverload(op='aten.isnan', overload='out')>: <function isnan at 0x7d717ac4f920>, <OpOverload(op='aten.i0', overload='default')>: <function i0 at 0x7d717ac64360>, <OpOverload(op='aten.i0', overload='out')>: <function i0 at 0x7d717ac64360>, <OpOverload(op='aten.all', overload='out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.logsumexp', overload='out')>: <function logsumexp at 0x7d717ac66200>, <OpOverload(op='aten.sinc', overload='out')>: <function sinc at 0x7d717ac791c0>, <OpOverload(op='aten.logsumexp', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904ea20>, kernel=<OpOverload(op='aten.logsumexp', overload='names')>), <OpOverload(op='aten.nan_to_num', overload='default')>: <function nan_to_num at 0x7d717ac65800>, <OpOverload(op='aten.nan_to_num', overload='out')>: <function nan_to_num at 0x7d717ac65800>, <OpOverload(op='aten.selu_', overload='default')>: <function selu at 0x7d717aa2b060>, <OpOverload(op='aten.sigmoid', overload='out')>: <function sigmoid at 0x7d717ac67880>, <OpOverload(op='aten.ge_', overload='Scalar')>: <function ge at 0x7d717ab62e80>, <OpOverload(op='aten.sgn', overload='out')>: <function sgn at 0x7d717ac67d80>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_Scalar_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Scalar_Tensor_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.normal_', overload='default')>: <function normal_ at 0x7d717ab332e0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Tensor_Scalar_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_left_shift', overload='Scalar_Tensor_out')>: <function bitwise_left_shift at 0x7d717ac7b9c0>, <OpOverload(op='aten.bitwise_right_shift', overload='Tensor_out')>: <function bitwise_right_shift at 0x7d717ac902c0>, <OpOverload(op='aten.copysign', overload='Tensor')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='Scalar')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='out')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.copysign', overload='Scalar_out')>: <function copysign at 0x7d717ac904a0>, <OpOverload(op='aten.any', overload='out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.heaviside', overload='out')>: <function heaviside at 0x7d717ac936a0>, <OpOverload(op='aten.complex', overload='out')>: <function complex at 0x7d717ab97600>, <OpOverload(op='aten.lcm', overload='default')>: <function lcm at 0x7d717aca49a0>, <OpOverload(op='aten.lcm', overload='out')>: <function lcm at 0x7d717aca49a0>, <OpOverload(op='aten.logaddexp', overload='out')>: <function logaddexp at 0x7d717aca5260>, <OpOverload(op='aten.logaddexp2', overload='out')>: <function logaddexp2 at 0x7d717aca56c0>, <OpOverload(op='aten.logical_not', overload='out')>: <function logical_not at 0x7d717aca5760>, <OpOverload(op='aten.logical_or', overload='out')>: <function logical_or at 0x7d717aca5d00>, <OpOverload(op='aten.logical_xor', overload='out')>: <function logical_xor at 0x7d717aca6160>, <OpOverload(op='aten.renorm', overload='default')>: <function renorm at 0x7d717ace8220>, <OpOverload(op='aten.exp2_', overload='default')>: <function exp2 at 0x7d717ab62480>, <OpOverload(op='aten.rsub', overload='Tensor')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Scalar')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Tensor_out')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.rsub', overload='Scalar_out')>: <function rsub at 0x7d717acc02c0>, <OpOverload(op='aten.xlogy', overload='OutTensor')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutScalar_Self')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.xlogy', overload='OutScalar_Other')>: <function xlogy at 0x7d717acc0720>, <OpOverload(op='aten.addcdiv', overload='out')>: <function addcdiv at 0x7d717acc0ea0>, <OpOverload(op='aten.addcmul', overload='out')>: <function addcmul at 0x7d717acc1260>, <OpOverload(op='aten.clamp', overload='out')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp', overload='Tensor_out')>: <function clamp at 0x7d717acc1620>, <OpOverload(op='aten.clamp_min', overload='out')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_min', overload='Tensor_out')>: <function clamp_min at 0x7d717acc18a0>, <OpOverload(op='aten.clamp_max', overload='out')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.clamp_max', overload='Tensor_out')>: <function clamp_max at 0x7d717acc1b20>, <OpOverload(op='aten.all', overload='dims_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='all_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dimname')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.all', overload='dimname_out')>: <function all at 0x7d717acc2c00>, <OpOverload(op='aten.any', overload='dims_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='all_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.any', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904cf40>, kernel=<OpOverload(op='aten.any', overload='dimname')>), <OpOverload(op='aten.any', overload='dimname_out')>: <function any at 0x7d717acc2e80>, <OpOverload(op='aten.std_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047d80>, kernel=<OpOverload(op='aten.std_mean', overload='correction_names')>), <OpOverload(op='aten.std_mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047e20>, kernel=<OpOverload(op='aten.std_mean', overload='names_dim')>), <OpOverload(op='aten.std_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c220>, kernel=<OpOverload(op='aten.std_mean', overload='default')>), <OpOverload(op='aten.std_mean', overload='correction')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.std_mean', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047ce0>, kernel=<OpOverload(op='aten.std_mean', overload='dim')>), <OpOverload(op='aten.std', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904cfe0>, 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 0x7d70b904cae0>, kernel=<OpOverload(op='aten.std', overload='dim')>), <OpOverload(op='aten.std', overload='correction')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.std', overload='correction_names_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.mean', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c540>, kernel=<OpOverload(op='aten.mean', overload='names_dim')>), <OpOverload(op='aten.mean', overload='dtype_out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.mean', overload='names_out')>: <function mean at 0x7d717acc37e0>, <OpOverload(op='aten.var_mean', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9047880>, 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 0x7d70b90477e0>, kernel=<OpOverload(op='aten.var_mean', overload='dim')>), <OpOverload(op='aten.var_mean', overload='correction')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.var_mean', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90472e0>, 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 0x7d70b90474c0>, kernel=<OpOverload(op='aten.var_mean', overload='names_dim')>), <OpOverload(op='aten.var_mean', overload='correction_out')>: <function var_mean at 0x7d717acc3420>, <OpOverload(op='aten.addr', overload='out')>: <function addr at 0x7d717ace85e0>, <OpOverload(op='aten.native_layer_norm', overload='out')>: <function native_layer_norm at 0x7d717acc36a0>, <OpOverload(op='aten.index_select', overload='out')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.constant_pad_nd', overload='out')>: <function constant_pad_nd at 0x7d717ace96c0>, <OpOverload(op='aten.stft', overload='center')>: <function stft at 0x7d717ace9120>, <OpOverload(op='aten.index_select', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e700>, kernel=<OpOverload(op='aten.index_select', overload='dimname')>), <OpOverload(op='aten.flip', overload='out')>: <function flip at 0x7d717ace9ee0>, <OpOverload(op='aten.istft', overload='default')>: <function istft at 0x7d717aceaa20>, <OpOverload(op='aten.renorm', overload='out')>: <function renorm at 0x7d717ace8220>, <OpOverload(op='aten.repeat', overload='out')>: <function repeat at 0x7d717aceade0>, <OpOverload(op='aten.roll', overload='out')>: <function roll at 0x7d717aceb420>, <OpOverload(op='aten.rot90', overload='default')>: <function rot90 at 0x7d717aceb6a0>, <OpOverload(op='aten.stack', overload='out')>: <function stack at 0x7d717aceb9c0>, <OpOverload(op='aten.unbind', overload='Dimname')>: <function unbind at 0x7d717ad0c040>, <OpOverload(op='aten.index_fill', overload='int_Tensor')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Scalar')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='Dimname_Scalar')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f240>, 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 0x7d70b904f1a0>, kernel=<OpOverload(op='aten.index_fill', overload='Dimname_Tensor')>), <OpOverload(op='aten.index_fill', overload='int_Scalar_out')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill', overload='int_Tensor_out')>: <function index_fill at 0x7d717ad0c680>, <OpOverload(op='aten.index_fill_', overload='int_Tensor')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='int_Scalar')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='Dimname_Scalar')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.index_fill_', overload='Dimname_Tensor')>: <function index_fill_ at 0x7d717ad0c4a0>, <OpOverload(op='aten.diag', overload='out')>: <function diag at 0x7d717ad0c5e0>, <OpOverload(op='aten.index_select', overload='dimname_out')>: <function index_select at 0x7d717ad0ccc0>, <OpOverload(op='aten.diagonal_scatter', overload='out')>: <function diagonal_scatter at 0x7d717ad0cf40>, <OpOverload(op='aten.diagonal', overload='Dimname')>: <function diagonal at 0x7d717ad0c180>, <OpOverload(op='aten.diag_embed', overload='out')>: <function diag_embed at 0x7d717ad0d260>, <OpOverload(op='aten.block_diag', overload='default')>: <function _block_diag_iterable at 0x7d717ad0d620>, <OpOverload(op='aten.block_diag', overload='out')>: <function _block_diag_iterable at 0x7d717ad0d620>, <OpOverload(op='aten.arange', overload='start_out')>: <function arange at 0x7d717ad0f9c0>, <OpOverload(op='aten.unfold_copy', overload='out')>: <function unfold_copy at 0x7d717ad0df80>, <OpOverload(op='aten.cumsum', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d940>, kernel=<OpOverload(op='aten.cumsum', overload='dimname')>), <OpOverload(op='aten.cumsum', overload='dimname_out')>: <function cumsum at 0x7d717ad0e020>, <OpOverload(op='aten.unfold_backward', overload='out')>: <function unfold_backward at 0x7d717ad216c0>, <OpOverload(op='aten.native_dropout', overload='out')>: <function native_dropout at 0x7d717ad22020>, <OpOverload(op='aten._softmax', overload='out')>: <function _softmax at 0x7d717ad223e0>, <OpOverload(op='aten._log_softmax', overload='out')>: <function _log_softmax at 0x7d717ad22660>, <OpOverload(op='aten._chunk_cat', overload='default')>: <function _chunk_cat at 0x7d717ad22e80>, <OpOverload(op='aten._chunk_cat', overload='out')>: <function _chunk_cat at 0x7d717ad22e80>, <OpOverload(op='aten.embedding_dense_backward', overload='default')>: <function embedding_dense_backward at 0x7d717ad22b60>, <OpOverload(op='aten.embedding_dense_backward', overload='out')>: <function embedding_dense_backward at 0x7d717ad22b60>, <OpOverload(op='aten.split_with_sizes_copy', overload='default')>: <function split_with_sizes_copy at 0x7d717ad22f20>, <OpOverload(op='aten.unsafe_split_with_sizes', overload='default')>: <function unsafe_split_with_sizes at 0x7d717ad231a0>, <OpOverload(op='aten.native_group_norm_backward', overload='default')>: <function native_group_norm_backward at 0x7d717ad23740>, <OpOverload(op='aten._fused_dropout', overload='out')>: <function _fused_dropout_decomposition at 0x7d717ad23f60>, <OpOverload(op='aten._fused_dropout', overload='default')>: <function _fused_dropout_decomposition at 0x7d717ad23f60>, <OpOverload(op='aten.addmm', overload='dtype_out')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='out')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.addmm', overload='dtype')>: <function addmm at 0x7d717ad22160>, <OpOverload(op='aten.native_group_norm_backward', overload='out')>: <function native_group_norm_backward_out at 0x7d717ad237e0>, <OpOverload(op='aten._addmm_activation', overload='default')>: <function _addmm_activation at 0x7d717ad21440>, <OpOverload(op='aten.native_batch_norm', overload='out')>: <function native_batch_norm at 0x7d717ad485e0>, <OpOverload(op='aten._batch_norm_with_update', overload='default')>: <function _batch_norm_with_update at 0x7d717ad48f40>, <OpOverload(op='aten._batch_norm_no_update', overload='default')>: <function _batch_norm_no_update at 0x7d717ad49120>, <OpOverload(op='aten._batch_norm_with_update_functional', overload='default')>: <function _batch_norm_with_update_functional at 0x7d717ad48fe0>, <OpOverload(op='aten.cudnn_batch_norm', overload='default')>: <function cudnn_batch_norm at 0x7d717ad49b20>, <OpOverload(op='aten.batch_norm_backward', overload='default')>: <function batch_norm_backward at 0x7d717ad49d00>, <OpOverload(op='aten.native_batch_norm_backward', overload='out')>: <function native_batch_norm_backward_out at 0x7d717ad49ee0>, <OpOverload(op='aten.native_batch_norm_backward', overload='default')>: <function native_batch_norm_backward at 0x7d717ad49da0>, <OpOverload(op='aten.lift', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.lift', overload='out')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.cudnn_batch_norm', overload='out')>: <function cudnn_batch_norm at 0x7d717ad49b20>, <OpOverload(op='aten.max_unpool2d', overload='default')>: <function max_unpool2d at 0x7d717ad4a0c0>, <OpOverload(op='aten.max_unpool2d', overload='out')>: <function max_unpool2d at 0x7d717ad4a0c0>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='default')>: <function cudnn_batch_norm_backward at 0x7d717ad4ad40>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>: <function adaptive_avg_pool2d at 0x7d717ad4b1a0>, <OpOverload(op='aten._adaptive_avg_pool2d', overload='out')>: <function adaptive_avg_pool2d at 0x7d717ad4b1a0>, <OpOverload(op='aten.upsample_nearest2d', overload='out')>: <function upsample_nearest2d at 0x7d717ad6cea0>, <OpOverload(op='aten.max_unpool3d', overload='default')>: <function max_unpool3d at 0x7d717ad49440>, <OpOverload(op='aten.max_unpool3d', overload='out')>: <function max_unpool3d at 0x7d717ad49440>, <OpOverload(op='aten.index_add_', overload='default')>: <function index_add_ at 0x7d717ad49f80>, <OpOverload(op='aten.index_add', overload='default')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.index_add', overload='out')>: <function index_add at 0x7d717ad4b4c0>, <OpOverload(op='aten.index_add', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904fce0>, kernel=<OpOverload(op='aten.index_add', overload='dimname')>), <OpOverload(op='aten.index_copy_', overload='default')>: <function index_copy_ at 0x7d717ad4b560>, <OpOverload(op='aten.index_copy', overload='default')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.index_copy', overload='out')>: <function index_copy at 0x7d717ad4bb00>, <OpOverload(op='aten.log_sigmoid_forward', overload='default')>: <function log_sigmoid_forward at 0x7d717ad4bc40>, <OpOverload(op='aten.uniform_', overload='default')>: <function uniform_ at 0x7d717ad6c400>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='out')>: <function upsample_nearest_exact1d at 0x7d717ad49c60>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='default')>: <function _upsample_nearest_exact3d at 0x7d717ad6da80>, <OpOverload(op='aten.upsample_nearest1d', overload='out')>: <function upsample_nearest1d at 0x7d717ad6d580>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='vec')>: <function _upsample_nearest_exact_vec at 0x7d717ad6d1c0>, <OpOverload(op='aten.upsample_nearest3d', overload='out')>: <function upsample_nearest3d at 0x7d717ad6d760>, <OpOverload(op='aten._upsample_nearest_exact3d', overload='out')>: <function _upsample_nearest_exact3d at 0x7d717ad6da80>, <OpOverload(op='aten.rnn_tanh', overload='input')>: <function rnn_tanh_input at 0x7d717ad6e2a0>, <OpOverload(op='aten.rnn_relu', overload='input')>: <function rnn_relu_input at 0x7d717ad6e480>, <OpOverload(op='aten.rnn_relu', overload='data')>: <function rnn_relu_data at 0x7d717ad6e660>, <OpOverload(op='aten.rnn_tanh', overload='data')>: <function rnn_tanh_data at 0x7d717ad6e840>, <OpOverload(op='aten._unsafe_masked_index_put_accumulate', overload='default')>: <function _unsafe_masked_index_put_accumulate at 0x7d717ad6e7a0>, <OpOverload(op='aten.gru', overload='data')>: <function gru_impl_data at 0x7d717ad6f1a0>, <OpOverload(op='aten._upsample_bilinear2d_aa', overload='vec')>: <function upsample_bilinear2d_aa_vec at 0x7d717ad6f560>, <OpOverload(op='aten._unsafe_masked_index', overload='default')>: <function _unsafe_masked_index at 0x7d717ad6ede0>, <OpOverload(op='aten._upsample_bicubic2d_aa', overload='vec')>: <function upsample_bicubic2d_aa_vec at 0x7d717ad6f740>, <OpOverload(op='aten.upsample_trilinear3d', overload='out')>: <function upsample_trilinear3d at 0x7d717ad8c400>, <OpOverload(op='aten.upsample_linear1d', overload='out')>: <function upsample_linear1d at 0x7d717ad6fe20>, <OpOverload(op='aten.upsample_linear1d', overload='vec')>: <function _upsample_linear_vec at 0x7d717ad6fc40>, <OpOverload(op='aten.nll_loss_forward', overload='output')>: <function nll_loss_forward at 0x7d717ad6c720>, <OpOverload(op='aten.nll_loss2d_forward', overload='output')>: <function nll_loss2d_forward at 0x7d717ad8d080>, <OpOverload(op='aten.nll_loss2d_forward', overload='default')>: <function nll_loss2d_forward at 0x7d717ad8d080>, <OpOverload(op='aten.affine_grid_generator', overload='default')>: <function affine_grid_generator at 0x7d717ad8db20>, <OpOverload(op='aten.affine_grid_generator', overload='out')>: <function affine_grid_generator at 0x7d717ad8db20>, <OpOverload(op='aten.mv', overload='out')>: <function mv at 0x7d717ad8e200>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>: <function binary_cross_entropy_with_logits at 0x7d717ad8e480>, <OpOverload(op='aten.binary_cross_entropy_with_logits', overload='out')>: <function binary_cross_entropy_with_logits at 0x7d717ad8e480>, <OpOverload(op='aten.upsample_bicubic2d', overload='out')>: <function upsample_bicubic2d_default at 0x7d717ad6e5c0>, <OpOverload(op='aten.reflection_pad1d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad2d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.reflection_pad3d', overload='out')>: <function _reflection_pad at 0x7d717ad8cae0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='default')>: <function multilabel_margin_loss_forward at 0x7d717ad8ede0>, <OpOverload(op='aten.replication_pad1d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad2d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.replication_pad3d', overload='out')>: <function _replication_pad at 0x7d717ad8ef20>, <OpOverload(op='aten.multi_margin_loss', overload='out')>: <function multi_margin_loss at 0x7d717ad8f2e0>, <OpOverload(op='aten.reflection_pad1d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad2d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad3d_backward', overload='default')>: <function _reflection_pad_backward at 0x7d717ad8f420>, <OpOverload(op='aten.reflection_pad3d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f420>, <OpOverload(op='aten.reflection_pad2d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.reflection_pad1d_backward', overload='grad_input')>: <function _reflection_pad_backward at 0x7d717ad8f600>, <OpOverload(op='aten.arange', overload='out')>: <function arange_default at 0x7d717adb42c0>, <OpOverload(op='aten.aminmax', overload='default')>: <function aminmax at 0x7d717ad8fc40>, <OpOverload(op='aten.aminmax', overload='out')>: <function aminmax at 0x7d717ad8fc40>, <OpOverload(op='aten.nansum', overload='default')>: <function nansum at 0x7d717ad8ff60>, <OpOverload(op='aten.nansum', overload='out')>: <function nansum at 0x7d717ad8ff60>, <OpOverload(op='aten.multi_margin_loss', overload='default')>: <function multi_margin_loss at 0x7d717ad8f2e0>, <OpOverload(op='aten.baddbmm', overload='dtype_out')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='out')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.baddbmm', overload='dtype')>: <function baddbmm at 0x7d717adb4c20>, <OpOverload(op='aten.__iand__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb62a0>, <OpOverload(op='aten.floor_divide', overload='out')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.floor_divide', overload='Scalar_out')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.__iand__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb62a0>, <OpOverload(op='aten.sgn', overload='default')>: <function sgn at 0x7d717ac67d80>, <OpOverload(op='aten._weight_norm_interface', overload='default')>: <function _weight_norm_interface at 0x7d717adb51c0>, <OpOverload(op='aten._weight_norm_interface', overload='out')>: <function _weight_norm_interface at 0x7d717adb51c0>, <OpOverload(op='aten.isin', overload='Tensor_Tensor')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Scalar')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Scalar_Tensor')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Tensor_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Tensor_Scalar_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.isin', overload='Scalar_Tensor_out')>: <function isin at 0x7d717adb5620>, <OpOverload(op='aten.take', overload='out')>: <function take at 0x7d717adb5b20>, <OpOverload(op='aten.hardsigmoid_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6480>, <OpOverload(op='aten.resize_as', overload='default')>: <function resize_as at 0x7d717adb5940>, <OpOverload(op='aten.resize_as', overload='out')>: <function resize_as at 0x7d717adb5940>, <OpOverload(op='aten.hardtanh_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5800>, <OpOverload(op='aten.addmv_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5e40>, <OpOverload(op='aten.fill_', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb60c0>, <OpOverload(op='aten.fill_', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb60c0>, <OpOverload(op='aten.hardswish_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6340>, <OpOverload(op='aten.gelu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6200>, <OpOverload(op='aten.__ilshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6020>, <OpOverload(op='aten.__ilshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6020>, <OpOverload(op='aten.__ior__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5580>, <OpOverload(op='aten.__ior__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5580>, <OpOverload(op='aten.__irshift__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5260>, <OpOverload(op='aten.__irshift__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5260>, <OpOverload(op='aten.__ixor__', overload='Tensor')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4fe0>, <OpOverload(op='aten.__ixor__', overload='Scalar')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4fe0>, <OpOverload(op='aten.leaky_relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4b80>, <OpOverload(op='aten.logit_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb4220>, <OpOverload(op='aten.relu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6660>, <OpOverload(op='aten.round_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb68e0>, <OpOverload(op='aten.round_', overload='decimals')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb68e0>, <OpOverload(op='aten.scatter_', overload='src')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_', overload='reduce')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_reduce_', overload='two')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6ca0>, <OpOverload(op='aten.index_copy_', overload='dimname')>: <function index_copy_ at 0x7d717ad4b560>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='out')>: <function miopen_batch_norm_backward at 0x7d717ad4a660>, <OpOverload(op='aten.scatter_', overload='value')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.scatter_', overload='value_reduce')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6a20>, <OpOverload(op='aten.where', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904d1c0>, kernel=<OpOverload(op='aten.where', overload='default')>), <OpOverload(op='aten.where', overload='self_out')>: <function where at 0x7d717acc2160>, <OpOverload(op='aten.count_nonzero', overload='out')>: <function count_nonzero at 0x7d717ab60400>, <OpOverload(op='aten.std', overload='names_out')>: <function std at 0x7d717acc39c0>, <OpOverload(op='aten.native_layer_norm_backward', overload='default')>: <function native_layer_norm_backward at 0x7d717ad23a60>, <OpOverload(op='aten.diag_embed', overload='default')>: <function diag_embed at 0x7d717ad0d260>, <OpOverload(op='aten.sum', overload='IntList_out')>: <function sum at 0x7d717acc2ca0>, <OpOverload(op='aten.sum', overload='dim_DimnameList')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9046f20>, kernel=<OpOverload(op='aten.sum', overload='dim_DimnameList')>), <OpOverload(op='aten.addmv', overload='out')>: <function addmv at 0x7d717ad236a0>, <OpOverload(op='aten.var', overload='dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90547c0>, kernel=<OpOverload(op='aten.var', overload='dim')>), <OpOverload(op='aten.addmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5d00>, <OpOverload(op='aten._euclidean_dist', overload='out')>: <function _euclidean_dist at 0x7d717aeeff60>, <OpOverload(op='aten.prod', overload='dim_Dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054f40>, kernel=<OpOverload(op='aten.prod', overload='dim_Dimname')>), <OpOverload(op='aten.prod', overload='int_out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.prod', overload='Dimname_out')>: <function prod at 0x7d717acc3060>, <OpOverload(op='aten.var', overload='correction_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054cc0>, kernel=<OpOverload(op='aten.var', overload='default')>), <OpOverload(op='aten.var', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054720>, kernel=<OpOverload(op='aten.var', overload='correction_names')>), <OpOverload(op='aten.baddbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5f80>, <OpOverload(op='aten.var', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9054860>, kernel=<OpOverload(op='aten.var', overload='names_dim')>), <OpOverload(op='aten.var', overload='out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.var', overload='correction_names_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.miopen_batch_norm_backward', overload='default')>: <function miopen_batch_norm_backward at 0x7d717ad4a660>, <OpOverload(op='aten.transpose_copy', overload='int')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.transpose at 0x7d717ab63420>, <OpOverload(op='aten.amax', overload='out')>: <function amax at 0x7d717acc3240>, <OpOverload(op='aten.hardsigmoid_backward', overload='grad_input')>: <function hardsigmoid_backward at 0x7d717aecb7e0>, <OpOverload(op='aten.amin', overload='out')>: <function amin at 0x7d717acc3100>, <OpOverload(op='aten.zero_', overload='default')>: <function zero at 0x7d717ab636a0>, <OpOverload(op='aten.log_sigmoid_backward', overload='grad_input')>: <function log_sigmoid_backward at 0x7d717aecb1a0>, <OpOverload(op='aten.full', overload='out')>: <function full at 0x7d717ad0efc0>, <OpOverload(op='aten.hardsigmoid_backward', overload='default')>: <function hardsigmoid_backward at 0x7d717aecb7e0>, <OpOverload(op='aten.upsample_bilinear2d', overload='out')>: <function upsample_bilinear2d at 0x7d717ad8c180>, <OpOverload(op='aten.grid_sampler_2d', overload='out')>: <function grid_sampler_2d at 0x7d717ad8dee0>, <OpOverload(op='aten.lstm', overload='data')>: <function lstm_data_impl at 0x7d717ad6ee80>, <OpOverload(op='aten._to_copy', overload='out')>: <function _to_copy at 0x7d717ad48c20>, <OpOverload(op='aten.svd', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904cc20>, kernel=<OpOverload(op='aten.svd', overload='default')>), <OpOverload(op='aten.softplus_backward', overload='default')>: <function softplus_backward at 0x7d717aecaca0>, <OpOverload(op='aten.dist', overload='out')>: <function dist at 0x7d717aeefce0>, <OpOverload(op='aten.normal', overload='Tensor_Tensor_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='Tensor_float_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.tanh_backward', overload='grad_input')>: <function tanh_backward at 0x7d717b313ce0>, <OpOverload(op='aten.std_mean', overload='correction_out')>: <function std_mean at 0x7d717acc3920>, <OpOverload(op='aten.normal', overload='float_Tensor_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='float_float_out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.normal', overload='out')>: <function normal at 0x7d717ab33600>, <OpOverload(op='aten.uniform', overload='default')>: <function uniform at 0x7d717ad6c5e0>, <OpOverload(op='aten._addmm_activation', overload='out')>: <function _addmm_activation at 0x7d717ad21440>, <OpOverload(op='aten.cudnn_batch_norm_backward', overload='out')>: <function cudnn_batch_norm_backward at 0x7d717ad4ad40>, <OpOverload(op='aten._prelu_kernel_backward', overload='default')>: <function _prelu_kernel_backward at 0x7d717aeed580>, <OpOverload(op='aten.floor_divide', overload='Scalar')>: <function floor_divide at 0x7d717adb4ea0>, <OpOverload(op='aten.embedding', overload='out')>: <function embedding at 0x7d717ad228e0>, <OpOverload(op='aten.frexp', overload='Tensor_out')>: <function frexp at 0x7d717ac92980>, <OpOverload(op='aten.take', overload='default')>: <function take at 0x7d717adb5b20>, <OpOverload(op='aten.broadcast_tensors', overload='default')>: <function broadcast_tensors at 0x7d717ace8cc0>, <OpOverload(op='aten.multilabel_margin_loss_forward', overload='output')>: <function multilabel_margin_loss_forward at 0x7d717ad8ede0>, <OpOverload(op='aten.norm', overload='dtype_out')>: <function norm at 0x7d717ab31760>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='default')>: <function _upsample_nearest_exact2d at 0x7d717ad6c540>, <OpOverload(op='aten._upsample_nearest_exact2d', overload='out')>: <function _upsample_nearest_exact2d at 0x7d717ad6c540>, <OpOverload(op='aten._reshape_alias', overload='default')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='default')>: <function binary_cross_entropy_backward at 0x7d717aeef7e0>, <OpOverload(op='aten.log_sigmoid_forward', overload='output')>: <function log_sigmoid_forward at 0x7d717ad4bc40>, <OpOverload(op='aten.sigmoid_backward', overload='grad_input')>: <function sigmoid_backward at 0x7d717aeca8e0>, <OpOverload(op='aten.sigmoid_backward', overload='default')>: <function sigmoid_backward at 0x7d717aeca8e0>, <OpOverload(op='aten.elu_backward', overload='grad_input')>: <function elu_backward at 0x7d717aeca840>, <OpOverload(op='aten.elu_backward', overload='default')>: <function elu_backward at 0x7d717aeca840>, <OpOverload(op='aten.hardsigmoid', overload='out')>: <function hardsigmoid at 0x7d717aecb880>, <OpOverload(op='aten.hardtanh_backward', overload='grad_input')>: <function hardtanh_backward at 0x7d717aeca660>, <OpOverload(op='aten.hardswish', overload='out')>: <function hardswish at 0x7d717aecbe20>, <OpOverload(op='aten.hardswish_backward', overload='default')>: <function hardswish_backward at 0x7d717aeec180>, <OpOverload(op='aten.hardswish_backward', overload='out')>: <function hardswish_backward at 0x7d717aeec180>, <OpOverload(op='aten.threshold_backward', overload='default')>: <function threshold_backward at 0x7d717aeec220>, <OpOverload(op='aten.threshold_backward', overload='grad_input')>: <function threshold_backward at 0x7d717aeec220>, <OpOverload(op='aten.leaky_relu_backward', overload='default')>: <function leaky_relu_backward at 0x7d717aeec540>, <OpOverload(op='aten.leaky_relu_backward', overload='grad_input')>: <function leaky_relu_backward at 0x7d717aeec540>, <OpOverload(op='aten.gelu_backward', overload='default')>: <function gelu_backward at 0x7d717aeec900>, <OpOverload(op='aten.gelu_backward', overload='grad_input')>: <function gelu_backward at 0x7d717aeec900>, <OpOverload(op='aten.silu', overload='out')>: <function silu at 0x7d717aeed120>, <OpOverload(op='aten.silu_backward', overload='grad_input')>: <function silu_backward at 0x7d717aeed1c0>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='default')>: <function rrelu_with_noise_backward at 0x7d717aeeda80>, <OpOverload(op='aten.rrelu_with_noise_backward', overload='out')>: <function rrelu_with_noise_backward at 0x7d717aeeda80>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='default')>: <function smooth_l1_loss_backward at 0x7d717aeee3e0>, <OpOverload(op='aten.log_sigmoid_backward', overload='default')>: <function log_sigmoid_backward at 0x7d717aecb1a0>, <OpOverload(op='aten.im2col', overload='out')>: <function im2col at 0x7d717ad21620>, <OpOverload(op='aten.mse_loss', overload='out')>: <function mse_loss at 0x7d717aeedb20>, <OpOverload(op='aten.huber_loss_backward', overload='default')>: <function huber_loss_backward at 0x7d717aeee7a0>, <OpOverload(op='aten.mse_loss_backward', overload='default')>: <function mse_loss_backward at 0x7d717aeedbc0>, <OpOverload(op='aten.smooth_l1_loss_backward', overload='grad_input')>: <function smooth_l1_loss_backward_out at 0x7d717aeee5c0>, <OpOverload(op='aten._safe_softmax', overload='default')>: <function safe_softmax at 0x7d717aeedee0>, <OpOverload(op='aten.smooth_l1_loss', overload='default')>: <function smooth_l1_loss at 0x7d717aeee340>, <OpOverload(op='aten.smooth_l1_loss', overload='out')>: <function smooth_l1_loss at 0x7d717aeee340>, <OpOverload(op='aten.huber_loss_backward', overload='out')>: <function huber_loss_backward_out at 0x7d717aeee980>, <OpOverload(op='aten.glu_backward', overload='default')>: <function glu_backward at 0x7d717aeeec00>, <OpOverload(op='aten.glu_backward', overload='grad_input')>: <function glu_backward at 0x7d717aeeec00>, <OpOverload(op='aten.nll_loss_backward', overload='grad_input')>: <function nll_loss_backward at 0x7d717aeeefc0>, <OpOverload(op='aten.nll_loss2d_backward', overload='default')>: <function nll_loss2d_backward at 0x7d717aeef2e0>, <OpOverload(op='aten.nll_loss2d_backward', overload='grad_input')>: <function nll_loss2d_backward at 0x7d717aeef2e0>, <OpOverload(op='aten.binary_cross_entropy', overload='default')>: <function binary_cross_entropy at 0x7d717aeef880>, <OpOverload(op='aten.binary_cross_entropy', overload='out')>: <function binary_cross_entropy at 0x7d717aeef880>, <OpOverload(op='aten.soft_margin_loss', overload='default')>: <function soft_margin_loss at 0x7d717aeedda0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='grad_input')>: <function soft_margin_loss_backward at 0x7d717aeed9e0>, <OpOverload(op='aten.col2im', overload='out')>: <function col2im at 0x7d717ad21580>, <OpOverload(op='aten.dist', overload='default')>: <function dist at 0x7d717aeefce0>, <OpOverload(op='aten._euclidean_dist', overload='default')>: <function _euclidean_dist at 0x7d717aeeff60>, <OpOverload(op='aten.slice_backward', overload='out')>: <function slice_backward at 0x7d717ad20220>, <OpOverload(op='aten.select_backward', overload='default')>: <function select_backward at 0x7d717ad20900>, <OpOverload(op='aten.select_backward', overload='out')>: <function select_backward at 0x7d717ad20900>, <OpOverload(op='aten.diagonal_backward', overload='default')>: <function diagonal_backward at 0x7d717ad20b80>, <OpOverload(op='aten.diagonal_backward', overload='out')>: <function diagonal_backward at 0x7d717ad20b80>, <OpOverload(op='aten.logit_backward', overload='default')>: <function logit_backward at 0x7d717ad21760>, <OpOverload(op='aten._softmax_backward_data', overload='default')>: <function _softmax_backward_data at 0x7d717ad20cc0>, <OpOverload(op='aten._softmax_backward_data', overload='out')>: <function _softmax_backward_data at 0x7d717ad20cc0>, <OpOverload(op='aten._log_softmax_backward_data', overload='default')>: <function _log_softmax_backward_data at 0x7d717ad21300>, <OpOverload(op='aten._log_softmax_backward_data', overload='out')>: <function _log_softmax_backward_data at 0x7d717ad21300>, <OpOverload(op='aten.native_dropout_backward', overload='default')>: <function native_dropout_backward at 0x7d717ad20860>, <OpOverload(op='aten.native_dropout_backward', overload='out')>: <function native_dropout_backward at 0x7d717ad20860>, <OpOverload(op='aten.unfold_backward', overload='default')>: <function unfold_backward at 0x7d717ad216c0>, <OpOverload(op='aten.eq', overload='Tensor_out')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.sin', overload='out')>: <function sin at 0x7d717ac78cc0>, <OpOverload(op='aten.sign', overload='out')>: <function sign at 0x7d717ac782c0>, <OpOverload(op='aten.signbit', overload='default')>: <function signbit at 0x7d717ac787c0>, <OpOverload(op='aten.signbit', overload='out')>: <function signbit at 0x7d717ac787c0>, <OpOverload(op='aten.fmin', overload='out')>: <function fmin at 0x7d717ac91f80>, <OpOverload(op='aten.sinh', overload='out')>: <function sinh at 0x7d717ac66b60>, <OpOverload(op='aten.pow', overload='Scalar_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.div', overload='out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.sqrt', overload='out')>: <function sqrt at 0x7d717ac794e0>, <OpOverload(op='aten.minimum', overload='out')>: <function minimum at 0x7d717aca6e80>, <OpOverload(op='aten.bitwise_xor_', overload='Tensor')>: <function bitwise_xor at 0x7d717ab61bc0>, <OpOverload(op='aten.le', overload='Tensor_out')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.add', overload='out')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.add', overload='Scalar_out')>: <function add at 0x7d717ac7aca0>, <OpOverload(op='aten.atan2', overload='out')>: <function atan2 at 0x7d717ac7b100>, <OpOverload(op='aten.bitwise_and', overload='Scalar_Tensor_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.mse_loss_backward', overload='grad_input')>: <function mse_loss_backward at 0x7d717aeedbc0>, <OpOverload(op='aten.bitwise_and', overload='Scalar_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.bitwise_or', overload='Scalar_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Tensor_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_or', overload='Scalar_Tensor_out')>: <function bitwise_or at 0x7d717ac7be20>, <OpOverload(op='aten.bitwise_xor', overload='Scalar_Tensor_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.bitwise_xor', overload='Tensor_out')>: <function bitwise_xor at 0x7d717ac79e40>, <OpOverload(op='aten.div', overload='out_mode')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.div', overload='Scalar_out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.eq', overload='Scalar_out')>: <function eq at 0x7d717ac90b80>, <OpOverload(op='aten.fmax', overload='default')>: <function fmax at 0x7d717ac91b20>, <OpOverload(op='aten.fmax', overload='out')>: <function fmax at 0x7d717ac91b20>, <OpOverload(op='aten.fmod', overload='Tensor_out')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.gcd', overload='default')>: <function gcd at 0x7d717ac92020>, <OpOverload(op='aten.gcd', overload='out')>: <function gcd at 0x7d717ac92020>, <OpOverload(op='aten.scatter_add_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6b60>, <OpOverload(op='aten.ge', overload='Tensor_out')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.ge', overload='Scalar_out')>: <function ge at 0x7d717ac92de0>, <OpOverload(op='aten.gt', overload='Scalar_out')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.gt', overload='Tensor_out')>: <function gt at 0x7d717ac93240>, <OpOverload(op='aten.hypot', overload='default')>: <function hypot at 0x7d717ac93b00>, <OpOverload(op='aten.hypot', overload='out')>: <function hypot at 0x7d717ac93b00>, <OpOverload(op='aten.index_reduce_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5a80>, <OpOverload(op='aten.narrow_copy', overload='default')>: <function pybind11_detail_function_record_v1_system_libstdcpp_gxx_abi_1xxx_use_cxx11_abi_1.narrow at 0x7d717ab95620>, <OpOverload(op='aten.igamma', overload='default')>: <function igamma at 0x7d717ac93f60>, <OpOverload(op='aten.igamma', overload='out')>: <function igamma at 0x7d717ac93f60>, <OpOverload(op='aten.igammac', overload='default')>: <function igammac at 0x7d717aca4400>, <OpOverload(op='aten.igammac', overload='out')>: <function igammac at 0x7d717aca4400>, <OpOverload(op='aten.le', overload='Scalar_out')>: <function le at 0x7d717aca4e00>, <OpOverload(op='aten.pad_sequence', overload='default')>: <function pad_sequence at 0x7d717ad4b600>, <OpOverload(op='aten.lt', overload='Scalar_out')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.eye', overload='m')>: <function eye at 0x7d717ad0cea0>, <OpOverload(op='aten.lt', overload='Tensor_out')>: <function lt at 0x7d717aca65c0>, <OpOverload(op='aten.maximum', overload='out')>: <function maximum at 0x7d717aca6a20>, <OpOverload(op='aten.index_copy', overload='dimname')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904f560>, kernel=<OpOverload(op='aten.index_copy', overload='dimname')>), <OpOverload(op='aten.renorm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb67a0>, <OpOverload(op='aten.remainder', overload='Scalar_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.mul', overload='out')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.mul', overload='Scalar_out')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.logical_and', overload='out')>: <function logical_and at 0x7d717aca5b20>, <OpOverload(op='aten.std', overload='correction_names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c5e0>, kernel=<OpOverload(op='aten.std', overload='correction_names')>), <OpOverload(op='aten.ne', overload='Scalar_out')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.ne', overload='Tensor_out')>: <function ne at 0x7d717aca7740>, <OpOverload(op='aten.nextafter', overload='out')>: <function nextafter at 0x7d717aca7ba0>, <OpOverload(op='aten.nextafter', overload='default')>: <function nextafter at 0x7d717aca7ba0>, <OpOverload(op='aten.pow', overload='Tensor_Scalar_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.pow', overload='Tensor_Tensor_out')>: <function pow at 0x7d717ac90fe0>, <OpOverload(op='aten.frexp', overload='Tensor')>: <function frexp at 0x7d717ac92980>, <OpOverload(op='aten.remainder', overload='Scalar_Tensor_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten._upsample_nearest_exact1d', overload='default')>: <function upsample_nearest_exact1d at 0x7d717ad49c60>, <OpOverload(op='aten.remainder', overload='Tensor_out')>: <function remainder at 0x7d717aca7f60>, <OpOverload(op='aten.slice_backward', overload='default')>: <function slice_backward at 0x7d717ad20220>, <OpOverload(op='aten.is_same_size', overload='default')>: <function is_same_size at 0x7d717ad8c7c0>, <OpOverload(op='aten._unsafe_view', overload='out')>: <function _reshape_alias at 0x7d717ad6f6a0>, <OpOverload(op='aten.sub', overload='out')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.split_with_sizes_copy', overload='out')>: <function split_with_sizes_copy at 0x7d717ad22f20>, <OpOverload(op='aten.xlogy_', overload='Scalar_Other')>: <function xlogy at 0x7d717ab613a0>, <OpOverload(op='aten.sub', overload='Scalar_out')>: <function sub at 0x7d717aca6f20>, <OpOverload(op='aten.binary_cross_entropy_backward', overload='grad_input')>: <function binary_cross_entropy_backward at 0x7d717aeef7e0>, <OpOverload(op='aten.squeeze', overload='dims')>: <function squeeze at 0x7d717aceb600>, <OpOverload(op='aten.transpose', overload='Dimname')>: <function transpose at 0x7d717ad0d9e0>, <OpOverload(op='aten.softplus_backward', overload='grad_input')>: <function softplus_backward at 0x7d717aecaca0>, <OpOverload(op='aten.as_strided_scatter', overload='out')>: <function as_strided_scatter at 0x7d717ace8b80>, <OpOverload(op='aten.nll_loss_backward', overload='default')>: <function nll_loss_backward at 0x7d717aeeefc0>, <OpOverload(op='aten.cat', overload='names')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904e200>, kernel=<OpOverload(op='aten.cat', overload='names')>), <OpOverload(op='aten.cat', overload='out')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.empty_strided', overload='out')>: <function empty_strided at 0x7d717ab30900>, <OpOverload(op='aten.asin', overload='out')>: <function asin at 0x7d717ac193a0>, <OpOverload(op='aten.asinh', overload='out')>: <function asinh at 0x7d717ac198a0>, <OpOverload(op='aten.atan', overload='out')>: <function atan at 0x7d717ac19da0>, <OpOverload(op='aten.atanh', overload='out')>: <function atanh at 0x7d717ac1a2a0>, <OpOverload(op='aten._fused_rms_norm_backward', overload='default')>: <function _fused_rms_norm_backward at 0x7d717ad23d80>, <OpOverload(op='aten.fmin', overload='default')>: <function fmin at 0x7d717ac91f80>, <OpOverload(op='aten.cosh', overload='out')>: <function cosh at 0x7d717ac1ba60>, <OpOverload(op='aten.std', overload='names_dim')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904c7c0>, kernel=<OpOverload(op='aten.std', overload='names_dim')>), <OpOverload(op='aten.conj_physical', overload='default')>: <function conj_physical at 0x7d717ac1b060>, <OpOverload(op='aten.unfold_copy', overload='default')>: <function unfold_copy at 0x7d717ad0df80>, <OpOverload(op='aten.bitwise_not', overload='out')>: <function bitwise_not at 0x7d717ac1a7a0>, <OpOverload(op='aten.ceil', overload='out')>: <function ceil at 0x7d717ac1aca0>, <OpOverload(op='aten.conj_physical', overload='out')>: <function conj_physical at 0x7d717ac1b060>, <OpOverload(op='aten.digamma', overload='out')>: <function digamma at 0x7d717ac1bf60>, <OpOverload(op='aten.erf', overload='out')>: <function erf at 0x7d717ac19940>, <OpOverload(op='aten.uniform', overload='out')>: <function uniform at 0x7d717ad6c5e0>, <OpOverload(op='aten.erfc', overload='out')>: <function erfc at 0x7d717ac4c7c0>, <OpOverload(op='aten.fmod', overload='Scalar_out')>: <function fmod at 0x7d717ac923e0>, <OpOverload(op='aten.silu_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb6de0>, <OpOverload(op='aten.exp', overload='out')>: <function exp at 0x7d717ac4ccc0>, <OpOverload(op='aten.tanh', overload='out')>: <function tanh at 0x7d717ac7a2a0>, <OpOverload(op='aten.expm1', overload='out')>: <function expm1 at 0x7d717ac4d1c0>, <OpOverload(op='aten.exp2', overload='out')>: <function exp2 at 0x7d717ac4d6c0>, <OpOverload(op='aten.log10', overload='out')>: <function log10 at 0x7d717ac65c60>, <OpOverload(op='aten.floor', overload='out')>: <function floor at 0x7d717ac4e200>, <OpOverload(op='aten.lgamma', overload='out')>: <function lgamma at 0x7d717ac64860>, <OpOverload(op='aten.tanh_backward', overload='default')>: <function tanh_backward at 0x7d717b313ce0>, <OpOverload(op='aten.lgamma', overload='default')>: <function lgamma at 0x7d717ac64860>, <OpOverload(op='aten.log', overload='out')>: <function log at 0x7d717ac64d60>, <OpOverload(op='aten.log1p', overload='out')>: <function log1p at 0x7d717ac65260>, <OpOverload(op='aten.reciprocal', overload='out')>: <function reciprocal at 0x7d717ac66ac0>, <OpOverload(op='aten.trunc', overload='out')>: <function trunc at 0x7d717ac7a7a0>, <OpOverload(op='aten.log2', overload='out')>: <function log2 at 0x7d717ac65760>, <OpOverload(op='aten.soft_margin_loss', overload='out')>: <function soft_margin_loss at 0x7d717aeedda0>, <OpOverload(op='aten.neg', overload='out')>: <function neg at 0x7d717ac66480>, <OpOverload(op='aten.fft_ifftshift', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9033740>, kernel=<OpOverload(op='aten.fft_ifftshift', overload='default')>), <OpOverload(op='aten.round', overload='decimals_out')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.clamp_max_', overload='default')>: <function clamp_max at 0x7d717ab620c0>, <OpOverload(op='aten.rsqrt', overload='out')>: <function rsqrt at 0x7d717ac67380>, <OpOverload(op='aten.addbmm_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5bc0>, <OpOverload(op='aten.abs', overload='out')>: <function abs at 0x7d717adb63e0>, <OpOverload(op='aten.digamma', overload='default')>: <function digamma at 0x7d717ac1bf60>, <OpOverload(op='aten.acos', overload='out')>: <function acos at 0x7d717adb7f60>, <OpOverload(op='aten.round', overload='out')>: <function round at 0x7d717ac66e80>, <OpOverload(op='aten.sum', overload='out')>: <function sum_default at 0x7d717adb4f40>, <OpOverload(op='aten.native_layer_norm_backward', overload='out')>: <function native_layer_norm_backward_out at 0x7d717ad23b00>, <OpOverload(op='aten.acosh', overload='out')>: <function acosh at 0x7d717ac18ea0>, <OpOverload(op='aten.cos', overload='out')>: <function cos at 0x7d717ac1b560>, <OpOverload(op='aten.cat', overload='names_out')>: <function cat at 0x7d717ace91c0>, <OpOverload(op='aten.tan', overload='out')>: <function tan at 0x7d717ac79da0>, <OpOverload(op='aten.clone', overload='out')>: <function clone at 0x7d717aca6ac0>, <OpOverload(op='aten.div', overload='Scalar_mode_out')>: <function div at 0x7d717ac90720>, <OpOverload(op='aten.mul', overload='Scalar')>: <function mul at 0x7d717aca72e0>, <OpOverload(op='aten.silu_backward', overload='default')>: <function silu_backward at 0x7d717aeed1c0>, <OpOverload(op='aten._fused_rms_norm', overload='default')>: <function _fused_rms_norm at 0x7d717ad23c40>, <OpOverload(op='aten.mish_backward', overload='default')>: <function mish_backward at 0x7d717aeeccc0>, <OpOverload(op='aten.sym_numel', overload='default')>: <function sym_numel at 0x7d717adb4cc0>, <OpOverload(op='aten.soft_margin_loss_backward', overload='default')>: <function soft_margin_loss_backward at 0x7d717aeed9e0>, <OpOverload(op='aten.slice_scatter', overload='out')>: <function slice_scatter at 0x7d717ad20680>, <OpOverload(op='aten.index_put_', overload='default')>: <function register_inplace.<locals>.inplace_op at 0x7d717adb5da0>, <OpOverload(op='aten.var', overload='names_out')>: <function var at 0x7d717acc3740>, <OpOverload(op='aten.diagonal_scatter', overload='default')>: <function diagonal_scatter at 0x7d717ad0cf40>, <OpOverload(op='aten.bitwise_and', overload='Tensor_out')>: <function bitwise_and at 0x7d717ac7b560>, <OpOverload(op='aten.as_strided_scatter', overload='default')>: <function as_strided_scatter at 0x7d717ace8b80>, <OpOverload(op='aten.lift_fresh', overload='default')>: <function nop_decomposition at 0x7d717ad494e0>, <OpOverload(op='aten.empty_like', overload='out')>: <function empty_like at 0x7d717ad0f740>, <OpOverload(op='aten.ones_like', overload='out')>: <function ones_like at 0x7d717ab30c20>, <OpOverload(op='aten.zeros_like', overload='out')>: <function zeros_like at 0x7d717ab309a0>, <OpOverload(op='aten.new_empty', overload='out')>: <function new_empty at 0x7d717ad0ef20>, <OpOverload(op='aten.new_empty_strided', overload='out')>: <function new_empty_strided at 0x7d717ace9620>, <OpOverload(op='aten.new_full', overload='out')>: <function new_full at 0x7d717ad0f380>, <OpOverload(op='aten.new_zeros', overload='out')>: <function new_zeros at 0x7d717ad0dd00>, <OpOverload(op='aten.new_ones', overload='out')>: <function new_ones at 0x7d717ad0c7c0>, <OpOverload(op='aten.item', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b904db20>, 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 0x7d70b904d800>, kernel=<OpOverload(op='aten.nonzero_numpy', overload='default')>), <OpOverload(op='aten.rot90', overload='out')>: <function rot90 at 0x7d717aceb6a0>, <OpOverload(op='aten.le_', overload='Tensor')>: <function le at 0x7d717ab63880>, <OpOverload(op='quantized.conv2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057a60>, 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 0x7d70b90707c0>, 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 0x7d70b90702c0>, kernel=<OpOverload(op='quantized.conv_unpack', overload='default')>), <OpOverload(op='sparse.qlinear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057920>, kernel=<OpOverload(op='sparse.qlinear_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056520>, kernel=<OpOverload(op='quantized.conv_transpose2d_output_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056840>, kernel=<OpOverload(op='quantized.conv_transpose2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055da0>, 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 0x7d70b9057240>, kernel=<OpOverload(op='quantized.conv_transpose2d_groups', overload='default')>), <OpOverload(op='quantized.conv3d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90700e0>, 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 0x7d70b9057f60>, 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 0x7d70b90709a0>, 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 0x7d70b9070680>, kernel=<OpOverload(op='quantized.conv3d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90704a0>, kernel=<OpOverload(op='quantized.conv_transpose1d_unpack', overload='default')>), <OpOverload(op='quantized.conv2d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070540>, kernel=<OpOverload(op='quantized.conv2d_dilation', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90579c0>, 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 0x7d70b9056a20>, kernel=<OpOverload(op='quantized.conv_transpose3d_groups', overload='default')>), <OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90562a0>, kernel=<OpOverload(op='quantized.make_quantized_cell_params_fp16', overload='default')>), <OpOverload(op='quantized.conv2d_groups', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056020>, kernel=<OpOverload(op='quantized.conv2d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90563e0>, kernel=<OpOverload(op='quantized.conv_transpose3d_transpose', overload='default')>), <OpOverload(op='quantized.conv2d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057380>, kernel=<OpOverload(op='quantized.conv2d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b90567a0>, kernel=<OpOverload(op='quantized.conv_transpose3d_padding', overload='default')>), <OpOverload(op='quantized.conv2d_output_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057600>, 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 0x7d70b9056700>, kernel=<OpOverload(op='quantized.conv3d_dilation', overload='default')>), <OpOverload(op='quantized.conv2d_padding', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070720>, kernel=<OpOverload(op='quantized.conv2d_padding', overload='default')>), <OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057ce0>, kernel=<OpOverload(op='quantized.conv_transpose2d_transpose', overload='default')>), <OpOverload(op='quantized.conv1d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057560>, kernel=<OpOverload(op='quantized.conv1d_unpack', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057880>, kernel=<OpOverload(op='quantized.conv_transpose3d_stride', overload='default')>), <OpOverload(op='quantized.linear_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070900>, kernel=<OpOverload(op='quantized.linear_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 0x7d70b90577e0>, 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 0x7d70b90568e0>, kernel=<OpOverload(op='quantized.conv3d_groups', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9055f80>, kernel=<OpOverload(op='quantized.conv_transpose3d_unpack', overload='default')>), <OpOverload(op='quantized.embedding_bag_unpack', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070860>, kernel=<OpOverload(op='quantized.embedding_bag_unpack', overload='default')>), <OpOverload(op='quantized.conv3d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056d40>, kernel=<OpOverload(op='quantized.conv3d_stride', overload='default')>), <OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9057ec0>, kernel=<OpOverload(op='quantized.conv_transpose3d_dilation', overload='default')>), <OpOverload(op='quantized.linear_unpack_fp16', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9056c00>, kernel=<OpOverload(op='quantized.linear_unpack_fp16', overload='default')>), <OpOverload(op='quantized.conv2d_stride', overload='default')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070180>, kernel=<OpOverload(op='quantized.conv2d_stride', overload='default')>), <OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>: functools.partial(<function _get_decomp_for_cia.<locals>._special_op_to_decompose_cia at 0x7d70b9070a40>, kernel=<OpOverload(op='profiler._record_function_exit', overload='_RecordFunction')>)}