Searched refs:enforce_output_layout (Results 1 – 1 of 1) sorted by relevance
187 def enforce_output_layout(gm: torch.fx.GraphModule): function269 enforce_output_layout(gm)