r2.11 cherry-pick: 239139d2ae6 "Fixing null pointer read in TensorArrayConcat when the step container is missing."
Refer to the original commit: https://github.com/tensorflow/tensorflow/commit/239139d2ae6a81ae9ba499ad78b56d9b2931538a