How to deal with prim::DictConstruct

[popojames]

Hello @masahi

I have tried using your TraceWrapper and it works on BertForSequenceClassification model.

Thanks