[Relay][heterogeneous pass] remove on_device op after annotation (#3204)
authorZhi <5145158+zhiics@users.noreply.github.com>
Tue, 21 May 2019 19:53:58 +0000 (12:53 -0700)
committerJared Roesch <roeschinc@gmail.com>
Tue, 21 May 2019 19:53:58 +0000 (12:53 -0700)
commit21935dcbf56ad3bd66ebff9891a6bc3865b8106d
tree4f4ae93b9e2dfc7bfc547c5237b9ff5574a1e04e
parent91b181d4a4b6c4e81c83565e0777f9ac1e29feb7
[Relay][heterogeneous pass] remove on_device op after annotation (#3204)

* remove on_device op after annotation

* Update src/relay/pass/device_annotation.cc

Co-Authored-By: MORINAGA <34588258+imorinaga@users.noreply.github.com>
src/relay/pass/device_annotation.cc
tests/python/relay/test_pass_annotation.py