message: Support other dtypes in BeamSearchDecoder initialization
diff: @@ -299,12 +299,13 @@ class BeamSearchDecoder(decoder.Decoder):
  """
  finished, start_inputs = self._finished, self._start_inputs
  + dtype = nest.flatten(self._initial_cell_state)[0].dtype
  log_probs = array_ops.one_hot( # shape(batch_sz, beam_sz)
  array_ops.zeros([self._batch_size], dtype=dtypes.int32),
  depth=self._beam_width,
  - on_value=0.0,
  - off_value=-np.Inf,
  - dtype=nest.flatten(self._initial_cell_state)[0].dtype)
  + on_value=ops.convert_to_tensor(0.0, dtype=dtype),
  + off_value=ops.convert_to_tensor(-np.Inf, dtype=dtype),
  + dtype=dtype)
  initial_state = BeamSearchDecoderState(
  cell_state=self._initial_cell_state,
  