更新时间:2021-03-18 GMT+08:00
GetWeightFormats函数
函数功能
查询op_type这个算子支持的权重数据类型。
函数原型
void GetWeightFormats(const std::string& op_type,
std::vector<domi::tagDomiTensorFormat>& format_vector);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
op_type |
输入 |
算子类型名称 |
format_vector |
输出 |
支持的权重数据类型列表 |
父主题: OpRegistry类