Avoid torch.compile graph-break in functional normalize fn#9314

lw wants to merge 2 commits intomainpytorch/vision:mainfrom

lw_normalize_graph_breakpytorch/vision:lw_normalize_graph_breakCopy head branch name to clipboard