Browse Source

add infer format for single op

tags/v1.3.0
y00500818 3 years ago
parent
commit
775d53be2c
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      ge/generator/ge_generator.cc

+ 3
- 0
ge/generator/ge_generator.cc View File

@@ -843,6 +843,9 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &in
Graph graph;
GE_CHK_STATUS(BuildSingleOpGraph(op_desc, inputs, outputs, name, graph),
"[Build][Graph] for single op:%s fail.", op_desc->GetName().c_str());
auto op = OpDescUtils::CreateOperatorFromOpDesc(op_desc);
GE_CHK_STATUS_RET(op_desc->CallInferFormatFunc(op),
"[Call][InferFormatFunc] for single op:%s fail.", op_desc->GetName().c_str());

// 2. check engine type when compile online
if (model_file_name == kFileNameSuffix) {


Loading…
Cancel
Save