I am new to Tensorflow, I was going through the code present on https://www.tensorflow.org/tutorials/distribute/parameter_server_training.In sequential model we pass the feature