SetBalanceWeight

Applicability

Product

Supported

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

x

Atlas inference product

x

Atlas training product

x

Function Description

Sets the weight of balanced distribution.

Prototype

1
void SetBalanceWeight(const BalanceWeight &balanceWeight) = 0

Parameters

Parameter

Input/Output

Description

balanceWeight

Input

Weight of balanced distribution

struct BalanceWeight {

int32_t rowNum = 0; // Number of rows in the weight matrix

int32_t colNum = 0; // Number of columns in the weight matrix

const int32_t *matrix = nullptr; // Pointer to the weight matrix, which is an array with rowNum rows and colNum columns. If this pointer is null, all elements of the matrix are 1.

};

Returns

None

Exception Handling

None

Constraints

None