manupa-arm commented on a change in pull request #6655:
URL: https://github.com/apache/incubator-tvm/pull/6655#discussion_r510795269
##
File path: python/tvm/relay/transform/transform.py
##
@@ -666,7 +666,7 @@ def PartitionGraph():
return _ffi_api.PartitionGraph()
-def
manupa-arm commented on a change in pull request #6655:
URL: https://github.com/apache/incubator-tvm/pull/6655#discussion_r510715272
##
File path: python/tvm/relay/transform/transform.py
##
@@ -666,7 +666,7 @@ def PartitionGraph():
return _ffi_api.PartitionGraph()
-def
manupa-arm commented on a change in pull request #6655:
URL: https://github.com/apache/incubator-tvm/pull/6655#discussion_r504052671
##
File path: tests/python/relay/test_pass_annotate_target.py
##
@@ -327,6 +327,42 @@ def after():
assert tvm.ir.structural_equal(expected,
manupa-arm commented on a change in pull request #6655:
URL: https://github.com/apache/incubator-tvm/pull/6655#discussion_r503308197
##
File path: src/relay/transforms/annotate_target.cc
##
@@ -62,19 +65,22 @@ class AnnotateTargetRewriter : public ExprRewriter {
manupa-arm commented on a change in pull request #6655:
URL: https://github.com/apache/incubator-tvm/pull/6655#discussion_r503299324
##
File path: src/relay/transforms/annotate_target.cc
##
@@ -40,11 +40,14 @@ static const PackedFunc* make_begin_op =
static const PackedFunc* m