masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1339295756
##
src/relax/transform/alter_op_impl.cc:
##
@@ -314,6 +377,8 @@ class AlterOpImplMutator : public ExprMutator {
Map cache_;
/*! \brief Input IRModule */
const IRM
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1339295756
##
src/relax/transform/alter_op_impl.cc:
##
@@ -314,6 +377,8 @@ class AlterOpImplMutator : public ExprMutator {
Map cache_;
/*! \brief Input IRModule */
const IRM
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1336715785
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -182,7 +182,15 @@ def te_layout_transform(data, name):
)
index_map: tvm.tir.IndexMap = call
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1335797265
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1335452675
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1324923990
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1324919768
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1324914791
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1324325599
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
masahi commented on code in PR #15679:
URL: https://github.com/apache/tvm/pull/15679#discussion_r1323414157
##
python/tvm/relax/transform/legalize_ops/manipulate.py:
##
@@ -205,3 +213,16 @@ def te_layout_transform(data, name):
output_dtype = call_args[0].struct_info.dtype
10 matches
Mail list logo