SetUserTag

Product Support

Product

Supported

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

Atlas inference product's AI Core

Atlas inference product's Vector Core

Atlas training products

Function

Adds user-defined information to a tensor. You can set the corresponding tag as required. You can use GetUserTag to obtain the tag information of a specified tensor and perform operations on the tensor based on the tag information.

Prototype

1
__aicore__ inline void SetUserTag(const TTagType tag)

Parameters

Table 1 Parameters

Parameter

Input/Output

Description

tag

Input

Tag information. The type TTagType corresponds to int32_t.

Returns

None

Restrictions

None

Example

For details, see Example.