Searched refs:rnn (Results 1 - 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/rnn/
H A D__init__.py17 See @{$python/contrib.rnn} guide.
83 from tensorflow.contrib.rnn.python.ops.core_rnn_cell import EmbeddingWrapper
84 from tensorflow.contrib.rnn.python.ops.core_rnn_cell import InputProjectionWrapper
85 from tensorflow.contrib.rnn.python.ops.core_rnn_cell import OutputProjectionWrapper
87 from tensorflow.contrib.rnn.python.ops.fused_rnn_cell import *
88 from tensorflow.contrib.rnn.python.ops.gru_ops import *
89 from tensorflow.contrib.rnn.python.ops.lstm_ops import *
90 from tensorflow.contrib.rnn.python.ops.rnn import *
91 from tensorflow.contrib.rnn
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dbidirectional_sequence_rnn_test.cc767 BidirectionalRNNOpModel rnn(/*batches=*/2, /*sequence_len=*/16,
770 rnn.SetFwWeights(weights);
771 rnn.SetBwWeights(weights);
772 rnn.SetFwBias(biases);
773 rnn.SetBwBias(biases);
774 rnn.SetFwRecurrentWeights(recurrent_weights);
775 rnn.SetBwRecurrentWeights(recurrent_weights);
777 rnn.ResetHiddenStates();
778 const int input_sequence_size = rnn.input_size() * rnn
[all...]
H A Dunidirectional_sequence_rnn_test.cc207 UnidirectionalRNNOpModel rnn(/*batches=*/2, /*sequence_len=*/16,
209 rnn.SetWeights(
233 rnn.SetBias({0.065691948, -0.69055247, 0.1107955, -0.97084129, -0.23957068,
238 rnn.SetRecurrentWeights({0.1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
255 rnn.ResetHiddenState();
256 const int input_sequence_size = rnn.input_size() * rnn.sequence_len();
259 rnn.SetInput(0, batch_start, batch_end);
260 rnn.SetInput(input_sequence_size, batch_start, batch_end);
262 rnn
[all...]
H A Dbasic_rnn_test.cc190 RNNOpModel rnn(2, 16, 8);
191 rnn.SetWeights(
215 rnn.SetBias({0.065691948, -0.69055247, 0.1107955, -0.97084129, -0.23957068,
220 rnn.SetRecurrentWeights({0.1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
237 rnn.ResetHiddenState();
239 (rnn.input_size() * rnn.num_batches());
242 float* batch_start = rnn_input + i * rnn.input_size();
243 float* batch_end = batch_start + rnn.input_size();
244 rnn
[all...]
H A Dbasic_rnn.cc32 namespace rnn { namespace in namespace:tflite::ops::builtin
122 } // namespace rnn
126 rnn::Prepare, rnn::Eval};
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
H A Dfused_rnn_cell.py24 from tensorflow.python.ops import rnn namespace
104 outputs, state = rnn.dynamic_rnn(
115 else: # non-dynamic rnn
118 outputs, state = rnn.static_rnn(
138 cell = tf.contrib.rnn.BasicRNNCell(10)
139 fw_lstm = tf.contrib.rnn.FusedRNNCellAdaptor(cell, use_dynamic_rnn=True)
140 bw_lstm = tf.contrib.rnn.TimeReversedFusedRNN(fw_lstm)
H A Drnn.py21 from tensorflow.python.ops import rnn namespace
35 Stacks several bidirectional rnn layers. The combined forward and backward
68 of the forward rnn.
70 of the backward rnn.
110 prev_layer, state_fw, state_bw = rnn.static_bidirectional_rnn(
137 Stacks several bidirectional rnn layers. The combined forward and backward
181 of the forward rnn.
183 of the backward rnn.
224 outputs, (state_fw, state_bw) = rnn.bidirectional_dynamic_rnn(
/external/tensorflow/tensorflow/python/keras/_impl/keras/layers/
H A Dwrappers_test.py154 rnn = keras.layers.SimpleRNN
169 rnn(output_dim), merge_mode=mode, input_shape=(timesteps, dim)))
185 rnn = keras.layers.SimpleRNN
195 rnn(output_dim), input_shape=(timesteps, dim)))
204 rnn = keras.layers.SimpleRNN
219 rnn(output_dim, return_sequences=True),
222 model.add(keras.layers.Bidirectional(rnn(output_dim), merge_mode=mode))
229 rnn(output_dim), merge_mode=mode)(inputs)
236 rnn = keras.layers.SimpleRNN
250 rnn(output_di
[all...]
/external/tensorflow/tensorflow/python/ops/
H A Dnn.py129 from tensorflow.python.ops.rnn import *
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
H A Dcudnn_rnn_test.py31 from tensorflow.contrib.rnn.python.ops import rnn as contrib_rnn_lib
45 from tensorflow.python.ops import rnn as rnn_lib
155 def rnn(self): member in class:CudnnTestModel
230 def _CreateCudnnCompatibleCanonicalRNN(rnn, inputs, is_bidi=False, scope=None):
231 mode = rnn.rnn_mode
232 num_units = rnn.num_units
233 num_layers = rnn.num_layers
235 # To reuse cuDNN-trained models, must use cudnn compatible rnn cells.
533 rnn
[all...]
H A Dcudnn_rnn_ops_benchmark.py24 from tensorflow.contrib import rnn as contrib_rnn
26 from tensorflow.contrib.rnn.python.ops import lstm_ops
33 from tensorflow.python.ops import rnn namespace
134 outputs, final_state = rnn.dynamic_rnn(
158 outputs, final_state = rnn.dynamic_rnn(
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
H A Dgrid_rnn_cell.py31 from tensorflow.contrib import rnn namespace
34 class GridRNNCell(rnn.RNNCell):
129 rnn.LSTMCell, num_units=num_units, state_is_tuple=state_is_tuple)
136 if not isinstance(self._cells[0], rnn.RNNCell):
335 cell_fn=lambda n: rnn.BasicRNNCell(num_units=n),
364 cell_fn=lambda n: rnn.BasicRNNCell(num_units=n),
379 return rnn.BasicLSTMCell(num_units=n, forget_bias=forget_bias)
410 return rnn.BasicLSTMCell(num_units=n, forget_bias=forget_bias)
440 return rnn.LSTMCell(
473 return rnn
[all...]
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
H A Dfused_rnn_cell_test.py15 """Tests for tensorflow.contrib.rnn.python.ops.fused_rnn_cell."""
23 from tensorflow.contrib.rnn.python.ops import fused_rnn_cell
29 from tensorflow.python.ops import rnn namespace
52 outputs, state = rnn.static_rnn(
120 # test bi-directional rnn
123 outputs, fw_state, bw_state = rnn.static_bidirectional_rnn(
H A Dcore_rnn_test.py15 """Tests for rnn module."""
26 from tensorflow.contrib import rnn as rnn_lib
39 from tensorflow.python.ops import rnn namespace
156 rnn.static_rnn(cell, inputs, dtype=dtypes.float32, sequence_length=4)
166 outputs, state = rnn.static_rnn(cell, inputs, dtype=dtypes.float32)
196 outputs, state = rnn.static_rnn(cell, inputs, dtype=dtypes.float32)
198 dropped_outputs, _ = rnn.static_rnn(
228 dynamic_outputs, dynamic_state = rnn.static_rnn(
279 prefix = prefix or "rnn"
297 return rnn
[all...]
H A Dgru_ops_test.py23 from tensorflow.contrib.rnn.python.kernel_tests import benchmarking
24 from tensorflow.contrib.rnn.python.ops import gru_ops
33 from tensorflow.python.ops import rnn namespace
53 _, output = rnn.dynamic_rnn(
117 outputs_dynamic, state_dynamic = rnn.dynamic_rnn(
130 outputs_dynamic, state_dynamic = rnn.dynamic_rnn(
243 outputs_dynamic, _ = rnn.dynamic_rnn(
261 outputs_dynamic, _ = rnn.dynamic_rnn(
362 outputs_dynamic, _ = rnn.dynamic_rnn(
382 outputs_dynamic, _ = rnn
[all...]
H A Dlstm_ops_test.py23 from tensorflow.contrib.rnn.python.kernel_tests import benchmarking
24 from tensorflow.contrib.rnn.python.ops import lstm_ops
32 from tensorflow.python.ops import rnn namespace
60 "rnn/lstm_cell/w_i_diag", shape=[cell_size], dtype=dtypes.float32)
62 "rnn/lstm_cell/w_f_diag", shape=[cell_size], dtype=dtypes.float32)
64 "rnn/lstm_cell/w_o_diag", shape=[cell_size], dtype=dtypes.float32)
67 "rnn/lstm_cell/kernel",
71 "rnn/lstm_cell/bias",
78 basic_outputs_op, basic_state_op = rnn.static_rnn(
102 name="rnn/lstm_cel
[all...]
/external/tensorflow/tensorflow/contrib/learn/python/learn/ops/
H A Dseq2seq_ops.py22 from tensorflow.contrib import rnn namespace
146 _, last_enc_state = rnn.static_rnn(
/external/tensorflow/tensorflow/core/profiler/g3doc/
H A Dadvise.md60 rnn.py:218:_rnn_step:_maybe_copy_some_..., cpu: 303.76ms, accelerator: 73.54ms, total: 377.30ms
61 rnn.py:216:_rnn_step:time >= max_seque..., cpu: 2.75ms, accelerator: 0us, total: 2.75ms
65 rnn.py:218:_rnn_step:_maybe_copy_some_..., cpu: 292.04ms, accelerator: 73.54ms, total: 365.58ms
66 rnn.py:216:_rnn_step:time >= max_seque..., cpu: 4.07ms, accelerator: 0us, total: 4.07ms
/external/tensorflow/tensorflow/python/profiler/internal/
H A Dmodel_analyzer_testlib.py29 from tensorflow.python.ops import rnn namespace
59 """Build the full model with conv,rnn,opt."""
66 out = rnn.dynamic_rnn(
/external/tensorflow/tensorflow/python/kernel_tests/
H A Drnn_test.py15 """Tests for rnn module."""
27 from tensorflow.contrib import rnn as contrib_rnn
40 from tensorflow.python.ops import rnn namespace
119 rnn.dynamic_rnn(
138 outputs, state = rnn.dynamic_rnn(cell, inputs, dtype=dtypes.float32)
147 outputs, state = rnn.dynamic_rnn(
162 outputs, state = rnn.dynamic_rnn(cell, inputs, dtype=dtypes.float32)
166 outputs, state = rnn.dynamic_rnn(
184 outputs, state = rnn.dynamic_rnn(
208 outputs, state = rnn
[all...]
/external/tensorflow/tensorflow/contrib/crf/python/ops/
H A Dcrf.py61 from tensorflow.python.ops import rnn namespace
140 _, alphas = rnn.dynamic_rnn(
482 backpointers, last_score = rnn.dynamic_rnn( # [B, T - 1, O], [B, O]
497 decode_tags, _ = rnn.dynamic_rnn( # [B, T - 1, 1]
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/
H A Drnn_ptb.py18 https://github.com/tensorflow/models/tree/master/tutorials/rnn/ptb
98 https://github.com/tensorflow/models/tree/master/tutorials/rnn/ptb
115 self.rnn = cudnn_rnn.CudnnLSTM(
118 self.rnn = RNN(hidden_dim, num_layers, self.keep_ratio)
119 self.track_layer(self.rnn)
139 y, _ = self.rnn(y, training=training)
357 "--no-use-cudnn-rnn",
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/
H A Dseq2seq_test.py28 from tensorflow.contrib.rnn.python.ops import core_rnn_cell
38 from tensorflow.python.ops import rnn namespace
54 _, enc_state = rnn.static_rnn(
107 _, enc_state = rnn.static_rnn(cell, inp, dtype=dtypes.float32)
325 enc_outputs, enc_state = rnn.static_rnn(cell, inp, dtype=dtypes.float32)
350 enc_outputs, enc_state = rnn.static_rnn(cell, inp, dtype=dtypes.float32)
376 enc_outputs, enc_state = rnn.dynamic_rnn(
400 enc_outputs, enc_state = rnn.dynamic_rnn(
428 enc_outputs, enc_state = rnn.static_rnn(cell, inp, dtype=dtypes.float32)
458 enc_outputs, enc_state = rnn
[all...]
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
H A Ddecoder_test.py28 from tensorflow.python.ops import rnn namespace
146 final_rnn_outputs, final_rnn_state = rnn.dynamic_rnn(
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
H A Ddecoder.py32 from tensorflow.python.ops import rnn namespace
41 _transpose_batch_time = rnn._transpose_batch_time # pylint: disable=protected-access

Completed in 486 milliseconds

12