用于获取SoftMax/SimpleSoftMax Tiling参数。
inline uint32_t GetSoftMaxMaxTmpSize(const ge::Shape srcShape, const uint32_t typeSize, const bool isReuseSource)
inline void TikcSoftMaxTilingFunc(const ge::Shape srcShape, const uint32_t workLocalSize, const uint32_t typeSize,
const bool isReuseSource, optiling::SoftMaxTiling& softmaxTiling)
接口 |
输入/输出 |
功能 |
|---|---|---|
srcShape |
输入 |
设置A矩阵的位置，数据格式，数据类型，是否转置等信息。 |
typeSize |
输入 |
计算的数据类型，比如half=2。 |
isReuseSource |
输入 |
是否复用src输入的空间，与kernel侧接口一致。 |
接口 |
输入/输出 |
功能 |
|---|---|---|
srcShape |
输入 |
输入的shape信息。 |
workLocalSize |
输入 |
剩余的可供softmax接口计算的空间大小。 |
typeSize |
输入 |
计算的数据类型，比如half=2。 |
isReuseSource |
输入 |
是否复用src输入的空间，与kernel侧接口一致。 |
softmaxTiling |
输出 |
输出softmax接口所需的tiling信息。 |
GetSoftMaxMaxTmpSize返回SoftMax/SimpleSoftMax接口能完成计算所需最小临时空间大小