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

update name of scale address in feed op

上级 9b15bb3f
...@@ -49,7 +49,7 @@ class FeedOp : public framework::OperatorBase<DeviceType> { ...@@ -49,7 +49,7 @@ class FeedOp : public framework::OperatorBase<DeviceType> {
auto input_ptr = input->data<float>(); auto input_ptr = input->data<float>();
Tensor *output = param_.Out(); Tensor *output = param_.Out();
auto output_ptr = output->mutable_data<half>(); auto output_ptr = output->mutable_data<half>();
auto out_address = output->fpga_args().scale_pointer(); auto output_scale_address = output->fpga_args().scale_pointer();
fpga::BypassArgs args; fpga::BypassArgs args;
args.convert_type = fpga::DATA_FP32_TO_FP16; args.convert_type = fpga::DATA_FP32_TO_FP16;
args.layout_type = fpga::LAYOUT_CHW_TO_HWC; args.layout_type = fpga::LAYOUT_CHW_TO_HWC;
...@@ -60,7 +60,7 @@ class FeedOp : public framework::OperatorBase<DeviceType> { ...@@ -60,7 +60,7 @@ class FeedOp : public framework::OperatorBase<DeviceType> {
args.image.pad_height = 0; args.image.pad_height = 0;
args.image.pad_width = 0; args.image.pad_width = 0;
args.output.address = output_ptr; args.output.address = output_ptr;
args.output.scale_address = out_address; args.output.scale_address = output_scale_address;
fpga::PerformBypass(args); fpga::PerformBypass(args);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册