Function _shape_tuple_func

python/tvm/exec/disco_worker.py:59–61

(x: Shape)

@register_global_func("tests.disco.shape_tuple", override=True)
def _shape_tuple_func(x: Shape):
    assert isinstance(x, Shape)
    return Shape(list(x) + [4, 5])


@register_global_func("tests.disco.test_callback", override=True)