未验证 提交 0f875ef3 编写于 作者: 石晓伟 提交者: GitHub

update operator-list function, test=develop (#3003)

上级 b908e834
......@@ -19,6 +19,10 @@
namespace paddle {
namespace lite {
const std::map<std::string, std::string> &GetOp2PathDict() {
return OpKernelInfoCollector::Global().GetOp2PathDict();
}
std::list<std::unique_ptr<KernelBase>> KernelRegistry::Create(
const std::string &op_type,
TargetType target,
......
......@@ -72,6 +72,8 @@ class OpKernelInfoCollector {
namespace paddle {
namespace lite {
const std::map<std::string, std::string> &GetOp2PathDict();
using KernelFunc = std::function<void()>;
using KernelFuncCreator = std::function<std::unique_ptr<KernelFunc>()>;
class LiteOpRegistry final : public Factory<OpLite, std::shared_ptr<OpLite>> {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册