[RELAY][BYOC] Preserve type information in Merge Composite (#5640)
authormbaret <55580676+mbaret@users.noreply.github.com>
Fri, 22 May 2020 02:24:23 +0000 (03:24 +0100)
committerGitHub <noreply@github.com>
Fri, 22 May 2020 02:24:23 +0000 (11:24 +0900)
commit0884659eb8c5fe51cc4cac9f2f8b6400f47fdee6
treed24179b53fbcdf1851d9e0bc5a73b6373b8915b2
parentc365c2aa874c02b82cea89b9c9c2b8bb9d4a5083
[RELAY][BYOC] Preserve type information in Merge Composite (#5640)

Keep the type information when extracting patterns
so that it can be used as part of 'check' functions.

Change-Id: I16cc70c3d013a794d2ceefb5bec815129c7b8825
src/relay/transforms/merge_composite.cc
tests/python/relay/test_pass_merge_composite.py