Updated on 2022-03-13 GMT+08:00

Formats

Function

Sets the input/output data formats supported by an operator.

Syntax

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);

Parameter Description

Parameter

Input/Output

Description

input_format

Input

Data format supported by the input

output_format

Input

Data format supported by the output