IsBasicBlockInSoftMax
功能说明
用于判断SoftMaxTiling结构是否符合基本块特征。
函数原型
1
|
bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2) |
参数说明
接口 |
输入/输出 |
功能 |
---|---|---|
tiling |
输入 |
待判断的SoftMaxTiling结构。 |
dataTypeSize |
输入 |
参与计算的srcTensor的数据类型大小,比如half=2。 |
返回值
- 返回true表示SoftMaxTiling结构满足基本块Tiling特征。
- 返回false表示SoftMaxTiling结构不满足基本块Tiling特征。
父主题: SoftMax