OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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