Searched defs:pre_grad (Results 1 – 1 of 1) sorted by relevance
119 def register_fusion(name: str, pre_grad=True): argument130 def list_group_batch_fusions(pre_grad=True) -> List[str]: argument1271 def generate_fusion_from_config(config_options: Dict[str, Any], pre_grad=True): argument1284 def group_batch_fusion_passes(graph: torch.fx.Graph, pre_grad=True): argument