Searched refs:_replace_sdpa_with_custom_op (Results 1 – 1 of 1) sorted by relevance
89 def _replace_sdpa_with_custom_op(module: torch.nn.Module): function98 _replace_sdpa_with_custom_op(child)104 _replace_sdpa_with_custom_op(module)