HomeSort by relevance Sort by last modified time
    Searched refs:CuDNNLSTM (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/python/keras/layers/
cudnn_recurrent_test.py 41 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
57 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
73 ('cudnnlstm', keras.layers.CuDNNLSTM),
81 num_states = 2 if layer_class is keras.layers.CuDNNLSTM else 1
98 ('cudnnlstm', keras.layers.CuDNNLSTM),
124 ('cudnnlstm', keras.layers.CuDNNLSTM),
132 num_states = 2 if layer_class is keras.layers.CuDNNLSTM else
    [all...]
cudnn_recurrent.py 38 """Private base class for CuDNNGRU and CuDNNLSTM layers.
338 @keras_export(v1=['keras.layers.CuDNNLSTM'])
339 class CuDNNLSTM(_CuDNNRNN):
401 super(CuDNNLSTM, self).__init__(
427 super(CuDNNLSTM, self).build(input_shape)
534 base_config = super(CuDNNLSTM, self).get_config()
__init__.py 166 from tensorflow.python.keras.layers.cudnn_recurrent import CuDNNLSTM
lstm_v2_test.py 809 cudnn_lstm_layer = keras.layers.CuDNNLSTM(rnn_state_size)
    [all...]
  /external/tensorflow/tensorflow/python/keras/saving/
hdf5_format_test.py 210 ('cudnnlstm', keras.layers.CuDNNLSTM, {
    [all...]

Completed in 237 milliseconds