2jv:BddlmZddlZddlmZGddeZdS)) annotationsN) CommonPoolc<eZdZ ddZdZdZdZdZdS)MaxPoolNc N|3t|t|ksHt|dks5|Pt|t|kst|dkr|||||||||Stj|dd|||||||| S)N)auto_pad ceil_mode dilations kernel_shapepads storage_orderstridesMAXr)minmax _max_poolr_run) selfxr r r r r rrs OC:\Projects\Historical\venv\Lib\site-packages\onnx/reference/ops/op_max_pool.pyrz MaxPool._run s  !Y3y>>11S^^q5H5H  S\\S\\%A%AS\\UVEVEV>>!##)+"      %'    c R)dtt|dzD|&dtt|D}|&dtt|D}t|tjfdtD} |jdd} d| D} |rtt| D]} t tj| | | | z|| dz || zdzz || z dz| | <| | dz || z| | | | dzk} | r| | xxdzcc<ntt| D]k} t tj| | | | z|| dz || zdzz || z dz| | <l|rV|d krO|d vrtt| D]} |d kr4t tj| | || z | | <n3t tj| | || z | | <| | dz || z|| dz || zdzz| | z }|dz| | df<|| | dfz | | df<nmtt| D]P} t tj| | || dz || zdzz dz|| z | | <Qt| dkr| |||||| ||| St| dkr| |||||| ||| St| d kr| |||||| ||| Std |jd)Ncg|]}dSr.0is r z%MaxPool._max_pool..Bs<<.Ds;;;Qq;;;rcg|]}dSr#rrs rr z%MaxPool._max_pool..Fs===q===rc6g|]}||zfSrr)rrn_dimsr s rr z%MaxPool._max_pool..Is+PPPQd1gtAJ'78PPPrcg|]}dSrr)rss rr z%MaxPool._max_pool..Ls???a???rrrNOTSET) SAME_UPPER SAME_LOWERr*zNot implemented yet for shape .) rangelennparrayshapeintceilsumfloor _max_pool_1d _max_pool_2d _max_pool_3d RuntimeError)rrr r r r r rrnew_padsinput_spatial_shapeoutput_spatial_shaper$need_to_reduce_out_size_in_ceil_modepad_ir&s ` @rrzMaxPool._max_pool6s9 <<???  323344 1 1*-G/2&qkoo//0 ,Q! 3y|CaGI"!* %   + +$Q')+a/AJ8"5a"88A;q>"I8J481(+++q0+++! 1$323344  *-H/2&qkoo//0 ,Q! 3y|CaGI"!* %   + +$Q''  H,,777s#67788<.iterationsf*Cf*Cf*CM** 8 8h$7 Q* <<vtZ88$$A1uuV  zVC!G_r%9%9#C!G_"##%sRx #$v:#7sRx   8 8r) r2tupler0zerosrBfullint64ravelr.r/outputreshape)rrr r r r r;rrr=global_poolingy_dimsyindicesx_dimschannelstotal_channelsr\rIrSrTrUrVrWrXrYrZr[s `` @@@@@@@@@rr7zMaxPool._max_pool_1ds!u%9::: HV17 + + +'&R@@@!9q X-&611GAJq\  8 8 8 8 8 8 8 8 8 8 8 8 8 8 8$~&&  A IaLLLL t{  q NN6**, ,v&&v(>(>??rc d} |jddt| z} tj| |j} tj| tjd} |j}|d}|dtdkr|dnd| dtdkr| dnd|d|z}| rdn|d| rdn|dzz|d|d|| | fd }t|D] }||t|j dkr | fS |  | fS) NFr!rArCrDrr,rc^|z}|z}tD]}|zdz }|dzz}tD]}|zdz }|dzz}|z|z} d} d} d} t||D]p} | dks| krt||D]O}|dks|kr| z|z}|dks|jdkr/| ||z| kr||z} | } |} Pq| | || z<dkr|z| zz| zn |z| z| zz|| z<dS)NrGrrrrrC)r.r2) rIrJrKrMrNrOpwwstartwend pool_indexrPrQw_indexrRw input_indexrSrTrUrV dilation_wrWr r;rX pooled_widthrrYstride_wwidthrZr[s rr\z'MaxPool._max_pool_2d..iterationsf*Cf*CM**  h$7 Q* << --B(]Xd^;F!LOj$@@D!#l!2R!7JB G G"64<< , ,q55AKK$!&vtZ!@!@ , ,A 1uuU (*+e)a-K*Q+ Q2O2O (!zVC+4E-F-K-K%+C+,=%>*+*+ ,z /1F3+,)A--F Wu_4w>>Z'1Gf4DD3+,,/  r r2r]r0r^rBr_r`r/rar.rbrc) rrr r r r r;rrr=rdrerfrgrhrirjr\rIrSrTrUrVrurWrXrvrYrwrxrZr[s ``` @@@@@@@@@@@@@rr8zMaxPool._max_pool_2ds!u%9::: HV17 + + +'&R@@@!9 ..22q q $' $5$5$9$9vayyq X-&611GAJ&611GAJ%-q\ q\ ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! F~&&  A IaLLLL t{  q NN6**, ,v&&v(>(>??rc J !"#d} |jddt| z} tj| |j} tj| tjd} |j}|d}|dtdkr|dnd!tdkr|dnd| dtdkr| dndtdkr| dnd|d |z}| rdn|d | rdn|d | rdn|d!zz"zz#|d |d|d|| |  !"#fd }t|D] }||t|j dkr | fS |  | fS) NFr!rArCrDrr,rc (|(z}|)z}|)z}t!D]p}|%zdz }|dzz}t"D]@}|&zdz }|dzz} t D]} | $zdz } | dzz} |"z z| zz| z} d}d}d}d}t||D]}|dks|krt|| D]c}|dks|'krt| | D]B}|dks|kr|'zz|zz|z}|||z|kr||z}|}|}|}Cd||| z<#dkr|(z|'zzz|zz|zn|(z|z|zz|z'zz|| z<BrdS)NrGrrmr)r!rr!rCrH)*rIrJrKrLrMrNrOrnrorppddstartdendrqrPrQrrd_indexrRrsdrtrSrTrUdepth dilation_drVrurWr r; pooled_depthrXrvrstride_drYrwrxrZr[s* rr\z'MaxPool._max_pool_3d..iterationbsf*Cf*Cf*CM**/ / h$7 Q* << --,,B(]Xd^;F!LOj$@@D#L11))!#h$!?% Q*(DD- (>??r)NNNNNNN)__name__ __module__ __qualname__rrr7r8r9rrrrr s ( ( ( ( T~H~H~H@6@6@6@pK@K@K@Zb@b@b@b@b@rr) __future__rnumpyr0"onnx.reference.ops._op_common_poolrrrrrrs#"""""999999R@R@R@R@R@jR@R@R@R@R@r