Fix typo in LINE#500 (#9626)

cell_input_fn -> probability_fn
2017-05-04 00:48:21 +08:00 · 2017-05-04 00:48:21 +08:00 · a1f5abe8e1
commit a1f5abe8e1
parent 3d3e30894d
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py
+++ b/tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py
@ -497,7 +497,7 @@ class AttentionWrapper(core_rnn_cell.RNNCell):
    if probability_fn is None:
      probability_fn = nn_ops.softmax
    else:
-      if not callable(cell_input_fn):
+      if not callable(probability_fn):
        raise TypeError(
            "probability_fn must be callable, saw type: %s"
            % type(probability_fn).__name__)