From 040826b70734998d0cb3dcf4c9f5aed5917d3244 Mon Sep 17 00:00:00 2001 From: wangliu Date: Wed, 6 Jun 2018 12:12:57 +0800 Subject: [PATCH] modify code style --- src/io/io.h | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/io/io.h b/src/io/io.h index f155c971c7..ff704b3b45 100644 --- a/src/io/io.h +++ b/src/io/io.h @@ -29,11 +29,11 @@ namespace paddle_mobile { template class Loader { -public: + public: const framework::Program 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); @@ -41,7 +41,7 @@ private: template class Executor { -public: + public: typedef typename PrecisionTrait

::ptype Ptype; Executor(const framework::Program p, int batch_size = 1, @@ -52,7 +52,7 @@ public: std::vector Predict(const std::vector &input, const std::vector &dims); -protected: + protected: Executor() = default; void InitMemory(); @@ -64,9 +64,9 @@ protected: std::shared_ptr Predict(const framework::Tensor &t, int block_id); std::map>>> - ops_of_block_; + std::vector>>> + ops_of_block_; bool use_optimize_ = false; }; -} // namespace paddle_mobile \ No newline at end of file +} // namespace paddle_mobile -- GitLab