lization_3d_axis2_epsilon_expanded_function_XU ULayerNormalization_test_layer_normalization_3d_axis2_epsilon_expanded_function_Mean2D