AllocTensorMsg

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

Allocates FlowMsg based on the shape, data type, and alignment size.

FlowMsgPtr AllocTensorMsg(const std::vector<int64_t> &shape, DataType data_type, uint32_t align = 512U)

Parameter	Input/Output	Description
shape	Input	Shape of tensor FlowMsg.
data_type	Input	Data type of tensor FlowMsg.
align	Input	Alignment size for the allocated memory address. The value range is [32, 64, 128, 256, 512, 1024]. This parameter is reserved.

Pointer to the allocated FlowMsg.

NULL is returned if the FlowMsg allocation fails.

None

Parent topic: FlowBufferFactory Data Type