/custom-op-bug

This is the reproducing repo for issue: https://github.com/tensorflow/tensorflow/issues/31568