def METHOD_NAME(test_case):
    default_shape = (256, 256)
    for device in ["cpu", "cuda"]:
        for check_func in check_func_list:
            tensor = flow.empty(*default_shape, device=flow.device(device))
            check_func["func"](tensor, **check_func["params"])
            try:
                check_func["checker"](test_case, tensor)
            except AssertionError as e:
                print(
                    f"Failed: {check_func['func'].__name__} {check_func['params']}"
                )
                raise e