o toiZ@@sfddlmZddlZddlmZddlmZmZmZddl Z ddl m Z m Z ddlmZddlmZddlmZmZerBdd lmZeZeZeed d k ZdEddZdFddZdGddZdHddZdId!d"Z dJd$d%Z!dKd&d'Z"dKd(d)Z#dLd,d-Z$dMd0d1Z%dNd4d5Z&dOd8d9Z'Gd:d;d;eZ(dPd>d?Z)e*d@krere)e(dAdBe)e(dCdDdSdS)Q) annotationsN defaultdict) TYPE_CHECKINGAny NamedTuple)defshelper)collect_sample_implementations)collect_snippets)ONNX_ML_DOMAINOpSchema)SequenceONNX_ML0vintreturnstrcCstj|rdSt|S)Nz∞)rr is_infiniter)rrE/home/ubuntu/.local/lib/python3.10/site-packages/onnx/defs/gen_doc.pydisplay_numbers rdomainoutputboolcCsd|v}|tkr |S| S)Nz-ml)r )rris_mlrrrshould_render_domain!sr schema_namecCs|r |d|S|S)N.r)rrrrrformat_name_with_domain(sr function_versions Sequence[int]cCsddd|DS)N, cSsg|]}t|qSr)r.0rrrr /z,format_function_versions..join)r!rrrformat_function_versions.sr*versionsSequence[OpSchema] changelogcs$dfdd|dddDS)Nr#c3(|]}tt|j|j|jVqdSNdisplay_version_linkr rname since_versionr$r-rr 3s&z"format_versions..r()r+r-rr4rformat_versions2s$r7OpSchema.AttrTypecCsJt|tjsJt|}||ddd}|ddkr#d|}|S)Nrr6szlist of ) isinstancer AttrTyperrfindlower)rr:rrrdisplay_attr_type6s  r?cCs|rd|dSdS)Nzthe 'z' operator setzthe default ONNX operator setrrrrrdisplay_domain?s rAcCs |r|SdS)Nzai.onnx (default)rr@rrrdisplay_domain_shortEsrBr2versioncCs&|d|}d|d|d|dS)N-z zr)r2rCr- name_with_verrrrr1Ksr1formal_parameterOpSchema.FormalParametercCsg}tjj|jkr dg}ntjj|jkr|jrdg}nddg}tjj}tjj}||j kr2| dn ||j kr<| dt |dkrDdSdd |d S) Noptionalvariadic heterogeneousdifferentiableznon-differentiabler (r#)) r FormalParameterOptionOptionaloptionVariadicis_homogeneousDifferentiationCategoryDifferentiableNonDifferentiabledifferentiation_categoryappendlenr))rGtagsrLnon_differentiablerrrgenerate_formal_parameter_tagsPs     "r]schemar c sd}|jr|d7}|ddd|jD7}|d7}|d7}|jtjjkr.|d7}n7|d|jr5dnd d |j 7}|d t |j d 7}t |d kre|d dfdd|ddD7}|jrj|S|jr|d7}|d7}t|jD][\}}d}|jrd}n-|jjrt|j}|jj}d.ddt|trfdd|D}n|}d||}|d|jdt|j|rd |d!ndd"7}|d#|jd$7}q||d%7}|d&7}|j|jkr|d t|jd't|jd!7}|d(7}|jr+|d7}|jD] } t | } |d| jd)| d*| j!d"7}|d#| jd$7}q|d%7}|d+7}|j"|j#krG|d t|j"d't|j#d!7}|d(7}|j$r{|d7}|j$D] } t | } |d| jd)| d*| j!d"7}|d#| jd$7}qV|d%7}|d,7}|d(7}|j%r|d7}|j%D]6} | j&} t | d-kr| d-}| d dD] }|d|7}q|d| j'd|d"7}|d#| jd$7}q|d%7}|S)/NrM css|] }d|VqdS)z N)rstrip)r%linerrrr5os  z!display_schema..z #### Version z/ No versioning maintained for experimental ops.z' This version of the operator has been deprecated availablez since version  of z. r9z% Other versions of this operator: {} r#c3r.r/r0r$r4rrr5s  r6z #### Attributes z
requiredvaluerrrcSs`t|trtt|d}t|dkrtd|dd}|St|ttfr,t|dSt|S)N (erOutf-8) r;floatrnproundrZbytes bytearraydecode)rf formattedrrr format_values  z$display_schema..format_valuecsg|]}|qSrr)r%val)rsrrr&r'z"display_schema..z default is z
z : rNrOz
z
z
z
z #### Inputsz -  zz : z #### Outputsz #### Type Constraintsr)rfrrr)(docr)lstrip splitlines support_levelr SupportType EXPERIMENTALrbr3rArrZformat attributessorteditemsre default_valuer2r get_attribute_value doc_stringr;listr?type description min_input max_inputrinputsr]type_str min_output max_outputoutputstype_constraintsallowed_type_strstype_param_str)r^r+r-r:_attroptrrinput_ option_strrtype_constraint allowedTypesallowedTypeStr allowedTyper)r-rsrdisplay_schemags            2 "  "   rlevelOpSchema.SupportTypecCs|tjjkrdSdS)Nzexperimental rM)r rzr{)rrrrsupport_level_strsrc@seZdZUded<ded<dS)Argsrrr-N)__name__ __module__ __qualname____annotations__rrrrrs  rargsNonec Cstjtjtjtjt}tj|d}ttj||jdddd}|d|d|dt d d }t D] }||j |j |qA|d t|D]f\}}t||jseqZd t|d }t|D]G\} } |d | dt|d 7}t| dd dD]-}t||jd|j } |d|jrdndd| | 7}|t||g|j7}|d 7}qqs||qZWdn1swYttj||jdddd}|d|d|dt dd } t D]}| |j t|j|j|q|d g} t}t| D]Z\}}t||js)qg}t|D]<\}}g}t|D]'\}}t|dd d}|d}|j|vrUq=||j||||fq=|||fq1| ||fq| D]\}}dt|d }|||d|dg}|D]E\}}|D]=\}}}|js|jr|j }|||||fqdt!|jt||t|||jrdndt"||j}||qq|r |d|D]$\}}}}dt!|jt||t||t"||jt#|}||q|d qy|d | D]\}}dt|d }|||D]\}}|D]\}}}d |jr@dnddt!|jt||t||$t||}|t|||j7}|d!7}|t%vr|d"7}tt%|D]!\}}|d#7}|d$|d%7}|d&|d'7}|d(7}|d!7}qu|$t&vr|d)7}|d#7}|d$|d%7}|d&t&|$d'7}|d(7}|d!7}||q4q.qWddS1swYdS)*NdocswrMrk)newlineencodingz. z## Operator Changelog a*This file is automatically generated from the [def files](/onnx/defs) via [this script](/onnx/defs/gen_doc.py). Do not modify directly and instead edit operator definitions.* For an operator input/output's differentiability, it can be differentiable, non-differentiable, or undefined. If a variable's differentiability is not specified, that variable has undefined differentiability. cSttSr/rrrrrrzmain..r_z# z ## Version rdcS|jSr/)r2r:rrrr)keyrDz### **{}**z (deprecated)z z## Operator Schemas cSs tddS)NcSrr/rrrrrr3rz(main....rrrrrr3s cSrr/)r3rrrrrHrr6z### z"|**Operator**|**Since version**|| z|-|-|-| z|{}{}{}|{}| z6|**Function**|**Since version**|**Function version**| z|{}{}|{}|{}| z## z*### {}**{}**ruz#### Examples z
z z z ```python z ``` z
z#### Sample Implementation )'ospathdirnamerealpath__file__r)openr-writerrget_all_schemas_with_historyrr3rYr~rrrrBrAr r2rbr|rrrysetadd has_functionhas_context_dependent_functionall_function_opset_versionsrr7r*r>SNIPPETSSAMPLE_IMPLEMENTATIONS)rbase_dirdocs_dirfoutdv_indexr^r versionmapr:rCunsorted_schemasrFindexoperator_schemas existing_ops _supportmapprocessed_supportmap_support_namemapprocessed_namemapnunsorted_versionsr+ supportmap function_opsrnamemapr!op_typesummarycoderrrmains&         ,    "                   $r__main__zOperators-ml.mdzChangelog-ml.mdz Operators.mdz Changelog.md)rrrr)rrrrrr)rrrrrr)r!r"rr)r+r,r-rrr)rr8rr)rrrr)r2rrCrr-rrr)rGrHrr)r^r r+r,r-rrr)rrrr)rrrr)+ __future__rr collectionsrtypingrrrnumpyrmonnxrr onnx.backend.sample.opsr onnx.backend.test.caser onnx.defsr r collections.abcrrrrgetenvrrrr r*r7r?rArBr1r]rrrrrrrrrsX                ~ 2