This page contains proposed changes for a future release of ROCm. Read the latest Linux release of ROCm documentation for your production environments.

ReLU

ReLU#

MIOpen: ReLU
ReLU

Functions

miopenStatus_t miopenGetPReLUBackwardWorkspaceSize (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, miopenTensorDescriptor_t weightDesc, size_t *sizeInBytes)
 Helper function to query the minimum workspace size required by the PReLU backward call. More...
 
miopenStatus_t miopenPReLUBackward (miopenHandle_t handle, void *workspace, size_t workspaceSizeInBytes, miopenTensorDescriptor_t inputDesc, const void *input, miopenTensorDescriptor_t weightDesc, const void *weight, miopenTensorDescriptor_t doutputDesc, const void *doutput, miopenTensorDescriptor_t dinputDesc, void *dinput, miopenTensorDescriptor_t dweightDesc, void *dweight)
 Execute a PReLU backward layer. More...
 

Detailed Description

Function Documentation

◆ miopenGetPReLUBackwardWorkspaceSize()

miopenStatus_t miopenGetPReLUBackwardWorkspaceSize ( miopenHandle_t  handle,
miopenTensorDescriptor_t  inputDesc,
miopenTensorDescriptor_t  weightDesc,
size_t *  sizeInBytes 
)

Helper function to query the minimum workspace size required by the PReLU backward call.

Parameters
handleMIOpen Handle (input)
inputDescTensor descriptor for input tensor (input)
weightDescTensor descriptor for weight tensor (input)
sizeInBytesPointer to data to return the minimum workspace size
Returns
miopenStatus_t
Examples
/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-miopen/checkouts/develop/include/miopen/miopen.h.

◆ miopenPReLUBackward()

miopenStatus_t miopenPReLUBackward ( miopenHandle_t  handle,
void *  workspace,
size_t  workspaceSizeInBytes,
miopenTensorDescriptor_t  inputDesc,
const void *  input,
miopenTensorDescriptor_t  weightDesc,
const void *  weight,
miopenTensorDescriptor_t  doutputDesc,
const void *  doutput,
miopenTensorDescriptor_t  dinputDesc,
void *  dinput,
miopenTensorDescriptor_t  dweightDesc,
void *  dweight 
)

Execute a PReLU backward layer.

Parameters
handleMIOpen handle (input)
workspaceAddress of the allocated workspace data (input)
workspaceSizeInBytesSize in bytes of the allocated workspace data (input)
inputDescTensor descriptor for input tensor (input)
inputData tensor input (input)
weightDescTensor descriptor for weight tensor (input)
weightData tensor weight (input)
doutputDescTensor descriptor for output gradient (input)
doutputGradient of output (input)
dinputDescTensor descriptor for input gradient (input)
dinputGradient of input (output)
dweightDescTensor descriptor for weight gradient (input)
dweightGradient of weight (output)
Examples
/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-miopen/checkouts/develop/include/miopen/miopen.h.