| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_rnn_test.cc | 786 BidirectionalRNNOpModel rnn(/*batches=*/2, /*sequence_len=*/16, in TEST() local
791 rnn.SetFwWeights(weights); in TEST()
792 rnn.SetBwWeights(weights); in TEST()
793 rnn.SetFwBias(biases); in TEST()
794 rnn.SetBwBias(biases); in TEST()
795 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
796 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST()
798 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST()
801 rnn.SetInput(0, batch_start, batch_end); in TEST()
802 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
[all …]
|
| D | unidirectional_sequence_rnn_test.cc | 277 UnidirectionalRNNOpModel rnn(/*batches=*/2, /*sequence_len=*/16, in TEST() local
279 rnn.SetWeights(rnn_weights); in TEST()
280 rnn.SetBias(rnn_bias); in TEST()
281 rnn.SetRecurrentWeights(rnn_recurrent_weights); in TEST()
283 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST()
286 rnn.SetInput(0, batch_start, batch_end); in TEST()
287 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
289 rnn.Invoke(); in TEST()
292 float* golden_end = golden_start + rnn.num_units() * rnn.sequence_len(); in TEST()
297 EXPECT_THAT(rnn.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
[all …]
|
| D | basic_rnn_test.cc | 259 RNNOpModel rnn(2, 16, 8); in TEST() local
260 rnn.SetWeights(rnn_weights); in TEST()
261 rnn.SetBias(rnn_bias); in TEST()
262 rnn.SetRecurrentWeights(rnn_recurrent_weights); in TEST()
265 (rnn.input_size() * rnn.num_batches()); in TEST()
268 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
269 float* batch_end = batch_start + rnn.input_size(); in TEST()
270 rnn.SetInput(0, batch_start, batch_end); in TEST()
271 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
273 rnn.Invoke(); in TEST()
[all …]
|
| D | basic_rnn.cc | 28 namespace rnn { namespace
256 static TfLiteRegistration r = {rnn::Init, rnn::Free, rnn::Prepare, rnn::Eval}; in Register_RNN()
|
| /external/tensorflow/tensorflow/contrib/rnn/ |
| D | __init__.py | 85 from tensorflow.contrib.rnn.python.ops.core_rnn_cell import EmbeddingWrapper
86 from tensorflow.contrib.rnn.python.ops.core_rnn_cell import InputProjectionWrapper
87 from tensorflow.contrib.rnn.python.ops.core_rnn_cell import OutputProjectionWrapper
89 from tensorflow.contrib.rnn.python.ops.fused_rnn_cell import *
90 from tensorflow.contrib.rnn.python.ops.gru_ops import *
91 from tensorflow.contrib.rnn.python.ops.lstm_ops import *
92 from tensorflow.contrib.rnn.python.ops.rnn import *
93 from tensorflow.contrib.rnn.python.ops.rnn_cell import *
95 from tensorflow.python.ops.rnn import _best_effort_input_batch_size as best_effort_input_batch_size
96 from tensorflow.python.ops.rnn import _transpose_batch_time as transpose_batch_time
[all …]
|
| /external/tensorflow/tensorflow/contrib/seq2seq/ |
| D | BUILD | 37 "//tensorflow/contrib/rnn:rnn_py",
54 "//tensorflow/python:rnn",
125 "//tensorflow/contrib/rnn:rnn_py",
132 "//tensorflow/python:rnn",
146 "//tensorflow/contrib/rnn:rnn_py",
153 "//tensorflow/python:rnn",
180 "//tensorflow/contrib/rnn:rnn_py",
187 "//tensorflow/python:rnn",
201 "//tensorflow/contrib/rnn:rnn_py",
208 "//tensorflow/python:rnn",
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | BUILD | 8 name = "rnn",
9 srcs = ["rnn.py"],
16 "//tensorflow/python:rnn",
45 ":rnn",
68 ":rnn",
91 ":rnn",
114 ":rnn",
132 ":rnn",
|
| D | rnn.py | 33 from tensorflow.python.ops.rnn import _best_effort_input_batch_size
34 from tensorflow.python.ops.rnn import _dynamic_rnn_loop
35 from tensorflow.python.ops.rnn import _should_cache
36 from tensorflow.python.ops.rnn import _transpose_batch_time
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | lstm_v2_test.py | 39 from tensorflow.python.keras.layers import recurrent_v2 as rnn unknown
71 layer = rnn.LSTM(
90 layer = rnn.LSTM(units, return_sequences=True)
100 layer = rnn.LSTM(units, input_shape=(None, embedding_dim))
113 model.add(rnn.LSTM(10, return_sequences=True, unroll=False))
114 model.add(rnn.LSTM(5, return_sequences=True, unroll=False))
121 layer_class = rnn.LSTM
137 layer = rnn.LSTM(units)
169 layer = rnn.LSTM(units)
188 layer = rnn.LSTM(units, stateful=True)
[all …]
|
| D | gru_v2_test.py | 38 from tensorflow.python.keras.layers import recurrent_v2 as rnn unknown
72 layer = rnn.GRU(1,
96 layer = rnn.GRU(rnn_state_size)
113 layer = rnn.GRU(units, input_shape=(None, embedding_dim))
126 model.add(rnn.GRU(10, return_sequences=True, unroll=False))
127 model.add(rnn.GRU(5, return_sequences=True, unroll=False))
134 layer_class = rnn.GRU
169 cudnn_layer = rnn.GRU(rnn_state_size,
203 layer = rnn.GRU(
232 layer = rnn.GRU(rnn_state_size)
[all …]
|
| D | wrappers_test.py | 298 rnn = keras.layers.SimpleRNN
313 rnn(output_dim), merge_mode=mode, input_shape=(timesteps, dim)))
343 rnn = keras.layers.SimpleRNN
353 rnn(output_dim), input_shape=(timesteps, dim)))
362 rnn = keras.layers.SimpleRNN
377 rnn(output_dim, return_sequences=True),
380 model.add(keras.layers.Bidirectional(rnn(output_dim), merge_mode=mode))
387 rnn(output_dim), merge_mode=mode)(inputs)
394 rnn = keras.layers.SimpleRNN
408 rnn(output_dim, stateful=True), merge_mode=mode)(inputs)
[all …]
|
| /external/tensorflow/tensorflow/contrib/estimator/python/estimator/ |
| D | rnn.py | 26 from tensorflow_estimator.contrib.estimator.python.estimator import rnn
30 rnn.__all__ = [s for s in dir(rnn) if not s.startswith('__')]
32 from tensorflow_estimator.contrib.estimator.python.estimator.rnn import *
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_test.py | 32 from tensorflow.contrib.rnn.python.ops import rnn as contrib_rnn_lib
46 from tensorflow.python.ops import rnn as rnn_lib
157 def rnn(self): member in CudnnTestModel
232 def _CreateCudnnCompatibleCanonicalRNN(rnn, inputs, is_bidi=False, scope=None): argument
233 mode = rnn.rnn_mode
234 num_units = rnn.num_units
235 num_layers = rnn.num_layers
535 rnn = model.rnn
540 model.rnn.saveable.format_converter._opaque_to_cu_canonical(
541 model.rnn.saveable._variables))
[all …]
|
| 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
134 outputs, final_state = rnn.dynamic_rnn(
158 outputs, final_state = rnn.dynamic_rnn(
|
| /external/tensorflow/tensorflow/python/ |
| D | __init__.py | 144 from tensorflow.python.ops import rnn
149 nn.dynamic_rnn = rnn.dynamic_rnn
150 nn.static_rnn = rnn.static_rnn
151 nn.raw_rnn = rnn.raw_rnn
152 nn.bidirectional_dynamic_rnn = rnn.bidirectional_dynamic_rnn
153 nn.static_state_saving_rnn = rnn.static_state_saving_rnn
|
| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| D | grid_rnn_cell.py | 32 from tensorflow.contrib import rnn
35 class GridRNNCell(rnn.RNNCell):
130 rnn.LSTMCell, num_units=num_units, state_is_tuple=state_is_tuple)
137 if not isinstance(self._cells[0], rnn.RNNCell):
338 cell_fn=lambda n: rnn.BasicRNNCell(num_units=n),
367 cell_fn=lambda n: rnn.BasicRNNCell(num_units=n),
382 return rnn.BasicLSTMCell(num_units=n, forget_bias=forget_bias)
413 return rnn.BasicLSTMCell(num_units=n, forget_bias=forget_bias)
443 return rnn.LSTMCell(
476 return rnn.LSTMCell(
[all …]
|
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/ |
| D | BUILD | 25 "//tensorflow/contrib/rnn:rnn_py",
32 "//tensorflow/python:rnn",
46 "//tensorflow/contrib/rnn:rnn_py",
55 "//tensorflow/python:rnn",
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| 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
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.dynamic_rnn(
[all …]
|
| D | lstm_ops_test.py | 24 from tensorflow.contrib.rnn.python.kernel_tests import benchmarking
25 from tensorflow.contrib.rnn.python.ops import lstm_ops
35 from tensorflow.python.ops import rnn
150 basic_outputs_op, basic_state_op = rnn.static_rnn(
228 output, _ = rnn.dynamic_rnn(
268 rnn.static_rnn(cell, inputs, dtype=dtypes.float32, scope="basic")
269 rnn.static_rnn(pcell, inputs, dtype=dtypes.float32, scope="peephole")
279 rnn.static_rnn(cell, inputs, dtype=dtypes.float32, scope="basic")
280 rnn.static_rnn(pcell, inputs, dtype=dtypes.float32, scope="peephole")
549 outputs = rnn.dynamic_rnn(cell, inputs, time_major=True, dtype=dtype)
[all …]
|
| D | fused_rnn_cell_test.py | 23 from tensorflow.contrib.rnn.python.ops import fused_rnn_cell
29 from tensorflow.python.ops import rnn
52 outputs, state = rnn.static_rnn(
123 outputs, fw_state, bw_state = rnn.static_bidirectional_rnn(
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_core_kernels.cmake | 96 "${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/blas_gemm.cc"
97 "${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/gru_ops.cc"
98 "${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/lstm_ops.cc"
99 "${tensorflow_source_dir}/tensorflow/contrib/rnn/ops/gru_ops.cc"
100 "${tensorflow_source_dir}/tensorflow/contrib/rnn/ops/lstm_ops.cc"
158 "${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/blas_gemm.cc"
159 "${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/gru_ops.cc"
160 "${tensorflow_source_dir}/tensorflow/contrib/rnn/kernels/lstm_ops.cc"
161 "${tensorflow_source_dir}/tensorflow/contrib/rnn/ops/gru_ops.cc"
162 "${tensorflow_source_dir}/tensorflow/contrib/rnn/ops/lstm_ops.cc"
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_test.py | 44 from tensorflow.python.ops import rnn
168 rnn.dynamic_rnn(
192 rnn.dynamic_rnn(cell, inputs, dtype=dtypes.int32)
207 outputs, state = rnn.dynamic_rnn(cell, inputs, dtype=dtypes.float32)
212 outputs, state = rnn.dynamic_rnn(
223 outputs, state = rnn.dynamic_rnn(cell, inputs, dtype=dtypes.float32)
227 outputs, state = rnn.dynamic_rnn(
245 outputs, state = rnn.dynamic_rnn(
265 outputs, state = rnn.dynamic_rnn(
281 rnn.dynamic_rnn(cell, inputs, dtype=dtypes.float32, sequence_length=[4])
[all …]
|
| D | rnn_cell_test.py | 43 from tensorflow.python.ops import rnn
206 rnn.static_rnn(cell, inputs, dtype=dtypes.float32, sequence_length=4)
217 outputs, state = rnn.static_rnn(cell, inputs, dtype=dtypes.float32)
251 outputs, state = rnn.static_rnn(cell, inputs, dtype=dtypes.float32)
253 dropped_outputs, _ = rnn.static_rnn(
284 dynamic_outputs, dynamic_state = rnn.static_rnn(
354 return rnn.static_rnn(cell, inputs, dtype=dtypes.float32, scope=scope)
396 outputs, _ = rnn.static_rnn(cell, inputs, dtype=dtypes.float32)
423 outputs, _ = rnn.static_rnn(cell, inputs, dtype=dtypes.float32)
455 outputs, state = rnn.static_state_saving_rnn(
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | fused_rnn_cell.py | 26 from tensorflow.python.ops import rnn
105 outputs, state = rnn.dynamic_rnn(
119 outputs, state = rnn.static_rnn(
|
| D | rnn.py | 21 from tensorflow.python.ops import rnn
110 prev_layer, state_fw, state_bw = rnn.static_bidirectional_rnn(
229 outputs, (state_fw, state_bw) = rnn.bidirectional_dynamic_rnn(
|