˲j_< dZddlZddlZddlZddlmZddlmZm Z m Z m Z ddl mZmZejejdZdZeD]'Zed ee je(ed e jd d d gede jddd gede jddd gede jdd dgede jdddgede jdddgdZeddejfdZdejfd Zdejfd!Zdejfd"Zed#dejfd$Zed%dejfd&Zed'dejfd(Zed)dejfd*Z ed+dejfd,Z!ed-dejfd.Z"ed/e j#d0d0d0d1d1dejfd2Z$ed3dejfd4Z%dejd5e&fd6Z'ed7e j#d0d8d0d0d0d0 dLdejfd:Z(ed;e j#d0d8d0d0d0d0 dLdejfd<Z)ed=e j#d0d8d0d0d0dMdejfd>Z*ed?e j#d0d8d0d0d0d0 dLdejfd@Z+edAe j#d0d8d0d0d0dMdejfdBZ,edCe j#d0d8d0d0d0d0 dLdejfdDZ-edE dMdejfdFZ.edGe j#d0dHd8d0d0d0d0 dLdejfdIZ/edJdejfdKZ0dS)Na Note [ONNX operators that are added/updated from opset 8 to opset 9] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ New operators: Compress ConstantOfShape EyeLike MaxUnpool OneHot Sinh Cosh Asinh Acosh Atanh Shrink IsNaN Sign Erf Scatter Where NonZero TfIdfVectorizer MeanVarianceNormalization Updated operators: BatchNormalization: removed spatial attribute. Greater, Less, Constant, MatMul, PRelu, Gemm, Flatten: more data types{integers} supported. Cast: more data types{string} supported. Upsample: moved scales from attribute to input. Scan N)_onnx) _type_utilserrorssymbolic_helpersymbolic_opset9) jit_utils registration)opset)nonzerowherescatter scatter_adderfsignisnangatherarange masked_fill index_fill index_copyrepeat_interleaveanyallzaten::zaten::upsample_nearest1dupsample_nearest1dnearest)decoratezaten::upsample_nearest2dupsample_nearest2dzaten::upsample_nearest3dupsample_nearest3dzaten::upsample_linear1dupsample_linear1dlinearzaten::upsample_bilinear2dupsample_bilinear2dzaten::upsample_trilinear3dupsample_trilinear3dcfd}|S)Nctj||\}}tjtj|}|rtjdStjdtjrtjdS|fdtdD}|d|S)Nalign_corners == Trueisz%torch._C.Value (output_size) indexingcg|]a}|dkrdnVt|z t|z z bS)?)floattypesizes).0idiminput output_sizes KC:\Projects\Historical\venv\Lib\site-packages\torch/onnx/symbolic_opset8.py z5_interpolate..symbolic_fn..ls} q55;qz233 **,,sQwZ899:rUpsamplemode_sscales_f) r_get_interpolate_attributes_interpolate_warning_maybe_get_scalar_unimplemented_maybe_get_const _is_valuerangeop) gr4r5argsscales align_cornersr3interpolate_modenames `` r6 symbolic_fnz!_interpolate..symbolic_fn^s / K ! !   ,-=>>>'9-HH  X"1$8OQVWW W%6{DII  $[ 1 1 "1=  > q# FttJ.>tPPPr8)rJr3rIrKs``` r6 _interpolaterMEs72QQQQQQQ. r8zaten::__interpolaterEctj|d}tj|s|rtjddStj|s)tj|rtjddStj|s)tj|rtjddStj||||||\}}|d|||S)Nb interpolater)zdynamic scales in opset 8zdynamic size in opset 8r9r:)rrA_is_noner@rB _interpolate_get_scales_and_moderD) rEr4size scale_factormoderHrecompute_scale_factor antialiasrGs r6 __interpolaterXxs $4]CHHM  #M 2 2V}V-m=TUUU  #L 1 1 o6O77 - 6     #D ) )Xo.G.M.MX-m=VWWW"C 5$ dMLFD 44 E$4 @ @@r8ctjjtjjtjjh}d}tj|dtjj}|tjjkr;|}||vr0|}tfd|D}nd|zStj d|f|zS)Nrc3fK|]+}d|tjjV,dS)Castto_iN)rD_C_onnxTensorProtoDataTypeFLOAT)r1argrEs r6 z-_try_cast_integer_to_float..sOVSw'B'HIIr8NzOnly floating datatype is supported for these operators: {Greater, Less, MatMul, PRelu, Gemm, Flatten}. This might cause the onnx model to be incorrect, if inputs have integer datatypes.) r JitScalarTypeHALFr`DOUBLE from_value UNDEFINED scalar_nametuplewarningswarn)rErFfloating_scalar_typesold_type arg0_types` r6_try_cast_integer_to_floatrps!&!'!( H)44 Q*4IK-777 0 0 0++--HDD T> !  P   ; r8cP||Sttd|||dS)N_cast_F)getattropset9)rEr4to_types r6 _cast_to_typervs3 .76-G-- . .q% ? ??r8ctj|}tj||}t|||\}}}||||Src)rr?_if_scalar_type_asrprD)rEr4otherop_name_s r6_comparison_operatorr|sQ  -e 4 4E  .ue < 8EBBOHe 144 54==x 44 54== =aIq00  / 6 6 @8EBBOHe 144 5Q4?? 44 5Q4?? ? >!UIw 7 77r8rc|tjj}ntj|}|jse|d|tjjd|}|d||S|d||d|S)N ConstantFillr)dtype_iinput_as_shape_ivalue_fr[r\)rrdr`ris_floating_pointrD onnx_type)rEr0r const_valuerresults r6_constant_fillrCs }!/5 !/66      0   -3==??   ttFF)>)>)@)@tAAAtt  ))++    r8z aten::emptyr2Fc*t||||||Src)zeros)rEr0rlayoutdevice pin_memory memory_formats r6emptyr[s E5&&* = ==r8zaten::empty_likec*t||||||Src) zeros_like)rEr4rrrrrs r6 empty_likeris avvz B BBr8z aten::zerosc&t|||dS)NrrrEr0rrrrs r6rrws !UE1 - --r8zaten::zeros_likecR|d|}t|||dS)NShaperrDrrEr4rrrrrshapes r6rr~+ DD% E !UE1 - --r8z aten::onesc&t|||dS)Nrrrs r6onesrs !UE1 - --r8zaten::ones_likecR|d|}t|||dS)Nrrrrs r6 ones_likerrr8z aten::fullc Ftj|d}tj|rQt|||||}t j||||dtjdStj |dd}t||||S)Nrrrrr2r) rrArBrrtaddrDrrrr) rEr0valuerrrrrtmps r6fullrs"25#>>K --<AueVV44z!S%j%,q//)R)RSSS*5#w??a{;;;r8zaten::full_likefcR|d|}t||||S)Nrr) rEr4 fill_valuerrrrrrs r6 full_likers+ DD% E !UE: 6 66r8z aten::repeatc tj|s)|dtj|}tj|r"t tj|}n$tj|d}t |}| r| }|t |z }|dkrDtj |||dtjdg|z|z}|d||S)Nrrr*rrTile)rrBrDr LongTensor_is_packed_listlen _unpack_listrAisCompleteTensorr/r0rtviewr)rErrepeatsrepeat_size_len const_repeatsr0 diff_dimss r6repeatrs  $W - -F$$z5+;G+D+D$EE&w//-o:7CCDD'8$GG m,,  !!###c%jj0 q==;4j%,sYQV?V2W2WXXD 44g & &&r8)FN)F)1__doc__ functoolsrkrtorch._Crr^ torch.onnxrrrrrttorch.onnx._internalrr partial onnx_symbolic_onnx_symbolicblock_listed_operatorsblock_listed_op_block_list_in_opset _apply_paramsrM GraphContextrXrprvr|rrrrrr parse_argsrrintrrrrrrrrrrrLr8r6rs$@ %%%%%%VVVVVVVVVVVV88888888#"<#=QGGG$.O.NN-O--..,,_== +o+,@!YOO P+o+,@!YOO P+o+,@!YOO P+o+,?HMM N+o+,A1hOO P +o+,BAxPP Q      06%&&AAAA'&AB)"8>@Y+@@@@ 'I2'''' <) <<<< 9) 9999 +9 !++++i$   +Y # + + + + M) MMMM2 Cc344 Y #   54 4  8y%888! 88 i, C    0 Cc3S99 > > > > >:9 >"##Cc3S99 C C C C C:9$# C Cc344..Y #...54. "##Cc3S99 . . . . .:9$# . Cc344..I "...54.!""Cc3S99 . . . . .:9#" . OT < < < < < <!""Cc3S#>> 7 7 7 7 7?>#" 7'i$''' '''r8