将数据格式字符串转化为Format类型值。
Format DataFormatToFormat(const std::string &str);
参数 |
输入/输出 |
说明 |
|---|---|---|
str |
输入 |
待转换的Format字符串形式 |
输入合法时，返回转换后的Format enum值，枚举定义请参考ge::Format，仅支持部分格式，详见约束说明；输入不合法时，返回FORMAT_RESERVED，并打印报错信息。
只支持以下几种格式：
"NCHW": FORMAT_NCHW
"NHWC": FORMAT_NHWC
"NDHWC": FORMAT_NDHWC
"NCDHW": FORMAT_NCDHW
"ND": FORMAT_ND
std::string format_str = "NHWC"; auto format = DataFormatToFormat(format_str); // 1