提交 645d3f69 编写于 作者: C Chon 提交者: GitHub

Merge pull request #804 from qnqinan/develop

add initialization of padding parameter in feed op closed #803
...@@ -56,6 +56,8 @@ class FeedOp : public framework::OperatorBase<DeviceType> { ...@@ -56,6 +56,8 @@ class FeedOp : public framework::OperatorBase<DeviceType> {
args.image.channels = input->dims()[1]; args.image.channels = input->dims()[1];
args.image.height = input->dims()[2]; args.image.height = input->dims()[2];
args.image.width = input->dims()[3]; args.image.width = input->dims()[3];
args.image.pad_height = 0;
args.image.pad_width = 0;
args.output.address = output_ptr; args.output.address = output_ptr;
fpga::PerformBypass(args); fpga::PerformBypass(args);
} }
......
...@@ -108,6 +108,7 @@ if ("FPGAnets" IN_LIST NET) ...@@ -108,6 +108,7 @@ if ("FPGAnets" IN_LIST NET)
set(DROPOUT_OP ON) set(DROPOUT_OP ON)
set(FUSION_CONVBNRELU_OP ON) set(FUSION_CONVBNRELU_OP ON)
set(FUSION_CONVBN_OP ON) set(FUSION_CONVBN_OP ON)
set(FUSION_CONVADD_OP ON)
set(FOUND_MATCH ON) set(FOUND_MATCH ON)
endif() endif()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册