Home
last modified time | relevance | path

Searched refs:layer_class (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent_test.py44 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
47 def test_cudnn_rnn_return_sequence(self, layer_class, return_sequences): argument
53 layer_class,
60 layer_class=[keras.layers.CuDNNGRU, keras.layers.CuDNNLSTM],
63 def test_cudnn_rnn_go_backward(self, layer_class, go_backwards): argument
69 layer_class,
79 def test_return_state(self, layer_class): argument
84 num_states = 2 if layer_class is keras.layers.CuDNNLSTM else 1
87 layer = layer_class(units, return_state=True, stateful=True)
104 def test_time_major_input(self, layer_class): argument
[all …]
Dsimplernn_test.py102 layer_class = keras.layers.SimpleRNN
106 layer = layer_class(
120 layer_class = keras.layers.SimpleRNN
126 model.add(layer_class(units=5, return_sequences=True, unroll=False))
131 layer_class = keras.layers.SimpleRNN
133 l1 = layer_class(units=1, stateful=stateful)
134 l2 = layer_class.from_config(l1.get_config())
145 layer_class = keras.layers.SimpleRNN
146 layer = layer_class(
170 layer_class = keras.layers.SimpleRNN
[all …]
Dgru_test.py139 layer_class = keras.layers.GRU
145 model.add(layer_class(units=5, return_sequences=True, unroll=False))
159 layer_class = keras.layers.GRU
169 layer = layer_class(
233 layer_class = keras.layers.GRU
237 layer = layer_class(
251 layer_class = keras.layers.GRU
253 l1 = layer_class(units=1, stateful=stateful)
254 l2 = layer_class.from_config(l1.get_config())
265 layer_class = keras.layers.GRU
[all …]
Dlstm_test.py128 layer_class = keras.layers.LSTM
132 layer = layer_class(
150 layer_class = keras.layers.LSTM
156 model.add(layer_class(units=5, return_sequences=True, unroll=unroll))
180 layer_class = keras.layers.LSTM
182 l1 = layer_class(units=1, stateful=stateful)
183 l2 = layer_class.from_config(l1.get_config())
342 layer_class = keras.layers.LSTM
349 layer = layer_class(units)
369 layer_class = keras.layers.LSTM
[all …]
Dgru_v2_test.py147 layer_class = rnn.GRU
149 l1 = layer_class(units=1, stateful=stateful)
150 l2 = layer_class.from_config(l1.get_config())
331 layer_class = rnn.GRU
337 model.add(layer_class(units=5, return_sequences=True, unroll=False))
392 layer_class = rnn.GRU
399 outputs, states = layer_class(units=5, return_state=True)(masked)
420 layer_class = rnn.GRU
424 layer = layer_class(
451 layer_class = rnn.GRU
[all …]
Dlstm_v2_test.py133 layer_class = rnn.LSTM
135 l1 = layer_class(units=1, stateful=stateful)
136 l2 = layer_class.from_config(l1.get_config())
302 layer_class = rnn.LSTM
309 layer = layer_class(units)
391 layer_class = rnn.LSTM
395 layer = layer_class(
408 layer_class = rnn.LSTM
414 model.add(layer_class(units=5, return_sequences=True, unroll=False))
616 layer_class = rnn.LSTM
[all …]
Drecurrent_test.py622 layer_class = keras.layers.SimpleRNN
626 layer = layer_class(units)
651 layer_class = keras.layers.SimpleRNN
658 layer = layer_class(units,
665 output = layer_class(units)(input2, initial_state=state)
/external/tensorflow/tensorflow/python/keras/engine/
Dcontrol_flow_test.py114 def test_control_flow_layer(self, layer_class): argument
115 model = testing_utils.get_model_from_layers([layer_class()],
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_weights_test.py202 self, layer_class, layer_args): argument
204 layer = layer_class(**layer_args)