| /external/tensorflow/tensorflow/python/ops/ |
| rnn_cell.py | 22 from tensorflow.python.ops.rnn_cell_impl import *
|
| rnn.py | 32 from tensorflow.python.ops import rnn_cell_impl
41 _concat = rnn_cell_impl._concat
166 return (not isinstance(rnn_cell, rnn_cell_impl.RNNCell)
433 rnn_cell_impl.assert_like_rnncell("cell_fw", cell_fw)
434 rnn_cell_impl.assert_like_rnncell("cell_bw", cell_bw)
604 rnn_cell_impl.assert_like_rnncell("cell", cell)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| rnn_test.py | 45 from tensorflow.python.ops import rnn_cell_impl
59 class Plus1RNNCell(rnn_cell_impl.RNNCell):
74 class ScalarStateRNNCell(rnn_cell_impl.RNNCell):
92 class UnbalancedOutputRNNCell(rnn_cell_impl.RNNCell):
111 class TensorArrayStateRNNCell(rnn_cell_impl.RNNCell):
136 self._cell = rnn_cell_impl.MultiRNNCell(
137 [rnn_cell_impl.LSTMCell(1) for _ in range(2)])
182 rnn_cell_impl.BasicRNNCell,
183 rnn_cell_impl.GRUCell,
184 rnn_cell_impl.BasicLSTMCell
[all...] |
| rnn_cell_test.py | 45 from tensorflow.python.ops import rnn_cell_impl
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| rnn_cell.py | 44 from tensorflow.python.ops import rnn_cell_impl
98 class CoupledInputForgetGateLSTMCell(rnn_cell_impl.RNNCell):
203 rnn_cell_impl.LSTMStateTuple(num_units, num_proj)
208 rnn_cell_impl.LSTMStateTuple(num_units, num_units)
320 rnn_cell_impl.LSTMStateTuple(c, m)
325 class TimeFreqLSTMCell(rnn_cell_impl.RNNCell):
499 class GridLSTMCell(rnn_cell_impl.RNNCell):
[all...] |
| core_rnn_cell.py | 35 from tensorflow.python.ops import rnn_cell_impl
42 RNNCell = rnn_cell_impl.RNNCell
43 _WEIGHTS_VARIABLE_NAME = rnn_cell_impl._WEIGHTS_VARIABLE_NAME
44 _BIAS_VARIABLE_NAME = rnn_cell_impl._BIAS_VARIABLE_NAME
223 rnn_cell_impl.assert_like_rnncell("cell", cell)
302 rnn_cell_impl.assert_like_rnncell("cell", cell)
356 rnn_cell_impl.assert_like_rnncell("cell", cell)
|
| gru_ops.py | 29 from tensorflow.python.ops import rnn_cell_impl
36 LayerRNNCell = rnn_cell_impl.LayerRNNCell # pylint: disable=invalid-name
|
| lstm_ops.py | 34 from tensorflow.python.ops import rnn_cell_impl
40 LayerRNNCell = rnn_cell_impl.LayerRNNCell # pylint: disable=invalid-name
343 Unlike `rnn_cell_impl.LSTMCell`, this is a monolithic op and should be much
393 return rnn_cell_impl.LSTMStateTuple(self._num_units, self._num_units)
442 new_state = rnn_cell_impl.LSTMStateTuple(cs, h)
576 final_state = rnn_cell_impl.LSTMStateTuple(final_cell_state, final_output)
597 The variable naming is consistent with `rnn_cell_impl.LSTMCell`.
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn.py | 31 from tensorflow.python.ops import rnn_cell_impl
165 rnn_cell_impl.assert_like_rnncell("cell", cell)
364 rnn_cell_impl.assert_like_rnncell("cell_fw", cell_fw)
365 rnn_cell_impl.assert_like_rnncell("cell_bw", cell_bw)
|
| rnn_cell.py | 34 from tensorflow.python.ops import rnn_cell_impl
40 class TfLiteRNNCell(rnn_cell_impl.LayerRNNCell):
160 class TFLiteLSTMCell(rnn_cell_impl.LayerRNNCell):
273 rnn_cell_impl.LSTMStateTuple(num_units, self._output_size)
508 rnn_cell_impl.LSTMStateTuple(c, m)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| basic_decoder.py | 31 from tensorflow.python.ops import rnn_cell_impl
64 rnn_cell_impl.assert_like_rnncell("cell", cell)
170 rnn_cell_impl.assert_like_rnncell("cell", cell)
|
| decoder.py | 36 from tensorflow.python.ops import rnn_cell_impl
46 _zero_state_tensors = rnn_cell_impl._zero_state_tensors # pylint: disable=protected-access
|
| attention_wrapper.py | 45 from tensorflow.python.ops import rnn_cell_impl
65 _zero_state_tensors = rnn_cell_impl._zero_state_tensors # pylint: disable=protected-access
[all...] |
| beam_search_decoder.py | 38 from tensorflow.python.ops import rnn_cell_impl
274 rnn_cell_impl.assert_like_rnncell("cell", cell) # pylint: disable=protected-access
[all...] |
| /external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/ |
| functional_rnn_test.py | 31 from tensorflow.python.ops import rnn_cell_impl
40 subcells = [rnn_cell_impl.LSTMCell(cell_size) for cell_size in cell_sizes]
41 return rnn_cell_impl.MultiRNNCell(subcells)
55 'gru': (rnn_cell_impl.GRUCell, [_NUM_UNITS]),
56 'lstm': (rnn_cell_impl.LSTMCell, [_NUM_UNITS]),
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| rnn_cell_test.py | 47 from tensorflow.python.ops import rnn_cell_impl
87 "root/ind_rnn_cell/%s_w:0" % rnn_cell_impl._WEIGHTS_VARIABLE_NAME,
88 "root/ind_rnn_cell/%s_u:0" % rnn_cell_impl._WEIGHTS_VARIABLE_NAME,
89 "root/ind_rnn_cell/%s:0" % rnn_cell_impl._BIAS_VARIABLE_NAME
136 cell = rnn_cell_impl.MultiRNNCell(
147 rnn_cell_impl._WEIGHTS_VARIABLE_NAME,
149 rnn_cell_impl._WEIGHTS_VARIABLE_NAME,
151 rnn_cell_impl._BIAS_VARIABLE_NAME,
153 rnn_cell_impl._WEIGHTS_VARIABLE_NAME,
155 rnn_cell_impl._WEIGHTS_VARIABLE_NAME
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_test.py | 47 from tensorflow.python.ops import rnn_cell_impl
243 single_cell = (lambda: rnn_cell_impl.BasicRNNCell(num_units, math_ops.tanh))
246 lambda: rnn_cell_impl.BasicRNNCell(num_units, gen_nn_ops.relu))
251 cell = rnn_cell_impl.MultiRNNCell(
769 lambda: rnn_cell_impl.MultiRNNCell([single_cell_fn()]),
807 return rnn_cell_impl.MultiRNNCell(
[all...] |
| cudnn_rnn_ops_test.py | 42 from tensorflow.python.ops import rnn_cell_impl
127 cell = rnn_cell_impl.LSTMCell(num_units, forget_bias=0., reuse=True)
132 initial_state=rnn_cell_impl.LSTMStateTuple(
158 cu_state_tuple_op = rnn_cell_impl.LSTMStateTuple(
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | 32 from tensorflow.python.ops import rnn_cell_impl
56 _BIAS_VARIABLE_NAME = rnn_cell_impl._BIAS_VARIABLE_NAME
57 _WEIGHTS_VARIABLE_NAME = rnn_cell_impl._WEIGHTS_VARIABLE_NAME
76 class CudnnCompatibleGRUCell(rnn_cell_impl.GRUCell):
[all...] |
| /external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/ops/ |
| seq2seq.py | 74 from tensorflow.python.ops import rnn_cell_impl
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| mirrored_strategy_multigpu_test.py | 53 from tensorflow.python.ops import rnn_cell_impl
736 cell_fw = rnn_cell_impl.LSTMCell(300)
737 cell_bw = rnn_cell_impl.LSTMCell(300)
[all...] |
| /external/tensorflow/tensorflow/python/debug/lib/ |
| session_debug_testlib.py | 49 from tensorflow.python.ops import rnn_cell_impl
67 class _RNNCellForTest(rnn_cell_impl.RNNCell):
[all...] |
