Home
last modified time | relevance | path

Searched refs:state_is_tuple (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
Dgrid_rnn_cell.py60 state_is_tuple=True, argument
110 if not state_is_tuple:
125 self._state_is_tuple = state_is_tuple
130 rnn.LSTMCell, num_units=num_units, state_is_tuple=state_is_tuple)
330 def __init__(self, num_units, state_is_tuple=True, output_is_tuple=True): argument
339 state_is_tuple=state_is_tuple,
357 state_is_tuple=True, argument
369 state_is_tuple=state_is_tuple,
379 state_is_tuple=True, argument
391 state_is_tuple=state_is_tuple,
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dmodels.py340 contrib_rnn.BasicLSTMCell, state_is_tuple=False)
356 state_is_tuple=False)
364 state_is_tuple=False)
370 [attn_fw_cell() for _ in range(num_layers)], state_is_tuple=False)
373 [attn_bw_cell() for _ in range(num_layers)], state_is_tuple=False)
393 state_is_tuple=False)
398 [attn_rnn_cell() for _ in range(num_layers)], state_is_tuple=False)
/external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
Drnn_cell_test.py295 rnn_cell_impl.BasicLSTMCell(1, state_is_tuple=True),
385 num_units=num_units, forget_bias=1.0, state_is_tuple=False)(x, m)
456 state_is_tuple=True)
506 state_is_tuple=True)
567 for state_is_tuple in [False, True]:
570 "state_is_tuple" + str(state_is_tuple),
579 state_is_tuple=state_is_tuple)
585 if state_is_tuple:
602 if not state_is_tuple:
780 for state_is_tuple in [False, True]:
[all …]
Dlstm_ops_test.py148 state_is_tuple=True,
246 for _ in range(2)], state_is_tuple=True)(x, ((m0, m1), (m2, m3)))
318 [rnn_cell.BasicLSTMCell(2, state_is_tuple=True) for _ in range(2)],
319 state_is_tuple=True)(x, ((m0, m1), (m2, m3)))
336 for _ in range(2)], state_is_tuple=True)(x, ((m0, m1), (m2, m3)))
369 rnn_cell.LSTMCell(2, use_peepholes=True, state_is_tuple=True)
372 state_is_tuple=True)(x, ((m0, m1), (m2, m3)))
389 state_is_tuple=True)(x, ((m0, m1), (m2, m3)))
Drnn_test.py65 state_is_tuple=False) for num_units in self.layers
72 state_is_tuple=False) for num_units in self.layers
244 state_is_tuple=False) for num_units in self.layers
251 state_is_tuple=False) for num_units in self.layers
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py58 state_is_tuple=True, argument
86 state_is_tuple=state_is_tuple,
190 state_is_tuple=True, argument
236 state_is_tuple=state_is_tuple,
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py392 num_units, initializer=initializer, state_is_tuple=False)
419 state_is_tuple=False)
450 state_is_tuple=False)
484 state_is_tuple=True)
531 state_is_tuple=True)
535 [_cell(i) for i in range(4)], state_is_tuple=True)
594 state_is_tuple=False)
620 state_is_tuple=False)
626 state_is_tuple=True)
696 state_is_tuple=False)
[all …]
Drnn_test.py725 state_is_tuple=False)
747 state_is_tuple=False)
858 state_is_tuple=False)
917 state_is_tuple): argument
924 state_is_tuple=state_is_tuple)
931 final_state = list(final_state) if state_is_tuple else [final_state]
962 inputs_list_t, sequence_length, state_is_tuple=False)
974 inputs_list_t, sequence_length, state_is_tuple=True)
991 state_is_tuple=False)
/external/tensorflow/tensorflow/contrib/legacy_seq2seq/python/kernel_tests/
Dseq2seq_test.py158 cell_nt = rnn_cell.BasicLSTMCell(2, state_is_tuple=False)
242 cell = functools.partial(rnn_cell.BasicLSTMCell, 2, state_is_tuple=True)
423 2, state_is_tuple=True)
425 cells=[single_cell() for _ in range(2)], state_is_tuple=True)
548 rnn_cell.BasicLSTMCell, 2, state_is_tuple=False)
643 return rnn_cell.BasicLSTMCell(2, state_is_tuple=True)
645 return dict((k, rnn_cell.BasicLSTMCell(2, state_is_tuple=True))
854 [rnn_cell.GRUCell(24) for _ in range(2)], state_is_tuple=True)
1030 cell = rnn_cell.BasicLSTMCell(2, state_is_tuple=True)
1041 cell = rnn_cell.BasicLSTMCell(2, state_is_tuple=False)
[all …]
/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py660 state_is_tuple=True, argument
695 if not state_is_tuple:
708 self._state_is_tuple = state_is_tuple
835 forget_bias=1.0, state_is_tuple=True, argument
881 if not state_is_tuple:
906 self._state_is_tuple = state_is_tuple
915 if state_is_tuple else num_units + num_proj)
920 if state_is_tuple else 2 * num_units)
1598 def __init__(self, cells, state_is_tuple=True): argument
1631 self._state_is_tuple = state_is_tuple
[all …]
Drnn.py1438 state_is_tuple = nest.is_sequence(state_size)
1441 if state_is_tuple != state_name_tuple:
1446 if state_is_tuple:
1467 if state_is_tuple:
1486 if state_is_tuple:
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py200 state_is_tuple=True, argument
243 if not state_is_tuple:
268 self._state_is_tuple = state_is_tuple
274 if state_is_tuple else num_units + self._output_size)
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars_test.py114 dim, forget_bias=0.0, state_is_tuple=True, reuse=False)
117 [lstm_cell() for _ in range(number_of_layers)], state_is_tuple=True)
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn_cell.py144 state_is_tuple=True, argument
183 if not state_is_tuple:
194 self._state_is_tuple = state_is_tuple
204 if state_is_tuple else num_units + num_proj)
209 if state_is_tuple else 2 * num_units)
529 state_is_tuple=True, argument
572 if not state_is_tuple:
579 self._state_is_tuple = state_is_tuple
596 if state_is_tuple:
653 state_is_tuple=self._state_is_tuple)
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.rnn_cell.-basic-l-s-t-m-cell.pbtxt108 …argspec: "args=[\'self\', \'num_units\', \'forget_bias\', \'state_is_tuple\', \'activation\', \'re…
Dtensorflow.nn.rnn_cell.-multi-r-n-n-cell.pbtxt107 …argspec: "args=[\'self\', \'cells\', \'state_is_tuple\'], varargs=None, keywords=None, defaults=[\…
Dtensorflow.nn.rnn_cell.-l-s-t-m-cell.pbtxt108 …clip\', \'num_unit_shards\', \'num_proj_shards\', \'forget_bias\', \'state_is_tuple\', \'activatio…
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt108 …clip\', \'num_unit_shards\', \'num_proj_shards\', \'forget_bias\', \'state_is_tuple\', \'activatio…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.experimental.nn.-t-f-lite-l-s-t-m-cell.pbtxt108 …clip\', \'num_unit_shards\', \'num_proj_shards\', \'forget_bias\', \'state_is_tuple\', \'activatio…
/external/tensorflow/tensorflow/contrib/grid_rnn/python/kernel_tests/
Dgrid_rnn_test.py724 2, use_peepholes=True, state_is_tuple=False, output_is_tuple=False)
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py3029 num_cells, forget_bias=0.0, state_is_tuple=True)
/external/tensorflow/
DRELEASE.md1736 `state_is_tuple=True`. For a quick fix while transitioning to the new
1737 default, simply pass the argument `state_is_tuple=False`.