Searched refs:custom_annotate_llama_matmul_16a8w (Results 1 – 2 of 2) sorted by relevance
151 custom_annotate_llama_matmul_16a8w,199 custom_annotations = (custom_annotate_llama_matmul_16a8w,)
105 def custom_annotate_llama_matmul_16a8w(gm: torch.fx.GraphModule) -> None: # noqa: C901 function