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

update name of scale address in feed op

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