IsBasicBlockInSoftMax
Function Description
Determines whether the SoftMaxTiling structure complies with the basic block features.
Prototype
1 | bool IsBasicBlockInSoftMax(optiling::SoftMaxTiling& tiling, const uint32_t dataTypeSize = 2) |
Parameters
API |
Input/Output |
Function |
|---|---|---|
tiling |
Input |
SoftMaxTiling structure to be determined. |
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.
Parent topic: SoftMax