1from torchgen.dest.lazy_ir import ( 2 generate_non_native_lazy_ir_nodes as generate_non_native_lazy_ir_nodes, 3 GenLazyIR as GenLazyIR, 4 GenLazyNativeFuncDefinition as GenLazyNativeFuncDefinition, 5 GenLazyShapeInferenceDefinition as GenLazyShapeInferenceDefinition, 6) 7from torchgen.dest.native_functions import ( 8 compute_native_function_declaration as compute_native_function_declaration, 9) 10from torchgen.dest.register_dispatch_key import ( 11 gen_registration_headers as gen_registration_headers, 12 gen_registration_helpers as gen_registration_helpers, 13 RegisterDispatchKey as RegisterDispatchKey, 14) 15from torchgen.dest.ufunc import ( 16 compute_ufunc_cpu as compute_ufunc_cpu, 17 compute_ufunc_cpu_kernel as compute_ufunc_cpu_kernel, 18 compute_ufunc_cuda as compute_ufunc_cuda, 19) 20