Searched refs:user_defined_kernel_grid_fn_code (Results 1 – 2 of 2) sorted by relevance
562 from torch._inductor.codegen.wrapper import user_defined_kernel_grid_fn_code570 fn_name, code = user_defined_kernel_grid_fn_code(
157 def user_defined_kernel_grid_fn_code( function799 grid_fn, code = user_defined_kernel_grid_fn_code(