https://github.com/tensorflow/addons/issues/2807
tensorflow/addons#2807