diff --git a/src/operators/feed_op.h b/src/operators/feed_op.h index 0ed3c448e207305ec4a4a4b2ccd926a1e100757d..33ccd574e3929c4a111801bcaaff221b22b73bb4 100644 --- a/src/operators/feed_op.h +++ b/src/operators/feed_op.h @@ -49,6 +49,7 @@ class FeedOp : public framework::OperatorBase { auto input_ptr = input->data(); Tensor *output = param_.Out(); auto output_ptr = output->mutable_data(); + auto out_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; @@ -59,6 +60,7 @@ class FeedOp : public framework::OperatorBase { args.image.pad_height = 0; args.image.pad_width = 0; args.output.address = output_ptr; + args.output.scale_address = out_address; fpga::PerformBypass(args); }