提交 ee896ecb 编写于 作者: W wuchenghui

add dsp_mode parameter

上级 fe34b8aa
......@@ -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
......@@ -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"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册