| 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 |