diff --git a/lite/api/cxx_api.cc b/lite/api/cxx_api.cc index 4016b02697369bf84a54bec9f84518c3b62c719c..ad3eb429b393514e170e68915b49c251164b38c2 100644 --- a/lite/api/cxx_api.cc +++ b/lite/api/cxx_api.cc @@ -28,13 +28,7 @@ namespace paddle { namespace lite { std::vector GetAllOps() { - const std::map &op2path = - OpKernelInfoCollector::Global().GetOp2PathDict(); - std::vector res; - for (const auto &op : op2path) { - res.push_back(op.first); - } - return res; + return OpLiteFactory::Global().GetAllOps(); } void Predictor::SaveModel(const std::string &dir, diff --git a/lite/core/op_registry.h b/lite/core/op_registry.h index 249a6aa72f7b65ed7697219eecbf74f0ee8df099..90a2b563af7e17a4806bd47cb883d9590cdab40f 100644 --- a/lite/core/op_registry.h +++ b/lite/core/op_registry.h @@ -97,6 +97,14 @@ class OpLiteFactory { return ss.str(); } + std::vector GetAllOps() const { + std::vector res; + for (const auto& op : op_registry_) { + res.push_back(op.first); + } + return res; + } + protected: std::map()>> op_registry_; };