企业官网建设流程全解析

--is_input_adjust_hw_layout

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

全量芯片支持

功能说明

与--input_fp16_nodes参数配合使用，指定网络模型输入数据类型为float16、数据格式为NC1HWC0。单独配置本参数无效，但模型转换成功。

关联参数

• --is_input_adjust_hw_layout参数设置为true，需要与--input_fp16_nodes参数配合使用，通过--input_fp16_nodes参数指定的节点输入数据类型为float16、输入数据格式为NC1HWC0。

  --is_input_adjust_hw_layout参数取值个数必须和--input_fp16_nodes参数指定节点个数匹配，多个参数取值使用英文逗号分隔。

• 当--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> ...

依赖约束

无。

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge