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

GetWeightFormats

Function

Queries the weight data type supported by the op_type operator.

Syntax

void GetWeightFormats(const std::string& op_type,

std::vector<domi::tagDomiTensorFormat>& format_vector);

Parameter Description

Parameter

Input/Output

Description

op_type

Input

Name of an operator type

format_vector

Output

Supported weight data types