| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | normalization_test.py | 115 def doOutputTest(self, input_shape, data_format, tol=1e-3): member in InstanceNormTest
146 self.doOutputTest((10, 10, 10, 30), 'NHWC', tol=1e-2)
149 self.doOutputTest((10, 10, 10, 30), 'NCHW', tol=1e-2)
152 self.doOutputTest((1, 100, 100, 1), 'NHWC', tol=1e-3)
155 self.doOutputTest((1, 100, 100, 1), 'NCHW', tol=1e-3)
158 self.doOutputTest((10, 10, 10, 10, 30), 'NHWC', tol=1e-2)
161 self.doOutputTest((10, 10, 10, 10, 30), 'NCHW', tol=1e-2)
164 self.doOutputTest((1, 100, 100, 1, 1), 'NHWC', tol=1e-3)
167 self.doOutputTest((1, 100, 100, 1, 1), 'NCHW', tol=1e-3)
305 def doOutputTest(self, member in GroupNormTest
[all …]
|
| D | layers_test.py | 2870 def doOutputTest(self, member in LayerNormTest
2926 self.doOutputTest((10, 300))
2930 self.doOutputTest((10, 300), begin_norm_axis=2)
2933 self.doOutputTest((100, 10, 10, 3))
2937 self.doOutputTest(
2939 self.doOutputTest(
2943 self.doOutputTest((10, 10, 10, 30))
2946 self.doOutputTest((10, 10, 10, 30), begin_norm_axis=3)
2949 self.doOutputTest((1, 100, 100, 1))
2952 self.doOutputTest((1, 100, 1000, 1), tol=1e-4, dtype=dtypes.float32)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization_test.py | 507 def doOutputTest(self, member in LayerNormalizationTest
572 self.doOutputTest((10, 300))
573 self.doOutputTest((10, 300), norm_axis=[0])
574 self.doOutputTest((10, 300), params_axis=[0, 1])
579 self.doOutputTest((10, 300), norm_axis=2)
583 self.doOutputTest((100, 10, 10, 3))
589 self.doOutputTest(
591 self.doOutputTest(shape, norm_axis=-1, tol=1e-4, dtype='float64')
595 self.doOutputTest((10, 10, 10, 30))
599 self.doOutputTest((10, 10, 10, 30), norm_axis=3)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_test.py | 1050 def doOutputTest(self, member in MomentsTest
1087 self.doOutputTest((10, 10), (0,), check_gradients=True)
1090 self.doOutputTest((10, 10), (0, 1), check_gradients=True)
1093 self.doOutputTest((10, 300), (0,))
1096 self.doOutputTest((10, 300), (1,))
1099 self.doOutputTest((10, 300), (0, 1))
1102 self.doOutputTest((10, 10, 10, 30), (0,))
1105 self.doOutputTest((10, 10, 10, 30), (1,))
1108 self.doOutputTest((10, 10, 10, 30), (3,))
1111 self.doOutputTest((10, 10, 10, 30), (0, 1, 2))
[all …]
|