def METHOD_NAME(test_case):
    _test_layer_norm(
        test_case, shape=[32, 1024], normalized_shape=[1024], dtype=flow.float16,
    )
    _test_layer_norm(test_case, shape=[16, 512], normalized_shape=[512])
    _test_layer_norm(test_case, shape=[15, 512], normalized_shape=[512])
    _test_layer_norm(test_case, shape=[16, 511], normalized_shape=[511])
    _test_layer_norm(test_case, shape=[13, 499], normalized_shape=[499])