trainable_initial_state_rnn
