GoodNight-bye
GoodNight-bye
hello, have you finished importing the DLRM model，i have the same task
> no i haven't 发自我的iPhone > […](#) > ------------------ Original ------------------ From: GoodNight-bye ***@***.***> Date: Mon,Jun 21,2021 4:26 PM To: apache/tvm ***@***.***> Cc: Ivy Zhang ***@***.***>, Author ***@***.***> Subject: Re:...
我这边是把网络拆分了，embedding的部分不放在NPU上执行，但是后面有个index算子的问题
embedding_bag is supported in relay/frontend/onnx.py, you can refer to this file.
I have done the job, DLRM is already running on our NPU. If you have any question, you can contact me.
> > I have done the job, DLRM is already running on our NPU. If you have any question, you can contact me. > > 您好， 我目前也遇到这个算子不支持的问题， 能给个联系方式吗？感谢！ aten::embedding_bag relay层直接参考onnx的实现，算子侧的话需要实现对应的take算子