click.option("--input-shape", type=str, default="1024,1024")

pytest.mark.parametrize("input_type", ["zeros", "ones", "rand01", "randn"])