GetFormats函数
函数功能
查询op_type这个算子支持的输入输出数据类型。
函数原型
void GetFormats(const std::string& op_type, std::vector<domi::tagDomiTensorFormat>& input_format_vector,std::vector<domi::tagDomiTensorFormat>& output_format_vector);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
op_type |
输入 |
算子类型名称 |
input_format_vector |
输出 |
支持的输入数据类型列表 |
output_format_vector |
输出 |
支持的输出数据类型列表 |