Searched refs:RNNKeys (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
D | state_saving_rnn_estimator.py | 88 final_state, name=rnn_common.RNNKeys.FINAL_STATE_KEY) 187 sequence_features[rnn_common.RNNKeys.LABELS_KEY] = labels 209 return '{}_{}'.format(rnn_common.RNNKeys.STATE_PREFIX, 0) 212 '{}_{}_c'.format(rnn_common.RNNKeys.STATE_PREFIX, 0), 213 '{}_{}_h'.format(rnn_common.RNNKeys.STATE_PREFIX, 0), 217 '{}_{}_c'.format(rnn_common.RNNKeys.STATE_PREFIX, i), 218 '{}_{}_h'.format(rnn_common.RNNKeys.STATE_PREFIX, i), 221 '{}_{}'.format(rnn_common.RNNKeys.STATE_PREFIX, i) 307 return '{}_{}'.format(rnn_common.RNNKeys.STATE_PREFIX, i) 490 labels = sequence_features.pop(rnn_common.RNNKeys.LABELS_KEY)
|
D | dynamic_rnn_estimator.py | 52 return '{}_{}'.format(rnn_common.RNNKeys.STATE_PREFIX, i) 378 if rnn_common.RNNKeys.STATE_PREFIX not in k 397 sequence_length_key=rnn_common.RNNKeys.SEQUENCE_LENGTH_KEY,
|
D | rnn_common.py | 42 class RNNKeys(object): class
|
D | dynamic_rnn_estimator_test.py | 481 if k.startswith(rnn_common.RNNKeys.STATE_PREFIX) 499 if k.startswith(rnn_common.RNNKeys.STATE_PREFIX):
|
D | state_saving_rnn_estimator_test.py | 286 rnn_common.RNNKeys.LABELS_KEY:
|