C5W4A1 Transformer Architecture with TensorFlow submission error