AllocTensorListMsg

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

Allocates a continuous memory block based on the input dtype shapes array to carry the tensor array.

Prototype

virtual std::shared_ptr<FlowMsg> AllocTensorListMsg(const std::vector<std::vector<int64_t>> &shapes, const std::vector<TensorDataType> &dataTypes, uint32_t align = 512U)

Parameters

Parameter	Input/Output	Description
shapes	Input	Shape list of the tensor list.
dataTypes	Input	DataType list of the tensor list.
align	Input	Alignment size of the allocated memory address. Values: [32, 64, 128, 256, 512, 1024].

Returns

Pointer to the allocated FlowMsg.

Exception Handling

NULL is returned if the tensor allocation fails.

Constraints

None

Parent topic: Class MetaRunContext