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