提交 f453b713 编写于 作者: H hedaoyuan

Refine code.

上级 a850dec9
...@@ -126,14 +126,11 @@ public: ...@@ -126,14 +126,11 @@ public:
inputData += inputChannels * inputHeight * inputWidth; inputData += inputChannels * inputHeight * inputWidth;
outputData += outputChannels * outputHeight * outputWidth; outputData += outputChannels * outputHeight * outputWidth;
} }
#ifdef PADDLE_MOBILE_INFERENCE
if (Device == DEVICE_TYPE_CPU) {
memory_.reset();
}
#endif
} }
}; };
#ifdef PADDLE_MOBILE_INFERENCE
/* /*
* \brief Forward calculation of convolution, optimized for mobile. * \brief Forward calculation of convolution, optimized for mobile.
*/ */
...@@ -284,6 +281,8 @@ public: ...@@ -284,6 +281,8 @@ public:
} }
}; };
#endif
/* /*
* \brief Backward input calculation of convolution. * \brief Backward input calculation of convolution.
*/ */
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册