Searched refs:moments_axes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | normalization.py | 114 moments_axes = list(range(inputs_rank))
115 del moments_axes[reduction_axis]
116 del moments_axes[0]
155 mean, variance = nn.moments(inputs, moments_axes, keep_dims=True)
314 moments_axes = [channels_axis + 1]
317 moments_axes.append(a + 1)
319 moments_axes.append(a)
361 inputs, moments_axes, keep_dims=True)
365 mean, variance = nn.moments(inputs, moments_axes, keep_dims=True)
|
| D | layers.py | 690 moments_axes = [0] + list(range(2, inputs_rank))
698 moments_axes = list(range(inputs_rank - 1))
774 mean, variance = nn.moments(inputs, moments_axes, keep_dims=True)
778 mean, variance = nn.moments(inputs, moments_axes)
782 inputs, moments_axes, batch_weights, keepdims=True)
786 mean, variance = nn.weighted_moments(inputs, moments_axes,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_test.py | 1052 moments_axes, argument
1060 input_values, axis=moments_axes, keepdims=keep_dims)
1062 input_values, axis=moments_axes, keepdims=keep_dims)
1068 inputs, moments_axes, keepdims=keep_dims)
|