Raise ValueError for out-of-bounds axis in LayerNormalization.build() (#23329)