FIX: Use element_shape in beam_search_decoder (#318)
FIX: Use element_shape in beam_search_decoder (#318)

Commit 1cd4a2b in TF removed `_element_shape` from TensorArray
tensorflow/addons,tensorflow/addons,tensorflow/addons