Is there high performance implementation tfa.image.median_filter2d on GPU

Special Interest Groups Addons
#1

Hi, we found the addon operation tfa.image.median_filter2d bad-perfomance on GPU. Is there suggestion to improve that on GPU? Thanks.