Searched refs:replace_with_vulkan_rotary_emb (Results 1 – 2 of 2) sorted by relevance
28 def replace_with_vulkan_rotary_emb(module: torch.nn.Module): function34 replace_with_vulkan_rotary_emb(child)
72 from .source_transformation.vulkan_rope import replace_with_vulkan_rotary_emb1084 transforms.append(replace_with_vulkan_rotary_emb)