Searched defs:SchedulerNode (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cuda_combined_scheduling.py | 79 def codegen_node(self, node: Union[FusedSchedulerNode, SchedulerNode]): argument
|
| H A D | cpp.py | 366 outer_fused_nodes: List[Union[FusedSchedulerNode, SchedulerNode]], argument
3474 def is_lowp_fp_scheduler(self, scheduler_node: SchedulerNode):
3821 def codegen_nodes(self, nodes: List[SchedulerNode]):
4158 def try_loop_split(self, nodes: List[SchedulerNode]):
4456 node: Union[OuterLoopFusedSchedulerNode, FusedSchedulerNode, SchedulerNode], argument
|
| H A D | simd.py | 1137 self, node: Union[scheduler.FusedSchedulerNode, scheduler.SchedulerNode]
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | scheduler.py | 825 class SchedulerNode(BaseSchedulerNode): class
3646 def codegen_node(self, node: Union[FusedSchedulerNode, SchedulerNode]) -> None: argument
3699 def debug_triton_code(node: Union[SchedulerNode, FusedSchedulerNode]) -> List[str]:
|