Searched refs:params_axis (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 863 params_axis=-1, argument
888 if isinstance(params_axis, list):
889 self.params_axis = params_axis[:]
890 elif isinstance(params_axis, int):
891 self.params_axis = params_axis
894 % type(params_axis))
924 if isinstance(self.params_axis, int):
925 self.params_axis = [self.params_axis]
926 for idx, x in enumerate(self.params_axis):
928 self.params_axis[idx] = ndims + x
[all …]
|
| D | normalization_test.py | 462 input_shape=(3, 4, 4), params_axis=1)
511 params_axis=-1, argument
541 norm_axis=norm_axis, params_axis=params_axis, dtype=dtype)
574 self.doOutputTest((10, 300), params_axis=[0, 1])
604 self.doOutputTest((10, 10, 10, 30), params_axis=[-2, -1])
606 params_axis=[-3, -2, -1])
613 params_axis=[-2, -1])
|
| /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…
|