SoftMax/SimpleSoftMax Tiling
功能说明
用于获取SoftMax/SimpleSoftMax Tiling参数。
函数原型
1
|
uint32_t GetSoftMaxMaxTmpSize(const ge::Shape& srcShape, const uint32_t dataTypeSize, const bool isReuseSource) |
1
|
uint32_t GetSoftMaxMinTmpSize(const ge::Shape& srcShape, const uint32_t dataTypeSize, const bool isReuseSource) |
1
|
void SoftMaxTilingFunc(const ge::Shape srcShape&, const uint32_t dataTypeSize, const uint32_t localWorkSpaceSize, optiling::SoftMaxTiling& softmaxTiling) |
参数说明
接口 |
输入/输出 |
功能 |
|---|---|---|
srcShape |
输入 |
输入srcTensor的shape信息。 |
dataTypeSize |
输入 |
参与计算的max和sum的数据类型，比如half=2。 |
isReuseSource |
输入 |
与kernel侧接口配置保持一致。 |
接口 |
输入/输出 |
功能 |
|---|---|---|
srcShape |
输入 |
输入srcTensor的shape信息。 |
localWorkSpaceSize |
输入 |
剩余的可供SoftMax接口计算的空间大小，单位为Byte。 |
dataTypeSize |
输入 |
参与计算的max和sum的数据类型，比如half=2。 |
softmaxTiling |
输出 |
输出SoftMax接口所需的tiling信息。 |
返回值
GetSoftMaxMaxTmpSize返回SoftMax/SimpleSoftMax接口能完成计算所需最大临时空间大小，单位为Byte。
GetSoftMaxMinTmpSize返回SoftMax/SimpleSoftMax接口能完成计算所需最小临时空间大小，单位为Byte。
父主题： SoftMax
