Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dreverse_sequence_op_test.py37 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)
121 seq_axis = 0
131 seq_axis=seq_axis,
146 seq_axis=1)
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dreverse_sequence_op_test.py34 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/
Drnn.py389 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
394 seq_axis=seq_axis,
397 return array_ops.reverse(input_, axis=[seq_axis])
405 seq_axis=time_axis,
423 seq_axis=time_axis,
/external/tensorflow/tensorflow/python/ops/
Drnn.py453 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
457 seq_axis=seq_axis, batch_axis=batch_axis)
459 return array_ops.reverse(input_, axis=[seq_axis])
467 seq_axis=time_axis,
479 seq_axis=time_axis, batch_axis=batch_axis)
Darray_ops.py3244 seq_axis=None, argument
3249 seq_axis = deprecation.deprecated_argument_lookup("seq_axis", seq_axis,
3256 seq_dim=seq_axis,
3269 input, seq_lengths, seq_axis=None, batch_axis=None, name=None): argument
3273 seq_dim=seq_axis,
Darray_grad.py633 seq_axis=op.get_attr("seq_dim"),
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt885 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\'], varargs=None,…
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py4352 seq_axis=parameters["seq_axis"])
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1881 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\', \'seq_dim\', \…
/external/tensorflow/
DRELEASE.md1464 * `tf.reverse_sequence`: `batch_dim` becomes `batch_axis`, `seq_dim` becomes `seq_axis`