提交 2905dcde 编写于 作者: X xzl

fix code format

上级 9f3eb91d
...@@ -66,16 +66,18 @@ class NeonDepthwiseConvFunction : public ConvFunctionBase { ...@@ -66,16 +66,18 @@ 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();
int newSize = batchSize * (inputChannels + 1) * padInputHeight * padInputWidth; int newSize =
batchSize * (inputChannels + 1) * 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,
inputPadding, inputPadding,
batchSize * inputChannels, batchSize * inputChannels,
inputHeight, inputHeight,
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.
先完成此消息的编辑!
想要评论请 注册