提交 e6459329 编写于 作者: E eclipsess

code style

上级 9bfe2c66
...@@ -17,14 +17,13 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE ...@@ -17,14 +17,13 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE. SOFTWARE.
==============================================================================*/ ==============================================================================*/
#pragma once #pragma once
#include "operators/elementwise_add_op.h"
#include "../test_include.h" #include "../test_include.h"
#include "operators/elementwise_add_op.h"
namespace paddle_mobile { namespace paddle_mobile {
namespace framework { namespace framework {
template<typename Dtype> template <typename Dtype> class TestElementwiseAddOp {
class TestElementwiseAddOp {
public: public:
TestElementwiseAddOp(const Program<Dtype> p) : program_(p) { TestElementwiseAddOp(const Program<Dtype> p) : program_(p) {
if (use_optimize_) { if (use_optimize_) {
...@@ -120,11 +119,10 @@ namespace paddle_mobile { ...@@ -120,11 +119,10 @@ namespace paddle_mobile {
op->Run(); op->Run();
} }
} }
}; };
template template class TestElementwiseAddOp<CPU>;
class TestElementwiseAddOp<CPU>; } // namespace framework
} // namespace framework
} // namespace paddle_mobile } // namespace paddle_mobile
int main() { int main() {
DLOG << "----------**********----------"; DLOG << "----------**********----------";
...@@ -162,4 +160,3 @@ int main() { ...@@ -162,4 +160,3 @@ int main() {
<< output_add_ptr[226]; << output_add_ptr[226];
return 0; return 0;
} }
...@@ -17,16 +17,15 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE ...@@ -17,16 +17,15 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE. SOFTWARE.
==============================================================================*/ ==============================================================================*/
#pragma once #pragma once
#include "operators/mul_op.h"
#include "../test_include.h" #include "../test_include.h"
#include "operators/mul_op.h"
namespace paddle_mobile { namespace paddle_mobile {
namespace framework { namespace framework {
template<typename Dtype> template <typename Dtype> class TestMulOp {
class TestMulOp {
public: public:
TestMulOp(const Program <Dtype> p) : program_(p) { TestMulOp(const Program<Dtype> p) : program_(p) {
if (use_optimize_) { if (use_optimize_) {
to_predict_program_ = program_.optimizeProgram; to_predict_program_ = program_.optimizeProgram;
} else { } else {
...@@ -108,7 +107,7 @@ namespace paddle_mobile { ...@@ -108,7 +107,7 @@ namespace paddle_mobile {
const framework::Program<Dtype> program_; const framework::Program<Dtype> program_;
std::shared_ptr<ProgramDesc> to_predict_program_; std::shared_ptr<ProgramDesc> to_predict_program_;
std::map<framework::BlockDesc, std::map<framework::BlockDesc,
std::vector<std::shared_ptr<OperatorBase < Dtype>>>> std::vector<std::shared_ptr<OperatorBase<Dtype>>>>
ops_of_block_; ops_of_block_;
bool use_optimize_ = false; bool use_optimize_ = false;
...@@ -122,14 +121,13 @@ namespace paddle_mobile { ...@@ -122,14 +121,13 @@ namespace paddle_mobile {
op->Run(); op->Run();
} }
} }
}; };
template template class TestMulOp<CPU>;
class TestMulOp<CPU>; } // namespace framework
} // namespace framework } // namespace paddle_mobile
} // namespaece paddle_mobile
int main () { int main() {
DLOG << "----------**********----------"; DLOG << "----------**********----------";
DLOG << "begin to run MulOp Test"; DLOG << "begin to run MulOp Test";
paddle_mobile::Loader<paddle_mobile::CPU> loader; paddle_mobile::Loader<paddle_mobile::CPU> loader;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册