IsBasicBlockInSoftMax
Function Usage
Determines whether the SoftMaxTiling structure complies with the basic block features.
Prototype
- Computation API in the AscendC::optiling namespace
1bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2)
- Computation API in the AscendC namespace
1bool IsBasicBlockInSoftMax(AscendC::tiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2)
Parameters
API |
Input/Output |
Function |
|---|---|---|
tiling |
Input |
SoftMaxTiling structure to be determined. The input parameter can be in the format of optiling::SoftMaxTiling or AscendC::tiling::SoftMaxTiling. |
dataTypeSize |
Input |
Data type of srcTensor involved in computation, for example, half = 2. |
Returns
- true: SoftMaxTiling structure meets the basic block tiling feature.
- false: SoftMaxTiling structure does not meet the basic block tiling feature.
Restrictions
None
Parent topic: SoftMax APIs