Home
last modified time | relevance | path

Searched refs:CuDNNLSTM (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent_test.py44 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
60 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
76 ('cudnnlstm', keras.layers.CuDNNLSTM),
84 num_states = 2 if layer_class is keras.layers.CuDNNLSTM else 1
101 ('cudnnlstm', keras.layers.CuDNNLSTM),
127 ('cudnnlstm', keras.layers.CuDNNLSTM),
135 num_states = 2 if layer_class is keras.layers.CuDNNLSTM else 1
166 ('cudnnlstm', keras.layers.CuDNNLSTM),
198 ('cudnnlstm', keras.layers.CuDNNLSTM),
255 for layer_class in [keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM]:
[all …]
Dcudnn_recurrent.py342 class CuDNNLSTM(_CuDNNRNN): class
404 super(CuDNNLSTM, self).__init__(
430 super(CuDNNLSTM, self).build(input_shape)
540 base_config = super(CuDNNLSTM, self).get_config()
D__init__.py254 from tensorflow.python.keras.layers.cudnn_recurrent import CuDNNLSTM
Dlstm_v2_test.py1033 cudnn_lstm_layer = keras.layers.CuDNNLSTM(rnn_state_size)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.layers.-cu-d-n-n-l-s-t-m.pbtxt1 path: "tensorflow.keras.layers.CuDNNLSTM"
3 is_instance: "<class \'tensorflow.python.keras.layers.cudnn_recurrent.CuDNNLSTM\'>"
Dtensorflow.keras.layers.pbtxt140 name: "CuDNNLSTM"
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_weights_test.py197 ('cudnnlstm', keras.layers.CuDNNLSTM, {
/external/tensorflow/
DRELEASE.md3049 …NNGRU) and [`tf.keras.layers.CuDNNLSTM`](https://www.tensorflow.org/versions/r1.9/api_docs/python/…