--is_input_adjust_hw_layout
功能说明
与--input_fp16_nodes参数配合使用,指定网络模型输入数据类型为float16、数据格式为NC1HWC0。单独配置本参数无效,但模型转换成功。
关联参数
- --is_input_adjust_hw_layout参数设置为true,需要与--input_fp16_nodes参数配合使用,通过--input_fp16_nodes参数指定的节点输入数据类型为float16、输入数据格式为NC1HWC0。
- 当--framework取值为1,且为MindSpore框架网络模型时,设置本参数无效,但模型转换成功。
参数取值
- true:指定网络模型输入数据类型为float16、数据格式为NC1HWC0。
- false:(默认值)不指定。
推荐配置及收益
无。
示例
- 若--input_fp16_nodes配置了一个节点:
atc --model=$HOME/module/resnet50_tensorflow*.pb --framework=3 --output=$HOME/module/out/tf_resnet50 --is_input_adjust_hw_layout=true --input_fp16_nodes="data" --soc_version=<soc_version>
- 若--input_fp16_nodes配置了多个节点:
atc --model=$HOME/module/resnet50_tensorflow*.pb --framework=3 --output=$HOME/module/out/tf_resnet50 --is_input_adjust_hw_layout="true,true" --input_fp16_nodes="data1;data2" --soc_version=<soc_version>
支持的型号
依赖约束
无。
父主题: 功能配置选项