Home
last modified time | relevance | path

Searched refs:final_outputs (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dstrip_pruning_vars_test.py163 final_outputs = self._get_outputs(
168 return final_outputs
175 def _check_output_equivalence(self, initial_outputs, final_outputs): argument
176 for initial_output, final_output in zip(initial_outputs, final_outputs):
188 final_outputs = self._get_final_outputs(output_tensor_names)
194 self._check_output_equivalence(initial_outputs, final_outputs)
205 final_outputs = self._get_final_outputs(output_tensor_names)
211 self._check_output_equivalence(initial_outputs, final_outputs)
222 final_outputs = self._get_final_outputs(output_tensor_names)
228 self._check_output_equivalence(initial_outputs, final_outputs)
/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Ddecoder_v2_test.py65 (final_outputs, unused_final_state, final_sequence_length) = my_decoder(
78 tuple(final_outputs.rnn_output.get_shape().as_list()))
81 tuple(final_outputs.sample_id.get_shape().as_list()))
84 final_outputs = self.evaluate(final_outputs)
98 final_outputs.rnn_output.shape)
100 _t((batch_size, time_steps)), final_outputs.sample_id.shape)
Ddecoder_test.py62 final_outputs, final_state, final_sequence_length = (
72 isinstance(final_outputs, basic_decoder.BasicDecoderOutput))
80 tuple(final_outputs.rnn_output.get_shape().as_list()))
83 tuple(final_outputs.sample_id.get_shape().as_list()))
87 "final_outputs": final_outputs,
Dbeam_search_decoder_test.py537 final_outputs, final_state, final_sequence_lengths = (
547 final_outputs, beam_search_decoder.FinalBeamSearchDecoderOutput)
551 beam_search_decoder_output = final_outputs.beam_search_decoder_output
557 tuple(final_outputs.predicted_ids.get_shape().as_list()))
561 'final_outputs': final_outputs,
649 final_outputs, final_state, final_sequence_lengths = bsd(
661 final_outputs, beam_search_decoder.FinalBeamSearchDecoderOutput)
665 beam_search_decoder_output = final_outputs.beam_search_decoder_output
672 tuple(final_outputs.predicted_ids.get_shape().as_list()))
676 'final_outputs': final_outputs,
Dattention_wrapper_v2_test.py326 final_outputs, final_state, _ = my_decoder(
331 self.assertIsInstance(final_outputs, basic_decoder.BasicDecoderOutput)
337 tuple(final_outputs.rnn_output.get_shape().as_list()))
339 tuple(final_outputs.sample_id.get_shape().as_list()))
371 "final_outputs": final_outputs,
414 final_outputs, final_state, _ = my_decoder(
418 self.assertIsInstance(final_outputs, basic_decoder.BasicDecoderOutput)
419 self.assertEqual(final_outputs.rnn_output.dtype, dtype)
441 final_outputs, final_state, _ = my_decoder(
445 self.assertIsInstance(final_outputs, basic_decoder.BasicDecoderOutput)
[all …]
Dattention_wrapper_test.py211 final_outputs, final_state, _ = decoder.dynamic_decode(my_decoder)
214 isinstance(final_outputs, basic_decoder.BasicDecoderOutput))
221 tuple(final_outputs.rnn_output.get_shape().as_list()))
223 tuple(final_outputs.sample_id.get_shape().as_list()))
258 'final_outputs': final_outputs,
311 final_outputs, final_state, _ = decoder.dynamic_decode(my_decoder)
313 isinstance(final_outputs, basic_decoder.BasicDecoderOutput))
314 self.assertEqual(final_outputs.rnn_output.dtype, dtype)
440 final_outputs, final_state, _ = decoder.dynamic_decode(my_decoder)
442 isinstance(final_outputs, basic_decoder.BasicDecoderOutput))
[all …]
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py461 final_outputs = nest.map_structure(lambda ta: ta.stack(), final_outputs_ta)
464 final_outputs, final_state = decoder.finalize(
465 final_outputs, final_state, final_sequence_lengths)
470 final_outputs = nest.map_structure(_transpose_batch_time, final_outputs)
472 return final_outputs, final_state, final_sequence_lengths
/external/tensorflow/tensorflow/python/ops/
Drnn.py883 final_outputs = tuple(ta.stack() for ta in output_final_ta)
885 for output, output_size in zip(final_outputs, flat_output_size):
890 final_outputs = output_final_ta
892 final_outputs = nest.pack_sequence_as(
893 structure=cell.output_size, flat_sequence=final_outputs)
895 final_outputs = nest.map_structure_up_to(
896 cell.output_size, lambda x: array_ops.stack(x, axis=0), final_outputs)
898 return (final_outputs, final_state)
/external/tensorflow/tensorflow/python/keras/
Dbackend.py3592 final_outputs = control_flow_ops.while_loop(
3597 new_states = final_outputs[3:]
3625 final_outputs = control_flow_ops.while_loop(
3629 new_states = final_outputs[2:]
3631 output_ta = final_outputs[1]
/external/tensorflow/tensorflow/contrib/autograph/examples/notebooks/
Ddev_summit_2018_demo.ipynb1166 " final_outputs = rnn_layer(hidden_outputs, upper_cell, batch_size, training)\n",
1170 " sequence_ends = tf.gather_nd(final_outputs, indices)\n",