[Shape Fns] Fix optional None for the actual shape functions

[Gamrix]

I think the person who edited mean.dim edited the python and the associated Torchscript version manually in two different ways. This diff fixes that all up. It also fixes the inconsistencies in the torch.nonzero generated file

Stack from ghstack:

• Editing gen_jit_shape_functions to make lintrunner happy #79571
• -> [Shape Fns] Fix optional None for the actual shape functions #83092
• [SSA] Clear Shape Cache on Load Error #82850

[facebook-github-bot]

🔗 Helpful links

• 🧪  See artifacts and rendered test results at hud.pytorch.org/pr/83092
• 📄  Preview Python docs built from this PR
• 📄  Preview C++ docs built from this PR
• ❓Need help or want to give feedback on the CI? Visit our office hours
• ↩️  [fb-only] Re-run with SSH instructions

❌ 1 New Failures

As of commit 1d63f53 (more details on the Dr. CI page):

Expand to see more

---

• 1/1 failures introduced in this PR

---

🕵️ 1 new failure recognized by patterns

The following CI failures do not appear to be due to upstream breakages

pull / linux-focal-py3.7-clang10-onnx / test (default, 1, 2, linux.2xlarge) (1/1)

Step: "Test" (full log | diagnosis details)

2022-08-09T21:17:14.2783405Z TypeError: cannot unpack non-iterable torch._C.Value object

2022-08-09T21:17:14.2776097Z   File "/opt/conda/lib/python3.7/site-packages/torch/onnx/utils.py", line 1062, in _model_to_graph
2022-08-09T21:17:14.2776731Z     module=module,
2022-08-09T21:17:14.2777489Z   File "/opt/conda/lib/python3.7/site-packages/torch/onnx/utils.py", line 624, in _optimize_graph
2022-08-09T21:17:14.2778150Z     graph = _C._jit_pass_onnx(graph, operator_export_type)
2022-08-09T21:17:14.2778995Z   File "/opt/conda/lib/python3.7/site-packages/torch/onnx/utils.py", line 1744, in _run_symbolic_function
2022-08-09T21:17:14.2779525Z     return symbolic_fn(g, *inputs, **attrs)
2022-08-09T21:17:14.2780466Z   File "/opt/conda/lib/python3.7/site-packages/torch/onnx/symbolic_helper.py", line 249, in wrapper
2022-08-09T21:17:14.2780955Z     return fn(g, *args, **kwargs)
2022-08-09T21:17:14.2781772Z   File "/opt/conda/lib/python3.7/site-packages/torch/onnx/symbolic_opset9.py", line 2263, in layer_norm
2022-08-09T21:17:14.2782513Z     g, input, normalized_shape, weight, bias, eps, cudnn_enable, False
2022-08-09T21:17:14.2783405Z TypeError: cannot unpack non-iterable torch._C.Value object 
2022-08-09T21:17:14.2784059Z (Occurred when translating layer_norm).
2022-08-09T21:17:14.2784353Z 
2022-08-09T21:17:14.2784554Z 
2022-08-09T21:17:14.2785768Z 
2022-08-09T21:17:14.2785780Z 
2022-08-09T21:17:14.2787510Z [gw0] �[31mFAILED�[0m test/onnx/test_pytorch_onnx_no_runtime.py �[1A
2022-08-09T21:17:14.2947255Z  �[36mtest/onnx/test_pytorch_onnx_no_runtime.py�[0m::TestONNXExport.test_source_range_propagation�[0m �[31m⨯�[0m�[31m99% �[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[32m█�[0m�[40m�[31m▉�[0m�[1B
2022-08-09T21:17:14.2948211Z 
2022-08-09T21:17:14.2948379Z 
2022-08-09T21:17:14.2948385Z 

---

This comment was automatically generated by Dr. CI (expand for details).

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.