Searched defs:batch_axis (Results 1 – 8 of 8) 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/contrib/gan/python/features/python/ |
| D | virtual_batchnorm_impl.py | 41 def _static_or_dynamic_batch_size(tensor, batch_axis): argument
140 batch_axis=0): 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 | 389 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 104 batch_axis=None, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 453 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| D | array_ops.py | 3245 batch_axis=None, argument
3269 input, seq_lengths, seq_axis=None, batch_axis=None, name=None): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 775 ctx, input_fn, inputs_structure_recorder, batch_axis, device, host_id): argument
1190 def __init__(self, input_fn, batch_axis, ctx): argument
2228 batch_axis=None, argument
2766 def _augment_model_fn(self, model_fn, batch_axis): argument
|