def METHOD_NAME(test_case):
    input = torch.ones(16, 128, 28, 28)
    running_mean = torch.randn(128)
    running_var = torch.randn(128)
    weight = torch.randn(128)
    bias = torch.randn(128)
    torch.nn.functional.batch_norm(
        input, running_mean, running_var, weight, bias, True
    )
    torch.nn.functional.batch_norm(
        input, running_mean, running_var, weight, bias, False
    )
    torch.nn.functional.batch_norm(
        input, running_mean, running_var, None, None, True
    )
    torch.nn.functional.batch_norm(
        input, running_mean, running_var, None, None, False
    )