Home
last modified time | relevance | path

Searched refs:cell_sizes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddynamic_rnn_estimator_test.py311 cell_sizes = [5, 3, 7]
316 [rnn_cell.LSTMCell(i) for i in cell_sizes])
360 cell_sizes = [4, 8, 7]
396 num_units=cell_sizes,
416 cell_sizes = [4, 7]
453 num_units=cell_sizes,
/external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/
Dfunctional_rnn_test.py39 def _CreateStackedLstmCell(*cell_sizes): argument
40 subcells = [rnn_cell_impl.LSTMCell(cell_size) for cell_size in cell_sizes]