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.py41 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 1
160 ('cudnnlstm', keras.layers.CuDNNLSTM),
192 ('cudnnlstm', keras.layers.CuDNNLSTM),
249 for layer_class in [keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM]:
[all …]
Dcudnn_recurrent.py339 class CuDNNLSTM(_CuDNNRNN): class
401 super(CuDNNLSTM, self).__init__(
427 super(CuDNNLSTM, self).build(input_shape)
534 base_config = super(CuDNNLSTM, self).get_config()
D__init__.py166 from tensorflow.python.keras.layers.cudnn_recurrent import CuDNNLSTM
Dlstm_v2_test.py809 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.pbtxt124 name: "CuDNNLSTM"
/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format_test.py210 ('cudnnlstm', keras.layers.CuDNNLSTM, {
/external/tensorflow/
DRELEASE.md376 …NNGRU) and [`tf.keras.layers.CuDNNLSTM`](https://www.tensorflow.org/versions/r1.9/api_docs/python/…