kh< dZddlmZddlmZddlmZGddej ZeZGdd ej Z y ) a  Depthwise Separable Conv Modules Basic DWS convs. Other variations of DWS exist with batch norm or activations between the DW and PW convs such as the Depthwise modules in MobileNetV2 / EfficientNet and Xception. Hacked together by / Copyright 2020 Ross Wightman )nn) create_conv2d)get_norm_act_layerc eZdZdZdddddddej ej ddf fd Zed Z ed Z d Z xZ S) SeparableConvNormActz4 Separable Conv w/ trailing Norm and Activation rF?TNc  tt| t|t ||z||||d|_tt ||z|| |||_t| | }| t| ni}||fd| i||_ y)NTstridedilationpadding depthwiserbias) drop_layer apply_act) superr__init__rintconv_dwconv_pwrdictbn)self in_channels out_channels kernel_sizerrrrchannel_multiplierpw_kernel_size norm_layer act_layerrrnorm_act_layer norm_kwargs __class__s V/var/www/teggl/fontify/venv/lib/python3.12/site-packages/timm/layers/separable_conv.pyrzSeparableConvNormAct.__init__s "D24$ [+==> HgO %  00 1<Y`gkm ,J B5?5Kdj1QS  RRkRc.|jjSNrrrs r(rz SeparableConvNormAct.in_channels!||'''r)c.|jjSr+rrr-s r(rz!SeparableConvNormAct.out_channels%||(((r)cl|j|}|j|}|j|}|Sr+)rrrrxs r(forwardzSeparableConvNormAct.forward)s. LLO LLO GGAJr)) __name__ __module__ __qualname____doc__r BatchNorm2dReLUrpropertyrrr5 __classcell__r's@r(rrsb>?TU_ahm$'bnn`b`g`gDS (())r)rcNeZdZdZ dfd ZedZedZdZxZ S)SeparableConv2dz Separable Conv c tt| t|t ||z||||d|_tt ||z|| |||_y)NTr r)rr@rrrrr) rrrr rrrrr!r"r's r(rzSeparableConv2d.__init__6s_ ot-/$ [+==> HgO %  00 1<Y`gkm r)c.|jjSr+r,r-s r(rzSeparableConv2d.in_channelsAr.r)c.|jjSr+r0r-s r(rzSeparableConv2d.out_channelsEr1r)cJ|j|}|j|}|Sr+)rrr3s r(r5zSeparableConv2d.forwardIs! LLO LLOr))r rrr Fr r) r6r7r8r9rr<rrr5r=r>s@r(r@r@3sAhm89 m(())r)r@N) r9torchrrcreate_norm_actrModulerSeparableConvBnActr@r)r(rJs>(/299D*biir)