Parser解析接口支持的配置参数
参数 |
说明 |
---|---|
INPUT_FP16_NODES |
指定输入数据类型为FP16的输入节点名称。 例如:"node_name1;node_name2",指定的节点必须放在双引号中,节点中间使用英文分号分隔。 示例: {ge::AscendString(ge::ir_option::INPUT_FP16_NODES), ge::AscendString("input1;input2")}, |
IS_INPUT_ADJUST_HW_LAYOUT |
用于指定网络输入数据类型是否为FP16,数据格式是否为NC1HWC0。 该参数需要与INPUT_FP16_NODES配合使用。若IS_INPUT_ADJUST_HW_LAYOUT参数设置为true,对应INPUT_FP16_NODES节点的输入数据类型为FP16,输入数据格式为NC1HWC0。 取值范围为false或true,默认值为false。 示例: {ge::AscendString(ge::ir_option::INPUT_FP16_NODES), ge::AscendString("input1;input2")}, {ge::AscendString(ge::ir_option::IS_INPUT_ADJUST_HW_LAYOUT), ge::AscendString("true,true")}, |
OUTPUT |
指定转图后计算图名称。 示例: {ge::AscendString(ge::ir_option::OUTPUT), ge::AscendString("newIssue")}, |
IS_OUTPUT_ADJUST_HW_LAYOUT |
用于指定网络输出的数据类型是否为FP16,数据格式是否为NC1HWC0。 该参数需要与OUT_NODES配合使用。 若IS_OUTPUT_ADJUST_HW_LAYOUT参数设置为true,对应OUT_NODES中输出节点的输出数据类型为FP16,数据格式为NC1HWC0。 取值:false或true,默认为false。 示例: {ge::AscendString(ge::ir_option::OUT_NODES), ge::AscendString("add_input:0")}, {ge::AscendString(ge::ir_option::IS_OUTPUT_ADJUST_HW_LAYOUT), ge::AscendString("true")}, |
OUT_NODES |
指定某层输出节点(算子)作为网络模型的输出或指定网络模型输出的名称。 如果不指定输出节点(算子名称),则模型的输出默认为最后一层的算子信息,如果指定,则以指定的为准。 某些情况下,用户想要查看某层算子参数是否合适,则需要将该层算子的参数输出,既可以在模型编译时通过该参数指定输出某层算子,模型编译后,在相应.om模型文件最后即可以看到指定输出算子的参数信息,如果通过.om模型文件无法查看,则可以将.om模型文件转换成json格式后查看。 支持三种格式:
|
ENABLE_SCOPE_FUSION_PASSES |
指定编译时需要生效的融合规则列表。 融合规则分类如下: 指定的融合规则必须放在双引号中,规则中间使用英文逗号分隔。 示例: {ge::AscendString(ge::ir_option::ENABLE_SCOPE_FUSION_PASSES), ge::AscendString("ScopePass1,ScopePass2")},
说明:
仅aclgrphParseTensorFlow支持该参数。 |
INPUT_DATA_NAMES |
指定模型文件输入节点的name和index属性的映射关系。系统按照输入name的顺序,设置对应输入节点的index属性。 示例: {ge::AscendString(ge::ir_option::INPUT_DATA_NAMES), ge::AscendString("Placeholder,Placeholder_1")}, |
INPUT_SHAPE |
模型输入的shape信息。 参数取值:
配置示例:
说明:
INPUT_SHAPE为可选设置。如果不设置,系统直接读取对应Data节点的shape信息,如果设置,以此处设置的为准,同时刷新对应Data节点的shape信息。 |