diff --git a/lite/core/op_registry.cc b/lite/core/op_registry.cc index b49670eefb8b2c6aae30cb041de4d055a2b9964c..94474f6eacc6d22f0b9edbfd755e95511012b4bf 100644 --- a/lite/core/op_registry.cc +++ b/lite/core/op_registry.cc @@ -19,6 +19,10 @@ namespace paddle { namespace lite { +const std::map &GetOp2PathDict() { + return OpKernelInfoCollector::Global().GetOp2PathDict(); +} + std::list> KernelRegistry::Create( const std::string &op_type, TargetType target, diff --git a/lite/core/op_registry.h b/lite/core/op_registry.h index a49682eea68240bfa178eb3d3351b8c7fb41048d..2c98289b852e2950a490f4e4b3b11ec217dd1070 100644 --- a/lite/core/op_registry.h +++ b/lite/core/op_registry.h @@ -72,6 +72,8 @@ class OpKernelInfoCollector { namespace paddle { namespace lite { +const std::map &GetOp2PathDict(); + using KernelFunc = std::function; using KernelFuncCreator = std::function()>; class LiteOpRegistry final : public Factory> {