aclSetTensorPlaceMent

Applicability

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 products

Atlas training products

Description

Sets the placement attribute of TensorDesc to identify the type of memory for storing tensor data.

Prototype

aclError aclSetTensorPlaceMent(aclTensorDesc *desc, aclMemType memType)

Parameters

Parameter

Input/Output

Description

desc

Output

Pointer to data of the aclTensorDesc type.

Call aclCreateTensorDesc to create data of the aclTensorDesc type in advance.

memType

Input

Value of the placement attribute.

Returns

0 on success; else, failure. For details, see aclError.