p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!14963 [MS][LITE] solve const fold bug for conv op 

From: @cjh9368
Reviewed-by: @hangangqiang,@hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang,@hangangqiang
This commit is contained in:
mindspore-ci-bot 2021-04-13 20:00:41 +08:00 committed by Gitee
parent a4c308a38b 3fba8ebd60
commit 5c53383950
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/tools/anf_exporter/anf_exporter.cc
View File

@ -669,7 +669,9 @@ int AnfExporter::ConvertInputParameter(const std::shared_ptr<AnfNode> &input_ano
MS_LOG(ERROR) << "schema tensor format is wrong, " << schema_tensor->format;
return RET_ERROR;
}
if (primitive_c->GetAttr(opt::kWeightFormat) != nullptr) {
// attr weightFormat is only used by conv-like ops' second input
if (output_cnode->inputIndex.size() == 1 && primitive_c->GetAttr(opt::kWeightFormat) != nullptr) {
schema_tensor->format = static_cast<schema::Format>(GetValue<int64_t>(primitive_c->GetAttr(opt::kWeightFormat)));
}
schema_tensor->name = param_node->name();