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

GetFormats

Function

Queries the input and output data types supported by the op_type operator.

Syntax

void GetFormats(const std::string& op_type, std::vector<domi::tagDomiTensorFormat>& input_format_vector,std::vector<domi::tagDomiTensorFormat>& output_format_vector);

Parameter Description

Parameter

Input/Output

Description

op_type

Input

Name of an operator type

input_format_vector

Output

Supported input data types

output_format_vector

Output

Supported output data types