更新时间: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

输出

支持的权重数据类型列表

分享:

    相关文档

    相关产品