Searched refs:_NUM_UNITS (Results 1 – 1 of 1) sorted by relevance
49 _NUM_UNITS = 7 variable in FunctionalRnnTest55 'gru': (rnn_cell_impl.GRUCell, [_NUM_UNITS]),56 'lstm': (rnn_cell_impl.LSTMCell, [_NUM_UNITS]),57 'stacked_lstm': (_CreateStackedLstmCell, [_NUM_UNITS] * 3)