| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_benchmark.py | 134 outputs, final_state = rnn.dynamic_rnn(
138 gradients = gradients_impl.gradients([outputs, final_state],
158 outputs, final_state = rnn.dynamic_rnn(
162 gradients = gradients_impl.gradients([outputs, final_state],
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| attention_wrapper_v2_test.py | 326 final_outputs, final_state, _ = my_decoder(
332 self.assertIsInstance(final_state, wrapper.AttentionWrapperState)
342 tuple(final_state.attention.get_shape().as_list()))
344 tuple(final_state.cell_state[0].get_shape().as_list()))
346 tuple(final_state.cell_state[1].get_shape().as_list()))
351 for history_array in final_state.alignment_history:
358 state_alignment_history = final_state.alignment_history.stack()
363 # Remove the history from final_state for purposes of the
365 final_state = final_state._replace(alignment_history=()) # pylint: disable=protected-acces
[all...] |
| attention_wrapper_test.py | 211 final_outputs, final_state, _ = decoder.dynamic_decode(my_decoder)
216 isinstance(final_state, wrapper.AttentionWrapperState))
218 isinstance(final_state.cell_state, rnn_cell.LSTMStateTuple))
226 tuple(final_state.attention.get_shape().as_list()))
228 tuple(final_state.cell_state.c.get_shape().as_list()))
230 tuple(final_state.cell_state.h.get_shape().as_list()))
235 for history_array in final_state.alignment_history:
243 state_alignment_history = final_state.alignment_history.stack()
250 # Remove the history from final_state for purposes of the
252 final_state = final_state._replace(alignment_history=()) # pylint: disable=protected-acces
[all...] |
| decoder_test.py | 62 final_outputs, final_state, final_sequence_length = (
73 self.assertTrue(isinstance(final_state, rnn_cell.LSTMStateTuple))
88 "final_state": final_state,
|
| beam_search_decoder_test.py | 537 final_outputs, final_state, final_sequence_lengths = (
549 final_state, beam_search_decoder.BeamSearchDecoderState)
562 'final_state': final_state,
649 final_outputs, final_state, final_sequence_lengths = bsd(
663 final_state, beam_search_decoder.BeamSearchDecoderState)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| state_saving_rnn_estimator.py | 69 final_state: The final state output by the RNN
72 rnn_outputs, final_state = rnn.static_state_saving_rnn(
86 # Use `identity` to rename `final_state`.
87 final_state = array_ops.identity(
88 final_state, name=rnn_common.RNNKeys.FINAL_STATE_KEY)
89 return activations, final_state
494 rnn_activations, final_state = construct_state_saving_rnn(
514 state_dict = state_tuple_to_dict(final_state)
|
| dynamic_rnn_estimator.py | 240 final_state: A `Tensor` or nested tuple of `Tensor`s representing the final
244 rnn_outputs, final_state = rnn.dynamic_rnn(
257 return activations, final_state
492 rnn_activations, final_state = construct_rnn(
515 state_dict = state_tuple_to_dict(final_state)
|
| dynamic_rnn_estimator_test.py | 182 activations, final_state = sess.run([activations_t, final_state_t])
187 self.assertAllEqual(expected_state_shape, final_state.shape)
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| decoder.py | 115 def finalize(self, outputs, final_state, sequence_lengths):
234 def finalize(self, outputs, final_state, sequence_lengths):
302 `(final_outputs, final_state, final_sequence_lengths)`.
458 final_state = res[2]
464 final_outputs, final_state = decoder.finalize(
465 final_outputs, final_state, final_sequence_lengths)
472 return final_outputs, final_state, final_sequence_lengths
|
| beam_search_decoder.py | 335 def finalize(self, outputs, final_state, sequence_lengths):
340 final_state: An instance of BeamSearchDecoderState. Passed through to the
345 `final_state.lengths`.
350 final_state: The same input instance of `BeamSearchDecoderState`.
355 math_ops.reduce_max(final_state.lengths, axis=1), dtypes.int32)
362 final_state = final_state._replace(cell_state=nest.map_structure(
364 t, outputs.parent_ids, final_state.lengths),
365 final_state.cell_state))
368 return outputs, final_state
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| rnn_test.py | 726 outputs, final_state = rnn.static_rnn(
734 gradients = gradients_impl.gradients(outputs + [final_state],
737 return control_flow_ops.group(final_state, *(gradients + outputs))
748 outputs, final_state = rnn.dynamic_rnn(
753 gradients = gradients_impl.gradients([outputs, final_state],
756 return control_flow_ops.group(final_state, outputs, *gradients)
859 outputs, final_state = rnn.static_rnn(
[all...] |
| rnn_cell_test.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| rnn.py | 192 final_state = np.vstack([
196 return (final_output, final_state)
216 A tuple of (`final_output`, `final_state`) as given by the pseudocode above:
218 final_state is either a single `Tensor` matrix, or a tuple of such
293 final_state = final_output_and_state[len(flat_zero_output):]
297 for substate, flat_substate in zip(final_state, flat_state):
303 final_state = nest.pack_sequence_as(
304 structure=state, flat_sequence=final_state)
306 return final_output, final_state
664 (outputs, final_state) = _dynamic_rnn_loop
[all...] |
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn.py | 259 outputs, final_state = _dynamic_rnn_loop(
276 return outputs, final_state
|
| /external/autotest/client/cros/networking/ |
| wifi_proxy.py | 271 (successful, final_state, duration) = result
275 'timed out in state: %s.' % (ssid, final_state))
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| lstm_ops.py | 576 final_state = rnn_cell_impl.LSTMStateTuple(final_cell_state, final_output)
577 return outputs, final_state
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| gradients_test.py | 121 inp, (_, final_state) = dynamic_lstm_model_fn(batch_size, state_size,
123 pfor_jacobian = gradients.batch_jacobian(final_state.c, inp, use_pfor=True)
128 gradient_ops.gradients(array_ops.gather(final_state.c, i, axis=1), inp)[0]
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| rnn_cell_test.py | [all...] |
