/external/tensorflow/tensorflow/contrib/rnn/ |
H A D | __init__.py | 17 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 D | bidirectional_sequence_rnn_test.cc | 767 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 D | unidirectional_sequence_rnn_test.cc | 207 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 D | basic_rnn_test.cc | 190 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 D | basic_rnn.cc | 32 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 D | fused_rnn_cell.py | 24 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 D | rnn.py | 21 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 D | wrappers_test.py | 154 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 D | nn.py | 129 from tensorflow.python.ops.rnn import *
|
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
H A D | cudnn_rnn_test.py | 31 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 D | cudnn_rnn_ops_benchmark.py | 24 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 D | grid_rnn_cell.py | 31 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 D | fused_rnn_cell_test.py | 15 """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 D | core_rnn_test.py | 15 """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 D | gru_ops_test.py | 23 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 D | lstm_ops_test.py | 23 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 D | seq2seq_ops.py | 22 from tensorflow.contrib import rnn namespace 146 _, last_enc_state = rnn.static_rnn(
|
/external/tensorflow/tensorflow/core/profiler/g3doc/ |
H A D | advise.md | 60 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 D | model_analyzer_testlib.py | 29 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 D | rnn_test.py | 15 """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 D | crf.py | 61 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 D | rnn_ptb.py | 18 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 D | seq2seq_test.py | 28 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 D | decoder_test.py | 28 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 D | decoder.py | 32 from tensorflow.python.ops import rnn namespace 41 _transpose_batch_time = rnn._transpose_batch_time # pylint: disable=protected-access
|