from .lazy_ir import LazyIR as LazyIR | |
from .lazy_ir import GenLazyShapeInferenceDefinition as GenLazyShapeInferenceDefinition | |
from .lazy_ir import GenLazyNativeFuncDefinition as GenLazyNativeFuncDefinition | |
from .register_dispatch_key import ( | |
RegisterDispatchKey as RegisterDispatchKey, | |
gen_registration_helpers as gen_registration_helpers, | |
gen_registration_headers as gen_registration_headers, | |
) | |
from .native_functions import compute_native_function_declaration as compute_native_function_declaration |