Home
last modified time | relevance | path

Searched refs:replace_sdpa_with_custom_op (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/examples/models/llava/
H A Dexport_llava.py24 replace_sdpa_with_custom_op,
104 source_transforms.append(replace_sdpa_with_custom_op)
H A Dmodel.py18 replace_sdpa_with_custom_op,
65 self.text_model = replace_sdpa_with_custom_op(self.text_model)
/aosp_15_r20/external/executorch/examples/models/llama/
H A Dexport_llama_lib.py68 replace_sdpa_with_custom_op,
1033 transforms.append(replace_sdpa_with_custom_op)
/aosp_15_r20/external/executorch/examples/models/llama/source_transformation/
H A Dsdpa.py101 def replace_sdpa_with_custom_op(module: torch.nn.Module) -> torch.nn.Module: function