Vh\dZddlmZddlmZddlZddlZddlZddlm Z  ddZ d dZ y) z2Dispatcher for AtenLib functions from onnx-script.) annotations)CallableN) registrationct}|jDcgc]9}|jdr&|jddjdd;}}tj j tj jfD]/}t|}|tj j ur||z }|D]}t||st||}t|tjjsA|jD]}t||} tj j#| j%} |j'| j(| j*| j,s)|j'| j(| j*ds|j/| 2|Scc}w) a1 Creates a set of OperatorBase and Callable objects that represent ONNX-supported PyTorch operations. Args: registry (OnnxRegistry): The ONNX registry for PyTorch. Returns: A collection of OperatorBase and Callable objects representing ONNX-supported PyTorch operations. zaten::z::.r)qualified_name) namespaceop_nameoverloadN)set_all_registered_ops startswithsplittorchopsatenprimsdirhasattrgetattr isinstance_opsOpOverloadPacket overloadsrOpNamefrom_qualified_namenameis_registered_opr r r add) registrytablek onnx_supported_aten_lookup_table op_namespace attr_names attr_nameop_overload_packet overload_name op_overloadinternal_op_names [/home/dcms/DCMS/lib/python3.12/site-packages/torch/onnx/_internal/fx/decomposition_table.py'_create_onnx_supports_op_overload_tabler-s69UE--/( << !  as#A&($( 9+ & 599>> ) : :J# +I<3!(y!A 0%**2M2MN!3!=!=!? + %&8-H #/#6#6#J#J#.#3#3#5$K$ ,,.88,44-66-...88,44!/IIk*' + + +@ LM(s>Gc,i}t|}tjjj D]\}}d|j vs||vr|||<tjj j D]\}}||vr |||<|S)aJ This function creates a dictionary of op overloads and their decomposition functions for ops that do not have ONNX symbolic functions. If an op already has an ONNX symbolic function, its decomposition function is excluded from the table. The decomposition table is a subset of PyTorch's built-in aten-to-aten decomposition. Args: registry (torch.onnx.OnnxRegistry): The ONNX registry for PyTorch. Returns: Dict[torch._ops.OperatorBase, Callable]: A dictionary that maps op overloads to their corresponding decomposition functions. z torch._refs)r-r_decompdecomposition_tableitems __module__core_aten_decompositions)r!r0_ONNX_SUPPORT_OP_OVERLOADSr* decomp_fns r,(create_onnx_friendly_decomposition_tabler6Is DF"I!R #(--"C"C"I"I"K 5 Y Y11 188 +4K( 5#(--"H"H"J"P"P"R5 Y 4 4 +4K(5 )returnz'set[torch._ops.OperatorBase | Callable])r8z'dict[torch._ops.OperatorBase, Callable]) __doc__ __future__rtypingrr torch._opstorch.fxtorch.onnx._internal.fxrr-r6r7r,r@s58" 08,8v+,+r7