Searched refs:replace_rms_norm_with_native_rms_norm (Results 1 – 2 of 2) sorted by relevance
11 def replace_rms_norm_with_native_rms_norm(module: torch.nn.Module): function22 replace_rms_norm_with_native_rms_norm(child)
60 from .source_transformation.rms_norm import replace_rms_norm_with_native_rms_norm1053 transforms.append(replace_rms_norm_with_native_rms_norm)1060 transforms.append(replace_rms_norm_with_native_rms_norm)