提交 886c997b 编写于 作者: qnqinan's avatar qnqinan

fix a bug in sigmoid kernel in FPGA track

上级 9908a741
...@@ -33,11 +33,9 @@ bool SigmoidKernel<FPGA, float>::Init(SigmoidParam<FPGA> *param) { ...@@ -33,11 +33,9 @@ bool SigmoidKernel<FPGA, float>::Init(SigmoidParam<FPGA> *param) {
args.input_data_type = fpga::DATA_TYPE_FP16; args.input_data_type = fpga::DATA_TYPE_FP16;
args.output_data_type = fpga::DATA_TYPE_FP16; args.output_data_type = fpga::DATA_TYPE_FP16;
args.image.address = input_ptr; args.image.address = input_ptr;
args.image.height = args.image.height = 1;
(input->dims().size() == 4) ? (uint32_t)input->dims()[2] : 1; args.image.width = 1;
args.image.width = args.image.channels = input->fpga_data_num;
(input->dims().size() == 4) ? (uint32_t)input->dims()[3] : 1;
args.image.channels = (uint32_t)input->dims()[1];
args.output.address = out->data<half>(); args.output.address = out->data<half>();
args.output.scale_address = out->scale; args.output.scale_address = out->scale;
args.output.activation.activation_type = activation_enable; args.output.activation.activation_type = activation_enable;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册