def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["enable_quantize"] = [True, False]
    arg_dict["fp16"] = [True, False]
    arg_dict["embedding_size"] = [128, 17]
    for kwargs in GenArgDict(arg_dict):
        _test_embedding_gradient_shuffle(test_case, **kwargs)