提交 570c659f 编写于 作者: W wuchenghui

add softmax support for dsp

上级 3270a015
TF_INPUT_NODE=input
TF_OUTPUT_NODE=output
TF_OUTPUT_BR_NODE=before_softmax_node
TF_MODEL_FILE_PATH=path/to/tf_model_opt.pb
MODEL_TAG=mobile_net
INPUT_SHAPE=1,64,64,3
......
......@@ -15,7 +15,6 @@ source $1
if [ x"$RUNTIME" = x"dsp" ]; then
DATA_TYPE="DT_UINT8"
DEVICE_TYPE="HEXAGON"
TF_OUTPUT_NODE=${TF_OUTPUT_BR_NODE}
elif [ x"$RUNTIME" = x"gpu" ]; then
DATA_TYPE="DT_HALF"
DEVICE_TYPE="OPENCL"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册