IsBasicBlockInSoftMax
Function
Determines whether the SoftMaxTiling structure complies with the base 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
Parameter |
Input/Output |
Description |
|---|---|---|
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 base block tiling feature.
- false: SoftMaxTiling structure does not meet the base block tiling feature.
Restrictions
None
Parent topic: SoftMax APIs