tensorflow / 2.3 / train / features