Revert bad merge that added uses of contrib back into ncf_main.py.
Revert bad merge that added uses of contrib back into ncf_main.py.

PiperOrigin-RevId: 283365340

tensorflow/tpu,tensorflow/tpu,tensorflow/tpu,tensorflow/tpu