Function _normalize_struct_info_proxy

python/tvm/relax/script/parser/entry.py:516–524

(annotation)

def _normalize_struct_info_proxy(annotation) -> StructInfoProxy:
    if annotation is None:
        return TupleProxy([])
    elif callable(annotation):
        return annotation()
    elif isinstance(annotation, StructInfoProxy):
        return annotation
    else:
        raise TypeError(f"Expected StructInfoProxy but got {type(annotation)}.")


def _normalize_struct_info(struct_info, dict_globals: dict[str, Any] | None = None) -> StructInfo: