From bd8e81bbdf87b56ccd5ac006888f8d62f6882197 Mon Sep 17 00:00:00 2001
From: wangliu <wangliu@baidu.com>
Date: Wed, 6 Jun 2018 12:11:20 +0800
Subject: [PATCH] modify code style

---
 src/io/io.h | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/src/io/io.h b/src/io/io.h
index 13a6761b81..f155c971c7 100644
--- a/src/io/io.h
+++ b/src/io/io.h
@@ -14,8 +14,8 @@ limitations under the License. */
 
 #pragma once
 
-#include <memory.h>
 #include <map>
+#include <memory>
 #include <string>
 #include <vector>
 
@@ -27,21 +27,21 @@ limitations under the License. */
 
 namespace paddle_mobile {
 
-template <typename Dtype, Precision P = Precision::FP32>
+template <typename Dtype = CPU, Precision P = Precision::FP32>
 class Loader {
- public:
+public:
   const framework::Program<Dtype, P> Load(const std::string &dirname,
                                           bool optimize = false);
 
- private:
+private:
   void LoadVar(framework::Variable *variable,
                const framework::VarDesc &var_desc,
                const std::string &file_path);
 };
 
-template <typename Dtype, Precision P = Precision::FP32>
+template <typename Dtype = CPU, Precision P = Precision::FP32>
 class Executor {
- public:
+public:
   typedef typename PrecisionTrait<P>::ptype Ptype;
 
   Executor(const framework::Program<Dtype> p, int batch_size = 1,
@@ -52,7 +52,7 @@ class Executor {
   std::vector<Ptype> Predict(const std::vector<Ptype> &input,
                              const std::vector<int64_t> &dims);
 
- protected:
+protected:
   Executor() = default;
 
   void InitMemory();
@@ -64,9 +64,9 @@ class Executor {
   std::shared_ptr<framework::Tensor> Predict(const framework::Tensor &t,
                                              int block_id);
   std::map<framework::BlockDesc,
-           std::vector<std::shared_ptr<framework::OperatorBase<Dtype>>>>
-      ops_of_block_;
+          std::vector<std::shared_ptr<framework::OperatorBase<Dtype>>>>
+          ops_of_block_;
   bool use_optimize_ = false;
 };
 
-}  // namespace paddle_mobile
+}  // namespace paddle_mobile
\ No newline at end of file
-- 
GitLab