tf.estimator.experimental.call_logit_fn