!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
pull/5326/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit e5409647bf

@ -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();

Loading…
Cancel
Save