用于判断SoftMaxTiling结构是否符合基本块特征。
bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize)；
接口 |
输入/输出 |
功能 |
|---|---|---|
tiling |
输入 |
待判断的SoftMaxTiling结构。 |
dataTypeSize |
输入 |
参与计算的srcTensor的数据类型大小，比如half=2。 |