Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dnormalization.py143 virtual_batch_size=None, argument
166 virtual_batch_size=virtual_batch_size,
198 virtual_batch_size=None, argument
333 virtual_batch_size=virtual_batch_size,
Dnormalization_test.py1092 inp, virtual_batch_size=-1)
1099 inp, virtual_batch_size=6)
1116 inp, virtual_batch_size=3)
1124 inp, virtual_batch_size=2)
1136 virtual_batch_size = 3
1151 virtual_batch_size=virtual_batch_size)
1153 ghost_shape = ([virtual_batch_size,
1154 shape[0] // virtual_batch_size,
1189 virtual_batch_size = 2
1205 virtual_batch_size=virtual_batch_size)
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py185 virtual_batch_size=None, argument
211 self.virtual_batch_size = virtual_batch_size
260 if self.virtual_batch_size is not None:
322 if self.virtual_batch_size is not None:
323 if self.virtual_batch_size <= 0:
393 if len(axis_to_dim) == 1 and self.virtual_batch_size is None:
401 if self.virtual_batch_size is not None:
743 if self.virtual_batch_size is not None:
750 constant_op.constant([self.virtual_batch_size, -1]),
764 if self.virtual_batch_size is not None:
[all …]
Dnormalization_test.py285 norm = normalization.BatchNormalization(virtual_batch_size=2)
325 norm = normalization_v2.BatchNormalization(virtual_batch_size=2)
353 normalization_v2.BatchNormalization(fused=True, virtual_batch_size=2)
381 norm = normalization_v2.BatchNormalization(virtual_batch_size=8)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.layers.-batch-normalization.pbtxt133 …enorm_clipping\', \'renorm_momentum\', \'fused\', \'trainable\', \'virtual_batch_size\', \'adjustm…
Dtensorflow.layers.-batch-normalization.pbtxt143 …enorm_clipping\', \'renorm_momentum\', \'fused\', \'trainable\', \'virtual_batch_size\', \'adjustm…
Dtensorflow.layers.pbtxt97 …\'renorm\', \'renorm_clipping\', \'renorm_momentum\', \'fused\', \'virtual_batch_size\', \'adjustm…