|
|
|
@@ -83,9 +83,11 @@ class ModelBuilder { |
|
|
|
|
|
|
|
Status CompileSingleOp(); |
|
|
|
|
|
|
|
Status CollectCheckAicpuAttr(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, std::set<std::string> &tf_engine_set); |
|
|
|
Status CollectCheckAicpuAttr(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, |
|
|
|
std::set<std::string> &tf_engine_set); |
|
|
|
|
|
|
|
Status SetModelCheckAicpuAttr(ge::Model &model, std::set<std::string> &aicpu_optype_set, std::set<std::string> &aicpu_tf_optype_set); |
|
|
|
Status SetModelCheckAicpuAttr(ge::Model &model, std::set<std::string> &aicpu_optype_set, |
|
|
|
std::set<std::string> &aicpu_tf_optype_set); |
|
|
|
|
|
|
|
uint64_t session_id_; |
|
|
|
|
|
|
|
|