Searched refs:norm_axis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 862 norm_axis=None, argument
878 if isinstance(norm_axis, list):
879 self.norm_axis = norm_axis[:]
880 elif isinstance(norm_axis, int):
881 self.norm_axis = norm_axis
882 elif norm_axis is None:
883 self.norm_axis = None
886 % type(norm_axis))
914 if self.norm_axis is None:
915 self.norm_axis = list(range(1, ndims))
[all …]
|
| D | normalization_test.py | 510 norm_axis=None, argument
514 if norm_axis is None:
516 elif isinstance(norm_axis, int):
517 if norm_axis < 0:
518 moments_axis = [norm_axis + ndim]
520 moments_axis = [norm_axis]
523 for dim in norm_axis:
541 norm_axis=norm_axis, params_axis=params_axis, dtype=dtype)
573 self.doOutputTest((10, 300), norm_axis=[0])
579 self.doOutputTest((10, 300), norm_axis=2)
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.experimental.-layer-normalization.pbtxt | 89 …argspec: "args=[\'self\', \'norm_axis\', \'params_axis\', \'epsilon\', \'center\', \'scale\', \'be…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.experimental.-layer-normalization.pbtxt | 89 …argspec: "args=[\'self\', \'norm_axis\', \'params_axis\', \'epsilon\', \'center\', \'scale\', \'be…
|