experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix typo in LINE#500 (#9626)

Browse Source 
cell_input_fn -> probability_fn
This commit is contained in:
weipingpku 2017-05-04 00:48:21 +08:00 committed by Vijay Vasudevan
parent 3d3e30894d
commit a1f5abe8e1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/contrib/seq2seq/python/ops/attention_wrapper.py
View File

@ -497,7 +497,7 @@ class AttentionWrapper(core_rnn_cell.RNNCell):
if probability_fn is None: if probability_fn is None:
probability_fn = nn_ops.softmax probability_fn = nn_ops.softmax
else: else:
if not callable(cell_input_fn): if not callable(probability_fn):
raise TypeError( raise TypeError(
"probability_fn must be callable, saw type: %s" "probability_fn must be callable, saw type: %s"
% type(probability_fn).__name__) % type(probability_fn).__name__)