|
|
@@ -816,7 +816,7 @@ Status ModelBuilder::CompileSingleOp() { |
|
|
|
return ge::SUCCESS; |
|
|
|
} |
|
|
|
|
|
|
|
Status ModelBuilder::CheckAicpuOp(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, std::set<std::string> &tf_engine_set) { |
|
|
|
Status ModelBuilder::CheckAicpuOp(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, std::set<std::string> &tf_engine_set) { |
|
|
|
GE_CHECK_NOTNULL(op_desc); |
|
|
|
std::string aicpu_optype; |
|
|
|
bool has_attr_check_cpu = AttrUtils.GetStr(op_desc, "needCheckCpu", aicpu_optype); |
|
|
|