From ee896ecbbe59f1e0f4bec1b03f9cadc5c5a94b4e Mon Sep 17 00:00:00 2001
From: wuchenghui <wuchenghui@xiaomi.com>
Date: Mon, 29 Jan 2018 10:29:26 +0800
Subject: [PATCH] add dsp_mode parameter

---
 model.config      | 1 +
 validate_model.sh | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/model.config b/model.config
index b9b75e95..ccfc1159 100644
--- a/model.config
+++ b/model.config
@@ -8,3 +8,4 @@ RUNTIME=gpu
 TUNING_OR_NOT=1
 ANDROID_ABI=armeabi-v7a # Or 'arm64-v8a'
 LIMIT_OPENCL_KERNEL_TIME=0
+DSP_MODE=0 # used only for dsp
diff --git a/validate_model.sh b/validate_model.sh
index b8cfdaf4..6ee71ec2 100755
--- a/validate_model.sh
+++ b/validate_model.sh
@@ -134,6 +134,9 @@ echo "Step 2: Convert tf model to mace model and optimize memory"
 bazel build //lib/python/tools:tf_converter || exit -1
 rm -rf ${MODEL_CODEGEN_DIR}
 mkdir -p ${MODEL_CODEGEN_DIR}
+if [ ${DSP_MODE} ]; then
+    DSP_MODE_FLAG="--dsp_mode=${DSP_MODE}"
+fi
 bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \
                                         --output=${MODEL_CODEGEN_DIR}/model.cc \
                                         --input_node=${TF_INPUT_NODE} \
@@ -143,6 +146,7 @@ bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \
                                         --output_type=source \
                                         --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \
                                         --model_tag=${MODEL_TAG} \
+                                        ${DSP_MODE_FLAG} \
                                         --obfuscate=True || exit -1
 
 echo "Step 3: Download mace static library"
-- 
GitLab