GetConcatTmpSize

Function Usage

Obtains the size of the temporary space required by the Concat API.

Prototype

1
uint32_t GetConcatTmpSize(const platform_ascendc::PlatformAscendC &ascendcPlatform, const uint32_t elemCount, const uint32_t dataTypeSize)

Parameters

Table 1 API parameters

Parameter

Input/Output

Description

ascendcPlatform

Input

Hardware platform information that is passed. For details about the definition of PlatformAscendC, see Constructor and Destructor.

elemCount

Input

Number of elements of the input data.

dataTypeSize

Input

Size of the input data, in bytes

Returns

Size of the temporary space required by the Concat API.

Restrictions

None

Example

1
2
3
4
fe::PlatFormInfos platform_info;
auto plat = platform_ascendc::PlatformAscendC(&platform_info);
const uint32_t elemCount = 128;
AscendC::GetConcatTmpSize(plat, elemCount, 2);