| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reverse_sequence_op_test.py | 36 batch_axis, argument
44 x, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=seq_lengths)
55 batch_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,
87 batch_axis = 2
88 self._testBothReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
122 batch_axis = 2
130 batch_axis=batch_axis,
145 batch_axis=0,
[all …]
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| D | virtual_batchnorm_test.py | 42 vbn = virtual_batchnorm.VBN(reference_batch, batch_axis=1)
49 vbn = virtual_batchnorm.VBN(reference_batch, axis=-1, batch_axis=-2)
73 batch_axis = 0
87 del vb_reduction_axes[batch_axis]
93 vb_mean = array_ops.squeeze(vb_mean, batch_axis)
94 vb_variance = array_ops.squeeze(vb_variance, batch_axis)
115 batch_axis = 0 if axis != 0 else 1 # axis and batch_axis can't same
116 vbn = virtual_batchnorm.VBN(batch, axis, batch_axis=batch_axis)
245 virtual_batchnorm.VBN(array_ops.zeros([1, 2]), name='vbn4', batch_axis=-3)
250 virtual_batchnorm.VBN(array_ops.zeros([1, 2]), name='vbn5', batch_axis=2)
[all …]
|
| D | virtual_batchnorm_impl.py | 41 def _static_or_dynamic_batch_size(tensor, batch_axis): argument
43 batch_size = array_ops.shape(tensor)[batch_axis]
140 batch_axis=0): argument
183 reference_batch, batch_axis)
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 384 batch_axis = 0
387 batch_axis = 1
389 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
395 batch_axis=batch_axis)
406 batch_axis=batch_axis)
424 batch_axis=batch_axis)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reverse_sequence_op_test.py | 33 batch_axis, argument
43 p, batch_axis=batch_axis, seq_axis=seq_axis, seq_lengths=lengths)
56 batch_axis=0,
83 batch_axis = 2
84 self._testReverseSequence(x, batch_axis, seq_axis, seq_lengths, truth)
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | sugar_test.py | 59 self.batch_axis = ('batch', range(self.batch_size))
66 self.batch_axis, self.row_axis, self.column_axis, self.channel_axis,
83 self.batch_axis, self.row_axis, self.column_axis,
98 self.batch_axis, self.row_axis, self.column_axis, depth_axis,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 448 batch_axis = 0
451 batch_axis = 1
453 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
457 seq_axis=seq_axis, batch_axis=batch_axis)
468 batch_axis=batch_axis)
479 seq_axis=time_axis, batch_axis=batch_axis)
|
| D | array_ops.py | 3245 batch_axis=None, argument
3251 batch_axis = deprecation.deprecated_argument_lookup("batch_axis", batch_axis,
3257 batch_dim=batch_axis,
3269 input, seq_lengths, seq_axis=None, batch_axis=None, name=None): argument
3274 batch_dim=batch_axis,
|
| D | array_grad.py | 632 batch_axis=op.get_attr("batch_dim"),
|
| /external/tensorflow/tensorflow/contrib/gan/python/estimator/python/ |
| D | tpu_gan_estimator_impl.py | 104 batch_axis=None, argument
258 batch_axis=batch_axis,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 775 ctx, input_fn, inputs_structure_recorder, batch_axis, device, host_id): argument
792 if batch_axis is not None:
830 shard_dimensions=batch_axis)
1190 def __init__(self, input_fn, batch_axis, ctx): argument
1210 self._batch_axis = batch_axis
2228 batch_axis=None, argument
2339 model_function = self._augment_model_fn(model_fn, batch_axis)
2766 def _augment_model_fn(self, model_fn, batch_axis): argument
2826 input_holders = _InputPipeline(input_fn, batch_axis, ctx)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 885 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\'], varargs=None,…
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 4351 batch_axis=parameters["batch_axis"],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1881 …argspec: "args=[\'input\', \'seq_lengths\', \'seq_axis\', \'batch_axis\', \'name\', \'seq_dim\', \…
|
| /external/tensorflow/ |
| D | RELEASE.md | 1464 * `tf.reverse_sequence`: `batch_dim` becomes `batch_axis`, `seq_dim` becomes `seq_axis`
|