Support exporting for AdaNet TPU when mode is not PREDICT.
Support exporting for AdaNet TPU when mode is not PREDICT.

TPU only supports exporting in PREDICT mode. If we're not in PREDICT mode and
are exporting, we assume we are exporting to CPU/GPU.

PiperOrigin-RevId: 282038568

tensorflow/adanet,tensorflow/adanet