tf.estimator.FeedFnHook