Handle case where there is no context in the estimator params (e.g., during export).
Handle case where there is no context in the estimator params (e.g., during export).

PiperOrigin-RevId: 286296772

tensorflow/mesh,tensorflow/mesh