add support for indexing to meshgrid (#62722)
authorMichael Dagitses <mikeyd@fb.com>
Thu, 16 Sep 2021 16:58:09 +0000 (09:58 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 16 Sep 2021 16:59:49 +0000 (09:59 -0700)
commit2c57bbf52189b096ccaa0c4c041703e20812c006
tree657b354dcdf5c1045bbae540b7180dabfdc70873
parent67bd2a31b52b01b83a72bb0c8d1c2ef4d95d103b
add support for indexing to meshgrid (#62722)

Summary:
This is step 3/7 of https://github.com/pytorch/pytorch/issues/50276. It only adds support for the argument but doesn't implement new indexing modes yet.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/62722

Test Plan:
Verified this is not FC breaking by adding logging to both meshgrid
overloads and then called meshgrid twice:

`meshgrid(*tensors)`
  and
`meshgrid(*tensors, indexing='ij')`

This confirmed that the former signature triggered the original native
function and the latter signature triggered the new native function.

Reviewed By: H-Huang

Differential Revision: D30394313

Pulled By: dagitses

fbshipit-source-id: e265cb114d8caae414ee2305dc463b34fdb57fa6
aten/src/ATen/native/TensorShape.cpp
aten/src/ATen/native/native_functions.yaml
test/test_tensor_creation_ops.py
torch/functional.py
torch/onnx/symbolic_opset9.py
torch/testing/_internal/common_methods_invocations.py