Formats函数
函数功能
设置算子支持的输入/输出数据格式。
函数原型
OpRegistrationData& Formats(
const std::initializer_list<domi::tagDomiTensorFormat>& input_formats,
const std::initializer_list<domi::tagDomiTensorFormat>& output_formats);
OpRegistrationData& Formats(
const domi::tagDomiTensorFormat& input_format,
const domi::tagDomiTensorFormat& output_format);
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
input_format |
输入 |
输入支持的数据格式 |
output_format |
输入 |
输出支持的数据格式 |