[Relay/TOPI][TFLite] Implemented MATRIX_SET_DIAG Operator for Relay/TOPI and TFLite...
authorRishabh Jain <56974688+jainris@users.noreply.github.com>
Thu, 27 Aug 2020 03:21:45 +0000 (08:51 +0530)
committerGitHub <noreply@github.com>
Thu, 27 Aug 2020 03:21:45 +0000 (11:21 +0800)
commit082f27ebf8b14f537f0d7686e8161db1684f3110
treed2c03af41c82fca1c2e731e69ccb2394df50677d
parentf6d3ceecbb5f93a17ad7d5934d770bd9346ff1b9
[Relay/TOPI][TFLite] Implemented MATRIX_SET_DIAG Operator for Relay/TOPI and TFLite Frontend. (#6303)

* Corrected docstring error.

* Minor changes.

* Changed MATRIX_SET_DIAG registration from broadcast to injective.
12 files changed:
include/tvm/topi/transform.h
python/tvm/relay/frontend/tflite.py
python/tvm/relay/op/_transform.py
python/tvm/relay/op/transform.py
python/tvm/topi/testing/__init__.py
python/tvm/topi/testing/matrix_set_diag.py [new file with mode: 0644]
python/tvm/topi/transform.py
src/relay/op/tensor/transform.cc
src/topi/transform.cc
tests/python/frontend/tflite/test_forward.py
tests/python/relay/test_op_level10.py
tests/python/topi/python/test_topi_transform.py