下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

DataFormatToFormat

函数功能

将数据格式字符串转化为Format类型值。

使用该接口需要包含type_utils.h头文件。

#include "graph/utils/type_utils.h"

函数原型

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
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词