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