message: Make raw_rnn accept scalar or TensorArray values for state.
diff: @@ -1125,6 +1125,12 @@ def raw_rnn(cell, loop_fn,
  def _copy_some_through(current, candidate):
  """Copy some tensors through via array_ops.where."""
  def copy_fn(cur_i, cand_i):
  + # TensorArray and scalar get passed through.
  + if isinstance(cur_i, tensor_array_ops.TensorArray):
  + return cand_i
  + if cur_i.shape.ndims == 0:
  + return cand_i
  + # Otherwise propagate the old or the new value.
  with ops.colocate_with(cand_i):
  return array_ops.where(elements_finished, cur_i, cand_i)
  return nest.map_structure(copy_fn, current, candidate)
  