Hello @masahi
I have tried using your TraceWrapper and it works on
**BertForSequenceClassification** model.
Thanks :)
---
[Visit
Topic](https://discuss.tvm.apache.org/t/how-to-deal-with-prim-dictconstruct/11978/6)
to respond.
You are receiving this because you enabled mailing list mod
The problem raised by @KZamudio is solved. The 1 in the tensor shape [a, **1**,
b, c] corresponds to the depthwise convolution (convolution is only computed on
1 channel at a time). The solution was to transform the depthwise convolution
into a grouped convolution with groups of size 16 as VTA
Thanks for asking. Some AWS teams may have some dependencies. I'll inform them
to comment on this thread if they have any concern.
---
[Visit
Topic](https://discuss.tvm.apache.org/t/byoc-how-backwards-compatible-does-the-tensorrt-partition-for-tensorrt-function-need-to-be/12957/3)
to resp
You can try something like
https://github.com/masahi/tvm-cutlass-eval/blob/master/bert/export.py#L26-L33
---
[Visit
Topic](https://discuss.tvm.apache.org/t/how-to-deal-with-prim-dictconstruct/11978/5)
to respond.
You are receiving this because you enabled mailing list mode.
To unsubscri
Hello @masahi
I have asked my questions and found out solution by myself in
[quote="masahi, post:2, topic:11978"]
Issue: Converting model from pytorch to relay model - #5 by popojames
[/quote]
I am facing prim::DictConstruct issue again. because I am customizing BERT
model with BERT config
cc @comaniac @Laurawly
---
[Visit
Topic](https://discuss.tvm.apache.org/t/byoc-how-backwards-compatible-does-the-tensorrt-partition-for-tensorrt-function-need-to-be/12957/2)
to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
