| from .lazy_ir import ( | |
| generate_non_native_lazy_ir_nodes as generate_non_native_lazy_ir_nodes, | |
| GenLazyIR as GenLazyIR, | |
| GenLazyNativeFuncDefinition as GenLazyNativeFuncDefinition, | |
| GenLazyShapeInferenceDefinition as GenLazyShapeInferenceDefinition, | |
| ) | |
| from .native_functions import ( | |
| compute_native_function_declaration as compute_native_function_declaration, | |
| ) | |
| from .register_dispatch_key import ( | |
| gen_registration_headers as gen_registration_headers, | |
| gen_registration_helpers as gen_registration_helpers, | |
| RegisterDispatchKey as RegisterDispatchKey, | |
| ) | |
| from .ufunc import ( | |
| compute_ufunc_cpu as compute_ufunc_cpu, | |
| compute_ufunc_cpu_kernel as compute_ufunc_cpu_kernel, | |
| compute_ufunc_cuda as compute_ufunc_cuda, | |
| ) | |