Searched defs:can_fuse (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/executorch/backends/xnnpack/_passes/ | ||
| H A D | fuse_batch_norm_with_conv.py | 130 def can_fuse( member in FuseBatchNormWithConvPass |
| /aosp_15_r20/external/executorch/backends/transforms/ | ||
| H A D | fuse_batch_norm_with_conv.py | 132 def can_fuse( member in FuseBatchNormWithConvPass |
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ | ||
| H A D | reinplace.py | 270 def can_fuse(): function |
| /aosp_15_r20/external/pytorch/torch/_inductor/ | ||
| H A D | scheduler.py | 1300 def can_fuse(cls, producer: BaseSchedulerNode, consumer: BaseSchedulerNode) -> bool: member in ForeachKernelSchedulerNode |
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ | ||
| H A D | simd.py | 951 def can_fuse(self, node1, node2): member in SIMDScheduling |