Update on "[ONNX] Use parameter values in onnx shape inference (#49706)"

Adds an additional run of onnx shape inference after constant folding, since initializer may have changed and affected shape inference.

Differential Revision: [D26050881](https://our.internmc.facebook.com/intern/diff/D26050881)

[ghstack-poisoned]

torch/csrc/jit/passes/onnx/shape_type_inference.h

@@ -1,8 +1,8 @@
 #pragma once
 
 #include <torch/csrc/jit/ir/ir.h>
-#include <torch/csrc/jit/python/python_arg_flatten.h>
 #include <torch/csrc/jit/passes/onnx/helper.h>
+#include <torch/csrc/jit/python/python_arg_flatten.h>
 
 namespace torch {
 namespace jit {