FilesLast commitLast update
9 months ago
9 months ago
10 months ago
1 month ago
10 months ago
9 months ago
2 months ago
1 month ago
5 days ago
README

MemSetV2

English | 简体中文

Product Support Status

Product Supported
Ascend 950PR&950DT products √
Atlas A3 products ×
Atlas A2 products ×
Atlas 200I/500 A2 inference products ×
Atlas inference products ×
Atlas training products ×

Function Description

  • Operator function: initializes the output and workspace specified by downstream operators to specified values.

Parameter Description

Parameter Name Input/Output/Attribute Description Data Type Data Format
x Input Tensor to be initialized, which is passed by the framework. INT8, INT16, INT32, INT64, UINT8, UINT16, UINT32, UINT64
BF16, FLOAT16, FLOAT, BOOL
ND
values_int Attribute Specifies the initial value of the int type of the tensor at the corresponding position. int -
values_float Attribute Specifies the initial value of the float type of the tensor at the corresponding position. float -
x Output Tensor to be initialized, which is passed by the framework. The output of this operator is the input, and initialization is performed in place. Same as the input Tensor x ND

Constraint Description

  • None.

Invocation Description

Invocation Method Invocation Sample Description
Graph-mode invocation test_geir_mem_set_v2 Invokes the MemSetV2 operator by building a graph through the operator IR.