Visible to Intel only — GUID: GUID-614A8947-2075-467D-A4A2-45F59CC3EA83
Visible to Intel only — GUID: GUID-614A8947-2075-467D-A4A2-45F59CC3EA83
PReLU
General
PReLU operation performs element-wise parametric ReLU operation on a given input tensor, based on the following mathematical formula:
Operation attributes
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
---|---|---|---|---|
Denotes the data format of the input and output data. |
string |
NCX , NXC (default) |
Optional |
|
Denotes whether to apply per_channel broadcast when slope is 1D tensor. |
bool |
false , true (default) |
Optional |
Broadcasting Rules
Only slope tensor supports broadcasting semantics. Slope tensor is uni-directionally broadcasted to if one of the following rules is met:
1: slope is 1D tensor and per_channel_broadcast is set to true, the length of slope tensor is equal to the length of of channel dimension.
2: slope is 1D tensor and per_channel_broadcast is set to false, the length of slope tensor is equal to the length of of the rightmost dimension.
3: slope is nD tensor, starting from the rightmost dimension, or or slope dimension i is empty.
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
src |
Required |
1 |
slope |
Required |
Outputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
dst |
Required |
Supported data types
PReLU operation supports the following data type combinations.
Src |
Dst |
Slope |
---|---|---|
f32 |
f32 |
f32 |
bf16 |
bf16 |
bf16 |
f16 |
f16 |
f16 |