https://github.com/tensorflow/graphics/issues/4047