[Relay][AlterOp] Improving support for broadcast layout alteration. (#4040)
authorAnimesh Jain <anijain@umich.edu>
Sun, 6 Oct 2019 04:18:58 +0000 (21:18 -0700)
committerYizhi Liu <liuyizhi@apache.org>
Sun, 6 Oct 2019 04:18:58 +0000 (12:18 +0800)
commitd703fb4e1f58ae56475defb52a6cea8cbd67f662
tree7c563aa9b781ad6f1828168d1d829435f2f07150
parent260d660ac0bf2cf35e38f5bdffc3c70584a5ea6f
[Relay][AlterOp] Improving support for broadcast layout alteration. (#4040)
include/tvm/data_layout.h
src/relay/op/tensor/transform.cc
src/relay/pass/alter_op_layout.cc
src/relay/pass/alter_op_layout.h
src/relay/pass/pattern_util.h
src/relay/qnn/op/convolution.cc
tests/python/relay/test_op_qnn_conv2d.py
tests/python/relay/test_pass_alter_op_layout.py