(self, model: nn.Module, replacements: List[KernelReplacement])
| 517 | """ |
| 518 | |
| 519 | def __init__(self, model: nn.Module, replacements: List[KernelReplacement]): |
| 520 | self.model = model |
| 521 | self.replacements = replacements |
| 522 | self._original_modules: Dict[str, nn.Module] = {} |
| 523 | self._applied: List[str] = [] |
| 524 | |
| 525 | def __enter__(self) -> nn.Module: |
| 526 | for repl in self.replacements: |
nothing calls this directly
no outgoing calls
no test coverage detected