提交 e5409647 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5326 Fixed bug of embedding look up run failed in pynative

Merge pull request !5326 from JoyLvliang/fixed-bug-of-embedding-look-up-run-failed-in-pynative
......@@ -522,6 +522,9 @@ void ConstructInputTensor(const OpExecInfoPtr &op_run_info, std::vector<int> *te
opt::ConstInputToAttrInfoRegister reg;
bool reg_exist = opt::ConstInputToAttrInfoRegistry::Instance().GetRegisterByOpName(op_run_info->op_name, &reg);
if (op_run_info->op_name == prim::kPrimEmbeddingLookup->name()) {
reg_exist = false;
}
op_prim->BeginRecordAddAttr();
size_t input_num = op_run_info->op_inputs.size();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册