https://github.com/tensorflow/addons/issues/550#issuecomment-537141212 /cc @ewilderj / @seanpmorgan @karmel