Intel® oneAPI Deep Neural Network Developer Guide and Reference
A newer version of this document is available. Customers should click here to go to the newest version.
PReLUBackward
General
PReLUBackward operation computes gradient for PReLU.
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  |  
       
Broadcasting Rules
Only slope tensor supports broadcasting semantics. Slope tensor is uni-directionally broadcasted to 
 if one of the following rules is met:
PyTorch case: slope is 1D tensor and broadcast per channel, length of slope is equal to the length of
 in channel dimension.PyTorch case: slope is 1D tensor and broadcast per tensor, length of slope is equal to 1.
Tensorflow case: slope is nD tensor and its dimensions must be equal to the
 dimensions starting from the second element: 
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  |  
       
2  |  
        diff_dst  |  
        Required  |  
       
Outputs
Index  |  
        Argument Name  |  
        Required or Optional  |  
       
|---|---|---|
0  |  
        diff_src  |  
        Required  |  
       
1  |  
        diff_slope  |  
        Required  |  
       
Supported data types
PReLUBackward operation supports the following data type combinations.
Src  |  
        Slope  |  
        Diff_dst  |  
        Diff_src  |  
        Diff_slope  |  
       
|---|---|---|---|---|
f32  |  
        f32  |  
        f32  |  
        f32  |  
        f32  |  
       
bf16  |  
        bf16  |  
        bf16  |  
        bf16  |  
        bf16  |  
       
f16  |  
        f16  |  
        f16  |  
        f16  |  
        f16  |