提交 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> {
args.image.channels = input->dims()[1];
args.image.height = input->dims()[2];
args.image.width = input->dims()[3];
args.image.pad_height = 0;
args.image.pad_width = 0;
args.output.address = output_ptr;
fpga::PerformBypass(args);
}
......
......@@ -108,6 +108,7 @@ if ("FPGAnets" IN_LIST NET)
set(DROPOUT_OP ON)
set(FUSION_CONVBNRELU_OP ON)
set(FUSION_CONVBN_OP ON)
set(FUSION_CONVADD_OP ON)
set(FOUND_MATCH ON)
endif()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册