From 10f51f745bfe87b3eedfee995a0f7260de665606 Mon Sep 17 00:00:00 2001 From: zhangyang0701 Date: Fri, 1 Mar 2019 14:58:48 +0800 Subject: [PATCH] support lod_mode for FPGA track --- src/io/api_paddle_mobile.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/io/api_paddle_mobile.cc b/src/io/api_paddle_mobile.cc index 7c391c0bf8..ebeabead13 100644 --- a/src/io/api_paddle_mobile.cc +++ b/src/io/api_paddle_mobile.cc @@ -41,10 +41,12 @@ bool PaddleMobilePredictor::Init(const PaddleMobileConfig &config) { config.memory_pack.combined_params_buf); } else if (!config.model_dir.empty()) { paddle_mobile_->Load(config.model_dir, config.optimize, - config.quantification, config.batch_size); + config.quantification, config.batch_size, + config.lod_mode); } else if (!config.prog_file.empty() && !config.param_file.empty()) { paddle_mobile_->Load(config.prog_file, config.param_file, config.optimize, - config.quantification, config.batch_size); + config.quantification, config.batch_size, + config.lod_mode); } else { LOG(kLOG_ERROR) << "fail to load inference model!"; return false; -- GitLab