[Relay/TOPI][OP] Add meshgrid op in Relay, TOPI, Pytorch frontend (#5961)
authorTrevor Morris <trevmorr@amazon.com>
Thu, 2 Jul 2020 02:14:33 +0000 (19:14 -0700)
committerGitHub <noreply@github.com>
Thu, 2 Jul 2020 02:14:33 +0000 (11:14 +0900)
commitaa035f4650926f5e714b02cbab6d974f0a17352f
tree8af23c4b8061cebe61673d4334133772fc390d41
parenta4ebf927e9e3b5a74a58c29a0d760c0ddfc10986
[Relay/TOPI][OP] Add meshgrid op in Relay, TOPI, Pytorch frontend (#5961)

* Add meshgrid op with pytorch importer

* Fix c++ lint

* Fix pylint

* Meshgrid: add scalar test for pytorch, add topi python wrapper

* Add indexing mode attr.

* Add MeshgridAttrs python binding

* c++ lint
13 files changed:
docs/api/python/topi.rst
docs/langref/relay_op.rst
include/tvm/relay/attrs/transform.h
python/tvm/relay/frontend/pytorch.py
python/tvm/relay/op/_transform.py
python/tvm/relay/op/op_attrs.py
python/tvm/relay/op/transform.py
src/relay/op/tensor/transform.cc
tests/python/frontend/pytorch/test_forward.py
tests/python/relay/test_op_level3.py
topi/include/topi/transform.h
topi/python/topi/transform.py
topi/src/transform.cc