Searched defs:batch_axis (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reverse_sequence_op_test.py | 36 batch_axis, argument
55 batch_axis, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reverse_sequence_op_test.py | 33 batch_axis, argument
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 392 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 457 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| D | array_ops.py | 4706 batch_axis=None, argument
4767 batch_axis=None, argument
|