def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["device"] = ["cuda", "cpu"]
    arg_dict["quantization_formula"] = ["google"]
    arg_dict["quantization_bit"] = [4, 8]
    arg_dict["quantization_scheme"] = ["symmetric", "affine"]
    arg_dict["weight_quant_per_layer"] = [True, False]
    arg_dict["input_quant_momentum"] = [0.95]
    for i in range(5):
        for arg in GenArgList(arg_dict):
            _test_qat_conv2d(test_case, *arg)