GetConcatTmpSize

Function Usage

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

Prototype

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

Parameters

Table 1 API parameters

API

Input/Output

Function

ascendcPlatform

Input

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

elemCount

Input

Number of input elements.

dataTypeSize

Input

Size of the input data, in bytes

Returns

Size of the temporary space required by the Concat API.

Availability

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);