提交 f453b713 编写于 作者: H hedaoyuan

Refine code.

上级 a850dec9
......@@ -126,14 +126,11 @@ public:
inputData += inputChannels * inputHeight * inputWidth;
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.
*/
......@@ -284,6 +281,8 @@ public:
}
};
#endif
/*
* \brief Backward input calculation of convolution.
*/
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册