提交 28c54a0f 编写于 作者: J Jianwei Xie 提交者: TensorFlower Gardener

Move callers from tf.nn.state_saving_rnn to tf.contrib.rnn.static_state_saving_rnn.

Change: 141559664
上级 4fc106f5
......@@ -684,7 +684,7 @@ class SequenceQueueingStateSaver(object):
inputs_by_time = tf.split(value=inputs, num_or_size_splits=num_unroll, axis=1)
assert len(inputs_by_time) == num_unroll
lstm_output, _ = tf.nn.state_saving_rnn(
lstm_output, _ = tf.contrib.rnn.static_state_saving_rnn(
cell,
inputs_by_time,
state_saver=batch,
......@@ -1302,7 +1302,7 @@ def batch_sequences_with_states(input_key, input_sequences, input_context,
inputs_by_time = tf.split(value=inputs, num_or_size_splits=num_unroll, axis=1)
assert len(inputs_by_time) == num_unroll
lstm_output, _ = tf.nn.state_saving_rnn(
lstm_output, _ = tf.contrib.rnn.static_state_saving_rnn(
cell,
inputs_by_time,
state_saver=batch,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册