Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 9f3eb91d 编写于 作者: Z zlx
浏览文件
操作

fix neon depthwise conv bug

上级 580340ee
显示空白变更内容
内联 并排
Showing with 4 addition and 6 deletion
paddle/function/neon/NeonDepthwiseConv.cpp
浏览文件 @ 9f3eb91d
...@@ -66,8 +66,7 @@ class NeonDepthwiseConvFunction : public ConvFunctionBase { ...@@ -66,8 +66,7 @@ class NeonDepthwiseConvFunction : public ConvFunctionBase {
float* inputPadding = inputData; float* inputPadding = inputData;
int padInputHeight = inputHeight + 2 * paddingH(); int padInputHeight = inputHeight + 2 * paddingH();
int padInputWidth = inputWidth + 2 * paddingW(); int padInputWidth = inputWidth + 2 * paddingW();
if (paddingH() > 0 || paddingW() > 0) { int newSize = batchSize * (inputChannels + 1) * padInputHeight * padInputWidth;
int newSize = batchSize * inputChannels * padInputHeight * padInputWidth;
resizeBuffer<Device>(newSize); resizeBuffer<Device>(newSize);
inputPadding = reinterpret_cast<float*>(memory_->getBuf()); inputPadding = reinterpret_cast<float*>(memory_->getBuf());
neon::Padding<float>::run(inputData, neon::Padding<float>::run(inputData,
...@@ -77,7 +76,6 @@ class NeonDepthwiseConvFunction : public ConvFunctionBase { ...@@ -77,7 +76,6 @@ class NeonDepthwiseConvFunction : public ConvFunctionBase {
inputWidth, inputWidth,
padInputHeight, padInputHeight,
padInputWidth); padInputWidth);
}
std::function<void( std::function<void(
const float*, const float*, int, int, int, int, int, int, float*)> const float*, const float*, int, int, int, int, int, int, float*)>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或