o )iM@s`ddlZddlZddlZddlZedddZd ddZeddd Zed d d Z dS)NcCstjdddS)N torchvisionZIMPLZMeta)torchlibraryLibraryrrn/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/_meta_registrations.py get_meta_lib srdefaultcsfdd}|S)Ncs,tjrttttjj||S)N)r extensionZ_has_opsrimplgetattrrops)fnop_name overload_namerrwrappers zregister_meta..wrapperr)rrrrrr register_metasrZ roi_alignc sdtddkddtjjkfddd}\}} } } || ||fS)NcSsdS)Nz$rois must have shape as Tensor[K, 5]rrrrrsz meta_roi_align..cdjdjS)NzMExpected tensor for input to have the same type as tensor for rois; but type  does not equal dtyperinputroisrrr r)r_checksizer new_empty) rr spatial_scale pooled_height pooled_widthsampling_ratioalignedZnum_rois_channelsheightwidthrrrmeta_roi_aligns   r+Z_roi_align_backwardc s0tjjkfdd||||fS)Ncr)NzLExpected tensor for grad to have the same type as tensor for rois; but type rrrgradrrrr-rz)meta_roi_align_backward..)rrrr!) r-rr"r#r$Z batch_sizer(r)r*r%r&rr,rmeta_roi_align_backward's   r.)r ) functoolsrZ torch.libraryZtorchvision.extensionr lru_cacherrr+r.rrrrs