Ύj"zddlZddlmZejejrdndZiZdZdS)Ncudacpuc &t|jt|f}|tvr!t jdd|jdtddd|jd|jdd|jdd}t jdd|jdtdd|jdd|jddd|jd}t j ||gd tt|<t j |ddddddddf|jddz d z z |ddddddddf|jddz d z z gd}t||z dddd}tj j ||d d d S)Ngg?)devicerg@bilinearborderT)inputgridmode padding_mode align_corners)strrsizebackwarp_tenGridtorchlinspaceshapeviewexpandcattopermutenn functional grid_sample)tenInputtenFlowk tenHorizontal tenVerticalgs g/home/vela/workspace/domain/tn-silver/web/tn-site/.runtime/video-rife/Practical-RIFE/model/warplayer.pywarpr's W^  c',,..112A   tS'-2B6RRRWW q!W]1%'''-vgmA.>GMRSDTVX'Y'Y nT3 a0@PPPUU q'-"A'''-vgmA.>B VWHX'Y'Y #i K (!---/RZZ iAaCAAA.8>!3Ds3Jc2QR AaCAAA.8>!3Ds3Jc2QRTUVXXG ! w &//1a;;A 8  * * aiy} * ~ ~~)rtorch.nnrrr is_availablerr'r(r&r,sl   7 7 9 9Dffu E Er(