Home
last modified time | relevance | path

Searched refs:_transpose_batch_time (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
Dbeam_search_ops_test.py31 def _transpose_batch_time(x): function
40 step_ids = _transpose_batch_time(
42 parent_ids = _transpose_batch_time(
45 expected_result = _transpose_batch_time([[[2, 2, 2], [6, 5, 6], [7, 8, 9],
59 step_ids = _transpose_batch_time(
61 parent_ids = _transpose_batch_time(
82 step_ids = _transpose_batch_time(
84 parent_ids = _transpose_batch_time(
87 expected_result = _transpose_batch_time([[[2, -1, 2], [6, 5, 6], [7, 8, 9],
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py36 from tensorflow.python.ops.rnn import _transpose_batch_time
215 flat_input = tuple(_transpose_batch_time(input_) for input_ in flat_input)
273 outputs = nest.map_structure(_transpose_batch_time, outputs)
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py45 _transpose_batch_time = rnn._transpose_batch_time # pylint: disable=protected-access variable
470 final_outputs = nest.map_structure(_transpose_batch_time, final_outputs)
Dsampler.py49 _transpose_batch_time = decoder._transpose_batch_time # pylint: disable=protected-access variable
234 inputs = nest.map_structure(_transpose_batch_time, inputs)
423 auxiliary_inputs = nest.map_structure(_transpose_batch_time,
Dhelper.py50 _transpose_batch_time = decoder._transpose_batch_time # pylint: disable=protected-access variable
250 inputs = nest.map_structure(_transpose_batch_time, inputs)
467 _transpose_batch_time, auxiliary_inputs)
/external/tensorflow/tensorflow/contrib/rnn/
D__init__.py96 from tensorflow.python.ops.rnn import _transpose_batch_time as transpose_batch_time
/external/tensorflow/tensorflow/python/ops/
Drnn.py45 def _transpose_batch_time(x): function
622 flat_input = tuple(_transpose_batch_time(input_) for input_ in flat_input)
678 outputs = nest.map_structure(_transpose_batch_time, outputs)