U <ºcÍ:ã@s°dZddlZddlZddlZddlmZmZmZmZ ddl mZmZej ejddZdZeD]Zede›ƒe e¡ƒq\d d „Zededd dƒgdededddƒgdededddƒgdededd dƒgdededddƒgdededddƒgddd„ƒƒƒƒƒƒZedƒejd œd!d"„ƒZejd œd#d$„Zejd œd%d&„Zejd œd'd(„Zed)ƒejd œd*d+„ƒZed,ƒejd œd-d.„ƒZed/ƒejd œd0d1„ƒZed2ƒejd œd3d4„ƒZed5ƒejd œd6d7„ƒZed8ƒejd œd9d:„ƒZed;ƒe dd?„ƒƒZ!ed@ƒejd œdAdB„ƒZ"eje#dCœdDdE„Z$edFƒe d/tmp/pip-unpacked-wheel-gikjz4vx/torch/onnx/symbolic_opset8.pyÚ_applyEsz_apply_params.._applyr)rrrrrrÚ _apply_paramsBsrzaten::upsample_nearest1dZupsample_nearest1déZnearest)Zdecoratezaten::upsample_nearest2dZupsample_nearest2dézaten::upsample_nearest3dZupsample_nearest3dézaten::upsample_linear1dZupsample_linear1dZlinearzaten::upsample_bilinear2dZupsample_bilinear2dzaten::upsample_trilinear3dZupsample_trilinear3dcs‡‡‡fdd„}|S)Ncst |ˆ|¡\}}t ˆ¡t |¡}|r8t ˆdˆ¡St ˆd¡‰t ˆ¡rZt ˆd¡S|dkr~‡‡‡fdd„tdˆƒDƒ}|jdˆˆ|dS) Núalign_corners == TrueÚisz%torch._C.Value (output_size) indexingcsDg|]<}|dkrdn*tˆˆ|ƒtˆ ¡ ¡ˆ|ƒ‘qS)éçð?)ÚfloatÚtypeÚsizes)Ú.0Úi)ÚdimÚinputÚoutput_sizerrÚ rsýÿÿz5_interpolate..symbolic_fn..rÚUpsample©Zmode_sZscales_f) rZ_get_interpolate_attributesZ_interpolate_warningÚ_maybe_get_scalarÚ_unimplementedÚ_maybe_get_constÚ _is_valueÚrangeÚop)Úgr%r&rÚscalesÚ align_corners©r$Úinterpolate_modeÚname)r%r&rÚsymbolic_fnds(ÿ ÿûz!_interpolate..symbolic_fnr)r5r$r4r6rr3rÚ_interpolateKsr7zaten::__interpolate©r0c Cst |d¡}t |¡s&|r&t dd¡St |¡sFt |¡rFt dd¡St |¡sft |¡rft dd¡St ||||||¡\}}|jd|||dS)NÚbZinterpolaterzdynamic scales in opset 8zdynamic size in opset 8r(r))rr,Z_is_noner+r-Z _interpolate_get_scales_and_moder/) r0r%ÚsizeZscale_factorÚmoder2Zrecompute_scale_factorZ antialiasr1rrrÚ __interpolate~s*ÿÿÿr<cshdddh}d}|d ¡ ¡}|dk rT|}||krJt‡fdd„|Dƒƒ}q^d|Sn t d¡|f|S) NZHalfZFloatÚDoublerc3s|]}t ˆ|d¡VqdS)FN)Úopset9Z_cast_Float)r"Úargr8rrÚ sz-_try_cast_integer_to_float..rzºOnly 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 Z scalarTypeÚtupleÚwarningsÚwarn)r0rZfloating_scalar_typesÚold_typeZ arg0_typerr8rÚ_try_cast_integer_to_float s ÿrEcCs$|dkr|Sttd|›ƒ||dƒS)NZ_cast_F)Úgetattrr>)r0r%Zto_typerrrÚ _cast_to_type¹srGcCs6t |¡}t ||¡}t|||ƒ\}}}| |||¡Sr)rr*Z_if_scalar_type_asrEr/)r0r%ÚotherZop_nameÚ_rrrÚ_comparison_operator¿s rJzaten::gtcCst|||dƒS)NZGreater©rJ©r0r%rHrrrÚgtÈsrMzaten::ltcCst|||dƒS)NZLessrKrLrrrÚltÍsrNz aten::bmmcCsDt |¡r2t|||ƒ\}}}t|| d||¡|ƒS| d||¡SdS)NZMatMul)rÚ_try_get_scalar_typerErGr/)r0ÚselfrHrDrrrÚbmmÒs rQzaten::matmulcCst|||ƒSr)rQ)r0rPrHrrrÚmatmulÛsrRzaten::prelucCsªt |¡}t |¡}|dk rD|dkrD|jd|ttd|dƒƒd}n"|dkrf|dgkrft ||dg¡}t |¡r˜t|||ƒ\}}}t || d||¡|ƒS| d||¡SdS)NrZ Unsqueezeé)Zaxes_irZPRelu) rZ_get_tensor_rankZ_get_tensor_sizesr/Úlistr.Z_squeeze_helperrOrErG)r0rPZweightZ self_rankZweight_sizesrDrrrÚpreluàs rUzaten::mmc Csšt ||¡}|dkr t d|¡‚|jdtjdgtj |¡ ¡dd}t |¡r„t||||ƒ\}}}}t||jd|||ddd |ƒS|jd|||ddd S) Nz/mm can only operate on tensors with known typesÚConstantr)Údtype©Zvalue_tÚGemmgr©Zbeta_fZalpha_f) rrOrZSymbolicValueErrorr/ÚtorchÚtensorrÚ JitScalarTypeÚ from_namerWrErG)r0rPrHÚscalar_typeZ zero_constantrDrrrÚmmðs4ÿÿþ ÿýr`zaten::addmmÚvÚtcCspt |¡rJt||||ƒ\}}}}t||jd|||t |¡t |¡d|ƒS|jd|||t |¡t |¡dSdS)NrYrZ)rrOrErGr/Z_scalar)r0rPZmat1Zmat2ÚbetaÚalpharDrrrÚaddmms, úö úrez aten::flattencCsøt |dd¡}t |dd¡}| ¡ ¡}|dkr8||}|dkrŒ||dkrŒt |¡r|t||ƒ\}}t||jd||d|ƒS|jd||dS|dkrè||dkrèt |¡rÔt||ƒ\}}t||jd||dd|ƒS|jd||ddSt ||||¡S) Nr#Ú start_dimÚend_dimrrSZFlatten)Zaxis_ir) rÚ _get_constr r$rOrErGr/r>Úflatten)r0r%rfrgZstart_dim_iZ end_dim_ir$rDrrrri(s0 ÿ ÿri)r0rWcCsr|dkrtjj}n t |¡}| ¡jsV|jd|tjj ¡d|d}|jd|| ¡dS|jd|| ¡d|dSdS)NZConstantFillrS)Zdtype_iZinput_as_shape_iZvalue_fZCast)Zto_i)rr]ÚFLOATrWZis_floating_pointr/Z onnx_type)r0r!rWÚconst_valuer_ÚresultrrrÚ_constant_fillEs& ûûrmzaten::emptyr#FcCst||||||ƒSr)Úzeros)r0r!rWÚlayoutÚdeviceÚ pin_memoryÚ memory_formatrrrÚempty]srszaten::empty_likecCst||||||ƒSr)Ú zeros_like)r0r%rWrorprqrrrrrÚ empty_likeksruzaten::zeroscCst|||dƒS)Nr©rm©r0r!rWrorprqrrrrnysrnzaten::zeros_likecCs| d|¡}t|||dƒS)NÚShaper©r/rm©r0r%rWrorprqrrÚshaperrrrt€srtz aten::onescCst|||dƒS)NrSrvrwrrrÚonessr|zaten::ones_likecCs| d|¡}t|||dƒS)NrxrSryrzrrrÚ ones_like•sr}z aten::fullc Csft |d¡}t |¡rFt|||||ƒ}t ||||jdt d¡d¡St |dd¡}t ||||ƒSdS)NrbrVrSrXr#rW)rr,r-rnr>Úaddr/r[r\rhrm) r0r!ÚvaluerWrorprqrkÚtmprrrÚfull¤s rzaten::full_likeÚfc Cs| d|¡}t||||ƒS)Nrxry) r0r%Z fill_valuerWrorprqrrr{rrrÚ full_like±srƒzaten::repeatc Csªt |¡s|jdt |¡d}t |¡r8tt |¡ƒ}nt |d¡}t|ƒ}| ¡rœ| ¡ ¡}|t|ƒ}|dkrœt |||jdt dg||¡d¡}| d||¡S)NrVrXrrrSZTile)rr-r/r[Z LongTensorZ_is_packed_listÚlenZ_unpack_listr,ZisCompleteTensorr r!r>Úviewr\)r0rPZrepeatsZrepeat_size_lenZ const_repeatsr!Z diff_dimsrrrÚrepeatÁs ÿr†)FN)FN)F)FN)F)FN)F)FN).Ú__doc__Ú functoolsrBr[Z torch.onnxrrrrr>Ztorch.onnx._internalrrÚpartialZ onnx_symbolicZ_onnx_symbolicZblock_listed_operatorsZblock_listed_opZ_block_list_in_opsetrr7ZGraphContextr<rErGrJrMrNrQrRrUr`Ú parse_argsreriÚintrmrsrurnrtr|r}rrƒr†rrrrÚsÚ ÿ þþþþþþÿ! ùÿùÿùÿ ùÿ ÿÿøÿ