Add ops(max_pool3d and max_pool3d_with_indices) | feat(atenlib)

[fatcat-z]

1. Add max_pool3d and max_pool3d_with_indices ops.
2. Refactor the code so max_pool2d and max_pool3d could share the same helper function. And move them to nn.py file.
3. Refactor the code so max_pool2d_with_indices and max_pool3d_indices could share the same helper function. And move them to nn.py file.
4. Add tests.

[codecov]

Codecov Report

Merging #618 (fc66470) into main (b132176) will increase coverage by 0.04%.
The diff coverage is 95.31%.

@@            Coverage Diff             @@
##             main     #618      +/-   ##
==========================================
+ Coverage   74.49%   74.54%   +0.04%     
==========================================
  Files         107      107              
  Lines       11305    11302       -3     
  Branches     1182     1177       -5     
==========================================
+ Hits         8422     8425       +3     
+ Misses       2561     2557       -4     
+ Partials      322      320       -2     

Impacted Files	Coverage Δ
onnxscript/function_libs/torch_aten/ops/nn.py	75.40% <93.87%> (+0.40%)	⬆️
...ipt/tests/function_libs/torch_aten/extra_opinfo.py	100.00% <100.00%> (ø)
...s/function_libs/torch_aten/ops_correctness_test.py	88.81% <100.00%> (ø)

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

[justinchuby]

Thanks for the explanation! This makes a lot of sense

[justinchuby]

Was it not supported or was there a bug?

Suggested change

	reason="this combinations is not supported.",
	reason="fixme: (what we reported in ORT)",