projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b8d385c) | patch
[BUG][ConvertLayout] Fix qnn.conv2d layout conversion too many values to unpack ...
authorlhutton1 <35535092+lhutton1@users.noreply.github.com>
Wed, 16 Sep 2020 08:55:13 +0000 (09:55 +0100)
committerGitHub <noreply@github.com>
Wed, 16 Sep 2020 08:55:13 +0000 (17:55 +0900)
commitbdfefbb03f5aab96ee677ee28a166dd6ab5dbf3f
tree2993afe60fda8a6f961b0e6b4878406115361a35tree | snapshot
parentb8d385c2ab741cb254aa700b48546799273276eecommit | diff
[BUG][ConvertLayout] Fix qnn.conv2d layout conversion too many values to unpack (#6442)

This patch follows a previous bugfix in #6419. I made a very simple oversight for qnn.conv2d in that tinfos also contains qnn parameters. Therefore, we need to extract data_info and weight_info differently.

Change-Id: Ib0ad01f427543371380d0bb604a77b5e0ec1103d
python/tvm/relay/qnn/op/layout_conversions.py diff | blob | history
tests/python/relay/test_pass_convert_op_layout.py diff | blob | history
Domain: Machine Learning / ML Framework;