tf.distribute.InputContext

A class wrapping information needed by an input function.