Searched refs:replace_sdpa_with_coreml_sdpa (Results 1 – 2 of 2) sorted by relevance
295 def replace_sdpa_with_coreml_sdpa(module: torch.nn.Module): function304 replace_sdpa_with_coreml_sdpa(child)
67 replace_sdpa_with_coreml_sdpa,1078 transforms.append(replace_sdpa_with_coreml_sdpa)