GetConcatTmpSize
功能说明
获取Concat接口所需的临时空间大小。
函数原型
1 | inline uint32_t GetConcatTmpSize(const platform_ascendc::PlatformAscendC &ascendcPlatform, const uint32_t elemCount, const uint32_t dataTypeSize) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
ascendcPlatform |
输入 |
传入硬件平台的信息，PlatformAscendC定义请参见构造及析构函数。 |
elemCount |
输入 |
输入元素个数。 |
dataTypeSize |
输入 |
输入数据大小（单位为字节）。 |
返回值
Concat接口所需的临时空间大小。
支持的型号
调用示例
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); |
父主题： Sort