下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

IsBasicBlockInSoftMax

功能说明

用于判断SoftMaxTiling结构是否符合基本块特征。

函数原型

1
bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2)

参数说明

表1 参数列表

接口

输入/输出

功能

tiling

输入

待判断的SoftMaxTiling结构。

dataTypeSize

输入

参与计算的srcTensor的数据类型大小,比如half=2。

返回值

  • 返回true表示SoftMaxTiling结构满足基本块Tiling特征。
  • 返回false表示SoftMaxTiling结构不满足基本块Tiling特征。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词