Browse Source

!987 clean cpp check in predict module

Merge pull request !987 from yangruoqi713/master
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
4ba0b372f4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      predict/module/tvm_kernel/lite/src/api/kernel_manager.cc

+ 1
- 1
predict/module/tvm_kernel/lite/src/api/kernel_manager.cc View File

@@ -797,7 +797,7 @@ void update_shape_flatten(const std::vector<DLTensor *> &tensors, int64_t *shape
}

std::string GetEltwiseMode(const OpCommonAttr &opAttr, const mindspore::predict::OpDef &opdef) {
auto &optype = opAttr.optype;
const auto optype = opAttr.optype;
std::string mode = "add";
if (optype == "Eltwise") {
auto op_mode = opdef.attr_as_Eltwise()->mode();


Loading…
Cancel
Save