HardSwish¶
General¶
HardSwish operation applies following formula on every element of \(\src\) tensor (the variable names follow the standard Naming Conventions):
\[dst = src * \frac{\min(\max(src + 3, 0), 6)}{6}\]
Operation attributes¶
HardSwish operation does not support any attribute.
Execution arguments¶
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs¶
Index |
Argu |
|
|---|---|---|
0 |
|
Required |
Outputs¶
Index |
Argu |
|
|---|---|---|
0 |
|
Required |
Supported data types¶
HardSwish operation supports the following data type combinations.
Src |
D |
|---|---|
f32 |
f32 |
bf16 |
bf16 |
f16 |
f16 |