Browse Source

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

pull/14963/head
cjh9368 4 years ago
parent
commit
3fba8ebd60
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/lite/tools/anf_exporter/anf_exporter.cc

+ 3
- 1
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();


Loading…
Cancel
Save