Searched refs:begin_norm_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 2207 begin_norm_axis=1, argument
2278 if begin_norm_axis < 0:
2279 begin_norm_axis = inputs_rank + begin_norm_axis
2280 if begin_params_axis >= inputs_rank or begin_norm_axis >= inputs_rank:
2283 (begin_params_axis, begin_norm_axis, inputs_rank))
2312 norm_axes = list(range(begin_norm_axis, inputs_rank))
|
| D | layers_test.py | 2873 begin_norm_axis=1, argument
2876 expected_mean = np.zeros(input_shape[:begin_norm_axis])
2877 expected_var_uncorrected = np.ones(input_shape[:begin_norm_axis])
2894 inputs, begin_norm_axis=begin_norm_axis, scope='LN')
2899 if begin_norm_axis < 0:
2900 begin_norm_axis = len(input_shape) + begin_norm_axis
2901 moments_axis = tuple(range(begin_norm_axis, len(input_shape)))
2930 self.doOutputTest((10, 300), begin_norm_axis=2)
2938 (100, 10, 10, 3), begin_norm_axis=3, tol=1e-4, dtype=dtypes.float64)
2940 (100, 10, 10, 3), begin_norm_axis=-1, tol=1e-4, dtype=dtypes.float64)
[all …]
|