Home
last modified time | relevance | path

Searched refs:virtual_batch_size (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/layers/
Dnormalization.py126 virtual_batch_size=None, argument
149 virtual_batch_size=virtual_batch_size,
183 virtual_batch_size=None, argument
307 virtual_batch_size=virtual_batch_size,
Dnormalization_test.py1043 inp, virtual_batch_size=-1)
1050 inp, virtual_batch_size=6)
1067 inp, virtual_batch_size=3)
1075 inp, virtual_batch_size=2)
1087 virtual_batch_size = 3
1102 virtual_batch_size=virtual_batch_size)
1104 ghost_shape = ([virtual_batch_size,
1105 shape[0] // virtual_batch_size,
1140 virtual_batch_size = 2
1156 virtual_batch_size=virtual_batch_size)
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py155 virtual_batch_size=None, argument
182 self.virtual_batch_size = virtual_batch_size
225 if self.virtual_batch_size is not None:
268 if self.virtual_batch_size is not None:
269 if self.virtual_batch_size <= 0:
316 if len(axis_to_dim) == 1 and self.virtual_batch_size is None:
323 if self.virtual_batch_size is not None:
573 if self.virtual_batch_size is not None:
577 expanded_shape = [self.virtual_batch_size, -1] + original_shape[1:]
588 if self.virtual_batch_size is not None:
[all …]
Dnormalization_test.py174 norm = normalization.BatchNormalization(virtual_batch_size=2)
208 norm = normalization_v2.BatchNormalization(virtual_batch_size=2)
236 normalization_v2.BatchNormalization(fused=True, virtual_batch_size=2)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dbatch_normalization.py200 if layer.virtual_batch_size:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.layers.-batch-normalization.pbtxt90 …enorm_clipping\', \'renorm_momentum\', \'fused\', \'trainable\', \'virtual_batch_size\', \'adjustm…
Dtensorflow.layers.-batch-normalization.pbtxt100 …enorm_clipping\', \'renorm_momentum\', \'fused\', \'trainable\', \'virtual_batch_size\', \'adjustm…
Dtensorflow.layers.pbtxt97 …\'renorm\', \'renorm_clipping\', \'renorm_momentum\', \'fused\', \'virtual_batch_size\', \'adjustm…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.layers.-batch-normalization.pbtxt90 …enorm_clipping\', \'renorm_momentum\', \'fused\', \'trainable\', \'virtual_batch_size\', \'adjustm…