diff --git a/src/operators/feed_op.h b/src/operators/feed_op.h index 3ca2b5668d1c9c34f341e0fe487819501a0ebe8d..286b0076ef2b9ad806f141c4d6124f1233dc78dc 100644 --- a/src/operators/feed_op.h +++ b/src/operators/feed_op.h @@ -47,7 +47,7 @@ class FeedOp : public framework::OperatorBase { fpga::BypassArgs args; args.convert_type = fpga::DATA_FP32_TO_FP16; args.layout_type = fpga::LAYOUT_CHW_TO_HWC; - args.image.address = nullptr; + args.image.address = (void *)input_ptr; args.image.channels = input->dims()[1]; args.image.height = input->dims()[2]; args.image.width = input->dims()[3];