下载
EN
注册

--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>

支持的型号

Atlas 200/300/500 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200I/500 A2 推理产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品

依赖约束

无。