fix include header

9cf706c9 · yejianwu · 1c3f9418 · 9cf706c9 · 9cf706c9
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

mace/core/registry.h mace/core/registry.h +1 -0

mace/kernels/proposal.h mace/kernels/proposal.h +1 -0

未找到文件。
--- a/mace/core/registry.h
+++ b/mace/core/registry.h
@@ -15,6 +15,7 @@
 #ifndef MACE_CORE_REGISTRY_H_
 #define MACE_CORE_REGISTRY_H_

+#include <functional>
 #include <map>
 #include <memory>
 #include <mutex>  // NOLINT(build/c++11)

--- a/mace/kernels/proposal.h
+++ b/mace/kernels/proposal.h
@@ -16,6 +16,7 @@
 #define MACE_KERNELS_PROPOSAL_H_

 #include <algorithm>
+#include <cmath>
 #include <vector>

 #include "mace/core/future.h"