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.
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.

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

◆ 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)