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

kuangll02 April 24, 2023, 1:56am 1

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

Topic		Replies	Views
Model Maker TF Lite Slow Inference General Discussion models , tflite , model_maker , help_request	11	5447	September 6, 2021
Lambda layer for smooth image General Discussion help_request	3	1227	February 22, 2022
OOM while training LLM General Discussion tfkeras , gpu , llm	1	143	April 24, 2024
Can you change WebGPU maxTextureDimension to improve inference performance? General Discussion tfjs , gpu , help_request	4	110	June 17, 2024
Distributed inference with JAX: GPU/TPU interconnect TensorFlow distributed-training , gpu , tf-probability	0	32	August 23, 2024