[BYOC][ETHOSN] Introduce further operator support (#6355)
authormbaret <55580676+mbaret@users.noreply.github.com>
Tue, 15 Sep 2020 22:05:19 +0000 (23:05 +0100)
committerGitHub <noreply@github.com>
Tue, 15 Sep 2020 22:05:19 +0000 (15:05 -0700)
commit4c5f0c06137010fb2914a77f34b329bcf4725dcd
treeba8bbc934fceb48d6b9cfd642fed785a05500ab3
parent69e4f44df8961332a8689453b252d57f62d32830
[BYOC][ETHOSN] Introduce further operator support (#6355)

* [BYOC][ETHOSN] Introduce further operator support

This PR introduces support for the following operators:
 - Quantized Fully Connected
 - Quantized Addition
 - Depth-to-space
 - Max/Avg Pool 2D
 - Quantized Relu (Clip)
 - Reshape
 - Quantized Sigmoid

Co-authored-by: Leo Blonk <Leo.Blonk@arm.com>
Co-authored-by: Tristan O'Connor <tristan.oconnor@arm.com>
Co-authored-by: Ramana Radhakrishnan <ramana.radhakrishnan@arm.com>
* Skip tf imports if not available

Change-Id: I11bcf4a78014fa63e7b8e3b0cb00eecfd6cb7760

* ethos -> ethosn

Change-Id: I1fb1a11d0765f6d69f04c24b9c24e08665b8af6a

* Reduce random testing in test_addition

Change-Id: Id06063a0a0cf5f01356df23dc5d4bbbcb47cfa99

* Reduce random testing in test fullyconnected

Change-Id: I330408dfabc4bd804373f100581ce909ff724052

* Fix dumb mistake with rename

Change-Id: I2c5007be485b323116a0e8bab0f9106ea5ec834b

* Added comments to update the hashes in network tests when necessary

Change-Id: I13828c918c959daa492b9ed942a882c86d6690d1

* Fix github name

Change-Id: Idaa70ab9c2ec8db2828d51d15e7c23f28670ec82

* Use black formatting

Change-Id: I538171bd547a16395bef155a1dad28e8b3e347f2

Co-authored-by: Leo Blonk <Leo.Blonk@arm.com>
Co-authored-by: Tristan O'Connor <tristan.oconnor@arm.com>
Co-authored-by: Ramana Radhakrishnan <ramana.radhakrishnan@arm.com>
18 files changed:
python/tvm/relay/op/contrib/ethosn.py
src/relay/backend/contrib/ethosn/codegen.cc
src/relay/backend/contrib/ethosn/codegen_ethosn.h
src/relay/backend/contrib/ethosn/ethosn_api.cc
src/relay/backend/contrib/ethosn/ethosn_api.h
src/runtime/contrib/ethosn/ethosn_runtime.cc
src/runtime/contrib/ethosn/ethosn_runtime.h
tests/python/contrib/test_ethosn/infrastructure.py
tests/python/contrib/test_ethosn/test_addition.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_conv2d.py
tests/python/contrib/test_ethosn/test_depth_to_space.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_fullyconnected.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_networks.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_pooling.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_relu.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_reshape.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_sigmoid.py [new file with mode: 0644]
tests/python/contrib/test_ethosn/test_topologies.py