Searched defs:GroupBatchFusionBase (Results 1 – 1 of 1) sorted by relevance
102 class GroupBatchFusionBase: class120 def decorator(fusion_cls: GroupBatchFusionBase):1204 rule: GroupBatchFusionBase, root_node: torch.fx.Node, fused_set: Set[torch.fx.Node]1246 def apply_group_batch_fusion(graph: torch.fx.GraphModule, rule: GroupBatchFusionBase):