benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py (2 lines):
	- line 745: inputs = features  # TODO: Should be using feature columns?
	- line 794: )  # For access by logger (TODO: Better way to access it?)