| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | reverse_sequence_op_test.py | 35 batch_axis, argument
43 x, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=seq_lengths)
57 batch_axis, argument
62 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth, True,
64 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth,
95 batch_axis = 2
96 self._testBothReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
132 batch_axis = 2
139 batch_axis=batch_axis,
154 batch_axis=0,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reverse_sequence_op_test.py | 29 batch_axis, argument
39 p, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=lengths)
52 batch_axis=0,
79 batch_axis = 2
80 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | recurrent_v2.py | 630 seq_axis, batch_axis = (0, 1)
632 seq_axis, batch_axis = (0, 1) if time_major else (1, 0)
672 inputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
684 outputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
1446 seq_axis, batch_axis = (0, 1)
1448 seq_axis, batch_axis = (0, 1) if time_major else (1, 0)
1490 inputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
1502 outputs, sequence_lengths, seq_axis=seq_axis, batch_axis=batch_axis)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 451 batch_axis = 0
454 batch_axis = 1
456 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
462 batch_axis=batch_axis)
473 batch_axis=batch_axis)
491 batch_axis=batch_axis)
|
| D | array_ops.py | 4982 batch_axis=None, argument
5028 batch_axis = deprecation.deprecated_argument_lookup("batch_axis", batch_axis,
5034 batch_dim=batch_axis,
5043 batch_axis=None, argument
5089 batch_dim=batch_axis,
|
| D | array_grad.py | 913 batch_axis=op.get_attr("batch_dim"),
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | reverse_sequence.py | 55 batch_axis=parameters["batch_axis"],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.estimator.tpu.-t-p-u-estimator.pbtxt | 24 …, \'train_batch_size\', \'eval_batch_size\', \'predict_batch_size\', \'batch_axis\', \'eval_on_tpu…
|
| D | tensorflow.pbtxt | 1973 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\', \'seq_dim\', \…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 965 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\'], varargs=None,…
|
| /external/tensorflow/ |
| D | RELEASE.md | 9455 becomes `axis` * `tf.reverse_sequence`: `batch_dim` becomes `batch_axis`,
|