[RELAY][FUSION] Enhance fusion rule that starts from elemwise and broadcast (#2932)
authorZhi <5145158+zhiics@users.noreply.github.com>
Wed, 1 May 2019 03:42:27 +0000 (11:42 +0800)
committermasahi <masahi129@gmail.com>
Wed, 1 May 2019 03:42:27 +0000 (12:42 +0900)
commitf88f45805de1be82cf9928c0d34c8dd48185cc35
tree90645f1d0feb9f487a73c2d95c75b821bbc4e2db
parent977896cbc9621a34b089cbcbd0e78071c6bcb9d6
[RELAY][FUSION] Enhance fusion rule that starts from elemwise and broadcast (#2932)

* [relay][bugfix] fuse injective to elemwise and broadcast

* enhance fusion for prarllel injectiveOD

* check if tensor in schedule

* fix codegen

* fix lint

* update

* lint
include/tvm/schedule.h
src/relay/backend/compile_engine.cc
src/relay/pass/fuse_ops.cc
src/schedule/schedule_lang.cc
tests/python/relay/test_pass_fuse_ops.py