| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reverse_sequence_op_test.py | 37 seq_axis, argument
44 x, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=seq_lengths)
56 seq_axis, argument
60 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth, True,
62 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth,
86 seq_axis = 0 # permute seq_axis and batch_axis (originally 2 and 0, resp.)
88 self._testBothReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
120 seq_axis = 0
129 seq_axis=seq_axis,
144 seq_axis=1)
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reverse_sequence_op_test.py | 34 seq_axis, argument
43 p, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=lengths)
57 seq_axis=1,
82 seq_axis = 0 # permute seq_axis and batch_axis (originally 2 and 0, resp.)
84 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 392 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
397 seq_axis=seq_axis,
400 return array_ops.reverse(input_, axis=[seq_axis])
408 seq_axis=time_axis,
426 seq_axis=time_axis,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 632 seq_axis, batch_axis = (0, 1)
634 seq_axis, batch_axis = (0, 1) if time_major else (1, 0)
637 init_h = array_ops.expand_dims(init_h, axis=seq_axis)
674 inputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
686 outputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
687 outputs = array_ops.reverse(outputs, axis=[seq_axis])
699 h = array_ops.squeeze(h, axis=seq_axis)
1447 seq_axis, batch_axis = (0, 1)
1449 seq_axis, batch_axis = (0, 1) if time_major else (1, 0)
1452 init_h = array_ops.expand_dims(init_h, axis=seq_axis)
[all …]
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | reverse_sequence.py | 60 seq_axis=parameters["seq_axis"])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 457 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
462 seq_axis=seq_axis,
465 return array_ops.reverse(input_, axis=[seq_axis])
473 seq_axis=time_axis,
491 seq_axis=time_axis,
|
| D | array_ops.py | 4705 seq_axis=None, argument
4750 seq_axis = deprecation.deprecated_argument_lookup("seq_axis", seq_axis,
4757 seq_dim=seq_axis,
4766 seq_axis=None, argument
4812 seq_dim=seq_axis,
|
| D | array_grad.py | 888 seq_axis=op.get_attr("seq_dim"),
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 957 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\'], varargs=None,…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1965 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\', \'seq_dim\', \…
|
| /external/tensorflow/ |
| D | RELEASE.md | 4137 * `tf.reverse_sequence`: `batch_dim` becomes `batch_axis`, `seq_dim` becomes `seq_axis`
|