/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-miopen/checkouts/docs-6.3.1/include/miopen/miopen.h File Reference#
miopen.h File Reference
#include <stddef.h>
#include <stdbool.h>
#include <miopen/config.h>
#include <miopen/export.h>
Go to the source code of this file.
Classes | |
struct | miopenConvAlgoPerf_t |
Perf struct for forward, backward filter, or backward data algorithms. More... | |
struct | miopenConvSolution_t |
Performance struct for forward, backward filter, or backward data algorithms in immediate mode. More... | |
struct | miopenTensorArgument_t |
Values of a tensor or scalar argument for the miopenRunSolution function. More... | |
Macros | |
#define | MIOPEN_DECLARE_OBJECT(name) |
#define | MIOPEN_API_VERSION_REDUCE_TENSOR 1 |
Version of TensorReduce API. Applications may use it to ensure backward compatibility with older library versions. More... | |
Typedefs | |
typedef void *(* | miopenAllocatorFunction) (void *context, size_t sizeBytes) |
Custom allocator function. More... | |
typedef void(* | miopenDeallocatorFunction) (void *context, void *memory) |
Custom deallocator function. More... | |
Functions | |
MIOPEN_DECLARE_OBJECT (miopenHandle) | |
Creates the miopenHandle_t type. More... | |
const char * | miopenGetErrorString (miopenStatus_t error) |
Get character string for an error code. More... | |
miopenStatus_t | miopenGetVersion (size_t *major, size_t *minor, size_t *patch) |
Method to return version of MIOpen. More... | |
miopenStatus_t | miopenCreate (miopenHandle_t *handle) |
Method to create the MIOpen handle object. More... | |
miopenStatus_t | miopenCreateWithStream (miopenHandle_t *handle, miopenAcceleratorQueue_t stream) |
Create a MIOpen handle with an accelerator stream. More... | |
miopenStatus_t | miopenDestroy (miopenHandle_t handle) |
Destroys the MIOpen handle. More... | |
miopenStatus_t | miopenSetStream (miopenHandle_t handle, miopenAcceleratorQueue_t streamID) |
Set accelerator command queue previously created. More... | |
miopenStatus_t | miopenGetStream (miopenHandle_t handle, miopenAcceleratorQueue_t *streamID) |
Get the previously created accelerator command queue. More... | |
miopenStatus_t | miopenSetAllocator (miopenHandle_t handle, miopenAllocatorFunction allocator, miopenDeallocatorFunction deallocator, void *allocatorContext) |
Set allocator for previously created miopenHandle. More... | |
miopenStatus_t | miopenGetKernelTime (miopenHandle_t handle, float *time) |
Get time for last kernel launched. More... | |
miopenStatus_t | miopenEnableProfiling (miopenHandle_t handle, bool enable) |
Enable profiling to retrieve kernel time. More... | |
MIOPEN_DECLARE_OBJECT (miopenFusionOpDescriptor) | |
Creates the miopenFusionOpDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenTensorDescriptor) | |
Creates the miopenTensorDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenSeqTensorDescriptor) | |
Creates the miopenSeqTensorDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenConvolutionDescriptor) | |
Creates the miopenConvolutionDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenPoolingDescriptor) | |
Creates the miopenPoolingDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenLRNDescriptor) | |
Creates the miopenLRNDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenActivationDescriptor) | |
Creates the miopenActivationDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenRNNDescriptor) | |
Creates the miopenRNNDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenCTCLossDescriptor) | |
Creates the miopenCTCLossDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenDropoutDescriptor) | |
Creates the miopenDropoutDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenReduceTensorDescriptor) | |
Creates the miopenReduceTensorDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenMhaDescriptor) | |
Creates the miopenMhaDescriptor_t type. More... | |
MIOPEN_DECLARE_OBJECT (miopenSoftmaxDescriptor) | |
Creates the miopenSoftmaxDescriptor_t type. More... | |
miopenStatus_t | miopenCreateTensorDescriptor (miopenTensorDescriptor_t *tensorDesc) |
Create a Tensor Descriptor. More... | |
miopenStatus_t | miopenSet4dTensorDescriptor (miopenTensorDescriptor_t tensorDesc, miopenDataType_t dataType, int n, int c, int h, int w) |
Set shape of 4D tensor. More... | |
miopenStatus_t | miopenSetNdTensorDescriptorWithLayout (miopenTensorDescriptor_t tensorDesc, miopenDataType_t dataType, miopenTensorLayout_t tensorLayout, const int *lens, int num_lens) |
Set shape of ND tensor with specific layout. More... | |
miopenStatus_t | miopenSet4dTensorDescriptorEx (miopenTensorDescriptor_t tensorDesc, miopenDataType_t dataType, int n, int c, int h, int w, int nStride, int cStride, int hStride, int wStride) |
Set shape and stride of 4D tensor. More... | |
miopenStatus_t | miopenGet4dTensorDescriptor (miopenTensorDescriptor_t tensorDesc, miopenDataType_t *dataType, int *n, int *c, int *h, int *w, int *nStride, int *cStride, int *hStride, int *wStride) |
Get the details of the tensor descriptor. More... | |
miopenStatus_t | miopenSetTensorDescriptor (miopenTensorDescriptor_t tensorDesc, miopenDataType_t dataType, int nbDims, const int *dimsA, const int *stridesA) |
Set shape of N-dimensional tensor. More... | |
miopenStatus_t | miopenSetTensorDescriptorV2 (miopenTensorDescriptor_t tensorDesc, miopenDataType_t dataType, int nbDims, const size_t *dimsA, const size_t *stridesA) |
Set shape of N-dimensional tensor. More... | |
miopenStatus_t | miopenSetTensorCastType (miopenTensorDescriptor_t tensorDesc, miopenDataType_t cast_type) |
Set the tensor cast type. More... | |
miopenStatus_t | miopenGetTensorDescriptorSize (miopenTensorDescriptor_t tensorDesc, int *size) |
Set shape of N-dimensional tensor. More... | |
miopenStatus_t | miopenGetTensorDescriptor (miopenTensorDescriptor_t tensorDesc, miopenDataType_t *dataType, int *dimsA, int *stridesA) |
Get the details of the N-dimensional tensor descriptor. More... | |
miopenStatus_t | miopenDestroyTensorDescriptor (miopenTensorDescriptor_t tensorDesc) |
Destroys the tensor descriptor. More... | |
miopenStatus_t | miopenCreateSeqTensorDescriptor (miopenSeqTensorDescriptor_t *tensorDesc) |
Create a Tensor Descriptor for sequence data. More... | |
miopenStatus_t | miopenDestroySeqTensorDescriptor (miopenSeqTensorDescriptor_t tensorDesc) |
Destroys the sequence data tensor descriptor. More... | |
miopenStatus_t | miopenOpTensor (miopenHandle_t handle, miopenTensorOp_t tensorOp, const void *alpha1, const miopenTensorDescriptor_t aDesc, const void *A, const void *alpha2, const miopenTensorDescriptor_t bDesc, const void *B, const void *beta, const miopenTensorDescriptor_t cDesc, void *C) |
Execute element-wise tensor operations. More... | |
miopenStatus_t | miopenSetTensor (miopenHandle_t handle, const miopenTensorDescriptor_t yDesc, void *y, const void *alpha) |
Fills a tensor with a single value. More... | |
miopenStatus_t | miopenScaleTensor (miopenHandle_t handle, const miopenTensorDescriptor_t yDesc, void *y, const void *alpha) |
Scales all elements in a tensor by a single value. More... | |
miopenStatus_t | miopenGetTensorNumBytes (miopenTensorDescriptor_t tensorDesc, size_t *numBytes) |
Returns number of bytes associated with tensor descriptor. More... | |
miopenStatus_t | miopenTransformTensor (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y) |
Copies one tensor to another tensor with a different layout/scale. More... | |
miopenStatus_t | miopenCreateConvolutionDescriptor (miopenConvolutionDescriptor_t *convDesc) |
Creates a convolution layer descriptor. More... | |
miopenStatus_t | miopenInitConvolutionDescriptor (miopenConvolutionDescriptor_t convDesc, miopenConvolutionMode_t c_mode, int pad_h, int pad_w, int stride_h, int stride_w, int dilation_h, int dilation_w) |
Creates a 2-D convolution layer descriptor. More... | |
miopenStatus_t | miopenInitConvolutionNdDescriptor (miopenConvolutionDescriptor_t convDesc, int spatialDim, const int *padA, const int *strideA, const int *dilationA, miopenConvolutionMode_t c_mode) |
Creates a N-dimensional convolution layer descriptor. More... | |
miopenStatus_t | miopenGetConvolutionSpatialDim (miopenConvolutionDescriptor_t convDesc, int *spatialDim) |
Retrieves the spatial dimension of a convolution layer descriptor. More... | |
miopenStatus_t | miopenGetConvolutionDescriptor (miopenConvolutionDescriptor_t convDesc, miopenConvolutionMode_t *c_mode, int *pad_h, int *pad_w, int *stride_h, int *stride_w, int *dilation_h, int *dilation_w) |
Retrieves a 2-D convolution layer descriptor's details. More... | |
miopenStatus_t | miopenGetConvolutionNdDescriptor (miopenConvolutionDescriptor_t convDesc, int requestedSpatialDim, int *spatialDim, int *padA, int *strideA, int *dilationA, miopenConvolutionMode_t *c_mode) |
Retrieves a N-dimensional convolution layer descriptor's details. More... | |
miopenStatus_t | miopenGetConvolutionGroupCount (miopenConvolutionDescriptor_t convDesc, int *groupCount) |
Get the number of groups to be used in Group/Depthwise convolution. More... | |
miopenStatus_t | miopenSetConvolutionGroupCount (miopenConvolutionDescriptor_t convDesc, int groupCount) |
Set the number of groups to be used in Group/Depthwise convolution. More... | |
miopenStatus_t | miopenSetTransposeConvOutputPadding (miopenConvolutionDescriptor_t convDesc, int adj_h, int adj_w) |
Set the output padding to be used in 2-D Transpose convolution. More... | |
miopenStatus_t | miopenSetTransposeConvNdOutputPadding (miopenConvolutionDescriptor_t convDesc, int spatialDim, const int *adjA) |
Set the output padding to be used in N-dimensional Transpose convolution. More... | |
miopenStatus_t | miopenGetConvolutionForwardOutputDim (miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t inputTensorDesc, const miopenTensorDescriptor_t filterDesc, int *n, int *c, int *h, int *w) |
Get the shape of a resulting 4-D tensor from a 2-D convolution. More... | |
miopenStatus_t | miopenGetConvolutionNdForwardOutputDim (miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t inputTensorDesc, const miopenTensorDescriptor_t filterDesc, int *nDim, int *outputTensorDimA) |
Get the shape of a resulting N-dimensional tensor from a (N-2)-dimensional convolution. More... | |
miopenStatus_t | miopenDestroyConvolutionDescriptor (miopenConvolutionDescriptor_t convDesc) |
Destroys the tensor descriptor object. More... | |
miopenStatus_t | miopenSetConvolutionAttribute (miopenConvolutionDescriptor_t convDesc, const miopenConvolutionAttrib_t attr, int value) |
Set the attribute of the convolution descriptor. More... | |
miopenStatus_t | miopenGetConvolutionAttribute (miopenConvolutionDescriptor_t convDesc, const miopenConvolutionAttrib_t attr, int *value) |
Get the attribute of the convolution descriptor. More... | |
miopenStatus_t | miopenConvolutionForwardGetSolutionCount (miopenHandle_t handle, const miopenTensorDescriptor_t wDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, size_t *solutionCount) |
Query the maximum number of solutions applicable for the given input/output and weights tensor descriptor for Convolution in the Forward direction. More... | |
miopenStatus_t | miopenConvolutionForwardGetSolution (miopenHandle_t handle, const miopenTensorDescriptor_t wDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, const size_t maxSolutionCount, size_t *solutionCount, miopenConvSolution_t *solutions) |
Query the applicable solutions for a convolution configuration described by input, output and convolution descriptors. More... | |
miopenStatus_t | miopenConvolutionForwardGetSolutionWorkspaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t wDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, const uint64_t solution_id, size_t *workSpaceSize) |
Returns the workspace size required for a particular solution id. More... | |
miopenStatus_t | miopenConvolutionForwardCompileSolution (miopenHandle_t handle, const miopenTensorDescriptor_t wDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, const uint64_t solution_id) |
Compiles the solution provided by the user, this solution may be acquired by the miopenConvolutionForwardGetSolution API call above. Compiling the solution ensures that the first API call to miopenConvolutionForwardImmediate does not cause a compile. More... | |
miopenStatus_t | miopenConvolutionForwardImmediate (miopenHandle_t handle, const miopenTensorDescriptor_t wDesc, const void *w, const miopenTensorDescriptor_t xDesc, const void *x, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, void *y, void *workSpace, size_t workSpaceSize, const uint64_t solution_id) |
Executes the Forward convolution operation based on the provided solution ID. More... | |
miopenStatus_t | miopenConvolutionBackwardDataGetSolutionCount (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t wDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, size_t *solutionCount) |
Query the maximum number of solutions applicable for the given input/output and weights tensor descriptor for backward Convolution w-r-t Data. More... | |
miopenStatus_t | miopenConvolutionBackwardDataGetSolution (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t wDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, const size_t maxSolutionCount, size_t *solutionCount, miopenConvSolution_t *solutions) |
Query the applicable solutions for a backward convolution w-r-t data as described by input, output and convolution descriptors. More... | |
miopenStatus_t | miopenConvolutionBackwardDataGetSolutionWorkspaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t wDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, const uint64_t solution_id, size_t *workSpaceSize) |
Returns the workspace size required for a particular solution id. More... | |
miopenStatus_t | miopenConvolutionBackwardDataCompileSolution (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t wDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, const uint64_t solution_id) |
Compiles the solution provided by the user, this solution may be acquired by the miopenConvolutionBackwardDataGetSolution API call above. Compiling the solution ensures that the first API call to miopenConvolutionBackwardDataImmediate does not cause a compile. More... | |
miopenStatus_t | miopenConvolutionBackwardDataImmediate (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t wDesc, const void *w, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, void *dx, void *workSpace, size_t workSpaceSize, const uint64_t solution_id) |
Executes the Backward convolution w-r-t data operation based on the provided solution ID. More... | |
miopenStatus_t | miopenConvolutionBackwardWeightsGetSolutionCount (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, size_t *solutionCount) |
Query the maximum number of solutions applicable for the given input/output and weights tensor descriptor for backward Convolution w-r-t Weights. More... | |
miopenStatus_t | miopenConvolutionBackwardWeightsGetSolution (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, const size_t maxSolutionCount, size_t *solutionCount, miopenConvSolution_t *solutions) |
Query the applicable solutions for a backward convolution w-r-t weights as described by input, output and convolution descriptors. More... | |
miopenStatus_t | miopenConvolutionBackwardWeightsGetSolutionWorkspaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, const uint64_t solution_id, size_t *workSpaceSize) |
Returns the workspace size required for a particular solution id. More... | |
miopenStatus_t | miopenConvolutionBackwardWeightsCompileSolution (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, const uint64_t solution_id) |
Compiles the solution provided by the user, this solution may be acquired by the miopenConvolutionBackwardWeightsGetSolution API call above. Compiling the solution ensures that the first API call to miopenConvolutionBackwardWeightsImmediate does not cause a compile. More... | |
miopenStatus_t | miopenConvolutionBackwardWeightsImmediate (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, void *dw, void *workSpace, size_t workSpaceSize, const uint64_t solution_id) |
Executes the Backward convolution w-r-t weights operation based on the provided solution ID. More... | |
miopenStatus_t | miopenConvolutionForwardGetWorkSpaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t wDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, size_t *workSpaceSize) |
Query the workspace size required for a forward convolution algorithm. More... | |
miopenStatus_t | miopenFindConvolutionForwardAlgorithm (miopenHandle_t handle, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t wDesc, const void *w, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t yDesc, void *y, const int requestAlgoCount, int *returnedAlgoCount, miopenConvAlgoPerf_t *perfResults, void *workSpace, size_t workSpaceSize, bool exhaustiveSearch) |
Search and run the forward convolutional algorithms and return a list of kernel times. More... | |
miopenStatus_t | miopenConvolutionForward (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t wDesc, const void *w, const miopenConvolutionDescriptor_t convDesc, miopenConvFwdAlgorithm_t algo, const void *beta, const miopenTensorDescriptor_t yDesc, void *y, void *workSpace, size_t workSpaceSize) |
Execute a forward convolution layer. More... | |
miopenStatus_t | miopenConvolutionForwardBias (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t bDesc, const void *b, const void *beta, const miopenTensorDescriptor_t yDesc, void *y) |
Calculate element-wise scale and shift of a tensor via a bias tensor. More... | |
miopenStatus_t | miopenConvolutionBackwardDataGetWorkSpaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t wDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, size_t *workSpaceSize) |
Query the workspace size required for a backward data convolution algorithm. More... | |
miopenStatus_t | miopenFindConvolutionBackwardDataAlgorithm (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t wDesc, const void *w, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dxDesc, void *dx, const int requestAlgoCount, int *returnedAlgoCount, miopenConvAlgoPerf_t *perfResults, void *workSpace, size_t workSpaceSize, bool exhaustiveSearch) |
Search and run the backwards data convolution algorithms and return a list of kernel times. More... | |
miopenStatus_t | miopenConvolutionBackwardData (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t wDesc, const void *w, const miopenConvolutionDescriptor_t convDesc, miopenConvBwdDataAlgorithm_t algo, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx, void *workSpace, size_t workSpaceSize) |
Execute a backward data convolution layer. More... | |
miopenStatus_t | miopenConvolutionBackwardWeightsGetWorkSpaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t xDesc, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, size_t *workSpaceSize) |
Get the GPU memory required for the backward weights convolution algorithm. More... | |
miopenStatus_t | miopenFindConvolutionBackwardWeightsAlgorithm (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t dwDesc, void *dw, const int requestAlgoCount, int *returnedAlgoCount, miopenConvAlgoPerf_t *perfResults, void *workSpace, size_t workSpaceSize, bool exhaustiveSearch) |
Search and run the backwards weights convolutional algorithms and return a list of kernel times. More... | |
miopenStatus_t | miopenConvolutionBackwardWeights (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const miopenConvolutionDescriptor_t convDesc, miopenConvBwdWeightsAlgorithm_t algo, const void *beta, const miopenTensorDescriptor_t dwDesc, void *dw, void *workSpace, size_t workSpaceSize) |
Execute a backward weights convolution layer. More... | |
miopenStatus_t | miopenConvolutionBackwardBias (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t dyDesc, const void *dy, const void *beta, const miopenTensorDescriptor_t dbDesc, void *db) |
Calculates the gradient with respect to the bias. More... | |
miopenStatus_t | miopenCreatePoolingDescriptor (miopenPoolingDescriptor_t *poolDesc) |
Creates a pooling layer descriptor. More... | |
miopenStatus_t | miopenSetPoolingIndexType (miopenPoolingDescriptor_t poolDesc, miopenIndexType_t index_type) |
Set index data type for pooling layer. The default indexing type is uint8_t. Users can set the index type to any of the miopenIndexType_t sizes; 8, 16, 32, or 64 bit unsigned integers. More... | |
miopenStatus_t | miopenGetPoolingIndexType (miopenPoolingDescriptor_t poolDesc, miopenIndexType_t *index_type) |
Get the index data type for pooling layer. The index type to any of the miopenIndexType_t sizes; 8, 16, 32, or 64 bit unsigned integers. More... | |
miopenStatus_t | miopenSetPoolingWorkSpaceIndexMode (miopenPoolingDescriptor_t poolDesc, miopenPoolingWorkspaceIndexMode_t workspace_index) |
Set workspace index mode for pooling layer. The default mode is miopenPoolingWorkSpaceIndexMask. More... | |
miopenStatus_t | miopenGetPoolingWorkSpaceIndexMode (miopenPoolingDescriptor_t poolDesc, miopenPoolingWorkspaceIndexMode_t *workspace_index) |
Get workspace index mode for pooling layer. More... | |
miopenStatus_t | miopenSet2dPoolingDescriptor (miopenPoolingDescriptor_t poolDesc, miopenPoolingMode_t mode, int windowHeight, int windowWidth, int pad_h, int pad_w, int stride_h, int stride_w) |
Sets a 2-D pooling layer descriptor details. More... | |
miopenStatus_t | miopenGet2dPoolingDescriptor (const miopenPoolingDescriptor_t poolDesc, miopenPoolingMode_t *mode, int *windowHeight, int *windowWidth, int *pad_h, int *pad_w, int *stride_h, int *stride_w) |
Gets a 2-D pooling layer descriptor details. More... | |
miopenStatus_t | miopenGetPoolingForwardOutputDim (const miopenPoolingDescriptor_t poolDesc, const miopenTensorDescriptor_t tensorDesc, int *n, int *c, int *h, int *w) |
Gets the shape of the output tensor for 2-D pooling. More... | |
miopenStatus_t | miopenSetNdPoolingDescriptor (miopenPoolingDescriptor_t poolDesc, const miopenPoolingMode_t mode, int nbDims, const int *windowDimA, const int *padA, const int *stridesA) |
Set details of a N-D pooling layer descriptor. More... | |
miopenStatus_t | miopenGetNdPoolingDescriptor (const miopenPoolingDescriptor_t poolDesc, int nbDimsRequested, miopenPoolingMode_t *mode, int *nbDims, int *windowDimA, int *padA, int *stridesA) |
Get details of a N-D pooling layer descriptor. More... | |
miopenStatus_t | miopenGetPoolingNdForwardOutputDim (const miopenPoolingDescriptor_t poolDesc, const miopenTensorDescriptor_t tensorDesc, int dims, int *tensorDimArr) |
Gets the shape of the output tensor for N-D pooling. More... | |
miopenStatus_t | miopenPoolingGetWorkSpaceSize (const miopenTensorDescriptor_t yDesc, size_t *workSpaceSize) |
Get the amount of GPU memory required for pooling. More... | |
miopenStatus_t | miopenPoolingGetWorkSpaceSizeV2 (const miopenPoolingDescriptor_t poolDesc, const miopenTensorDescriptor_t yDesc, size_t *workSpaceSize) |
Get the amount of GPU memory required for pooling. More... | |
miopenStatus_t | miopenPoolingForward (miopenHandle_t handle, const miopenPoolingDescriptor_t poolDesc, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y, bool do_backward, void *workSpace, size_t workSpaceSize) |
Execute a forward pooling layer. More... | |
miopenStatus_t | miopenPoolingBackward (miopenHandle_t handle, const miopenPoolingDescriptor_t poolDesc, const void *alpha, const miopenTensorDescriptor_t yDesc, const void *y, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx, void *workSpace) |
Execute a backward pooling layer. More... | |
miopenStatus_t | miopenDestroyPoolingDescriptor (miopenPoolingDescriptor_t poolDesc) |
Destroys the pooling descriptor object. More... | |
miopenStatus_t | miopenCreateLRNDescriptor (miopenLRNDescriptor_t *lrnDesc) |
Creates a local response normalization (LRN) layer descriptor. More... | |
miopenStatus_t | miopenSetLRNDescriptor (const miopenLRNDescriptor_t lrnDesc, miopenLRNMode_t mode, unsigned int lrnN, double lrnAlpha, double lrnBeta, double lrnK) |
Sets a LRN layer descriptor details. More... | |
miopenStatus_t | miopenGetLRNDescriptor (const miopenLRNDescriptor_t lrnDesc, miopenLRNMode_t *mode, unsigned int *lrnN, double *lrnAlpha, double *lrnBeta, double *lrnK) |
Gets a LRN layer descriptor details. More... | |
miopenStatus_t | miopenLRNGetWorkSpaceSize (const miopenTensorDescriptor_t yDesc, size_t *workSpaceSize) |
Determine the workspace requirements. More... | |
miopenStatus_t | miopenLRNForward (miopenHandle_t handle, const miopenLRNDescriptor_t lrnDesc, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y, bool do_backward, void *workSpace) |
Execute a LRN forward layer. More... | |
miopenStatus_t | miopenLRNBackward (miopenHandle_t handle, const miopenLRNDescriptor_t lrnDesc, const void *alpha, const miopenTensorDescriptor_t yDesc, const void *y, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx, const void *workSpace) |
Execute a LRN backward layer. More... | |
miopenStatus_t | miopenDestroyLRNDescriptor (miopenLRNDescriptor_t lrnDesc) |
Destroys the LRN descriptor object. More... | |
miopenStatus_t | miopenLayerNormForward (miopenHandle_t handle, miopenNormMode_t mode, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t weightDesc, const void *weight, const miopenTensorDescriptor_t biasDesc, const void *bias, const float epsilon, const int32_t normalized_dim, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t meanDesc, void *mean, const miopenTensorDescriptor_t rstdDesc, void *rstd) |
Execute a layernorm forward layer. More... | |
miopenStatus_t | miopenCatForward (miopenHandle_t handle, const int32_t xCount, const miopenTensorDescriptor_t *xDescs, const void *const *xs, const miopenTensorDescriptor_t yDesc, void *y, const int32_t dim) |
Execute a cat forward layer. More... | |
miopenStatus_t | miopenDeriveBNTensorDescriptor (miopenTensorDescriptor_t derivedBnDesc, const miopenTensorDescriptor_t xDesc, miopenBatchNormMode_t bn_mode) |
Derive tensor for gamma and beta from input tensor descriptor. More... | |
miopenStatus_t | miopenBatchNormalizationForwardTraining (miopenHandle_t handle, miopenBatchNormMode_t bn_mode, void *alpha, void *beta, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t bnScaleBiasMeanVarDesc, void *bnScale, void *bnBias, double expAvgFactor, void *resultRunningMean, void *resultRunningVariance, double epsilon, void *resultSaveMean, void *resultSaveInvVariance) |
Execute forward training layer for batch normalization. More... | |
miopenStatus_t | miopenBatchNormalizationForwardTraining_V2 (miopenHandle_t handle, miopenBatchNormMode_t bn_mode, void *alpha, void *beta, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t scaleDesc, const miopenTensorDescriptor_t biasVarDesc, const miopenTensorDescriptor_t savedMeanDesc, const miopenTensorDescriptor_t savedVarDesc, void *bnScale, void *bnBias, double expAvgFactor, void *resultRunningMean, void *resultRunningVariance, double epsilon, void *resultSaveMean, void *resultSaveInvVariance) |
Execute forward training layer for batch normalization. More... | |
miopenStatus_t | miopenBatchNormalizationForwardInference (miopenHandle_t handle, miopenBatchNormMode_t bn_mode, void *alpha, void *beta, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t bnScaleBiasMeanVarDesc, void *bnScale, void *bnBias, void *estimatedMean, void *estimatedVariance, double epsilon) |
Execute forward inference layer for batch normalization. More... | |
miopenStatus_t | miopenBatchNormalizationForwardInference_V2 (miopenHandle_t handle, miopenBatchNormMode_t bn_mode, void *alpha, void *beta, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t scaleDesc, const miopenTensorDescriptor_t biasDesc, const miopenTensorDescriptor_t estMeanDesc, const miopenTensorDescriptor_t estVarianceDesc, void *bnScale, void *bnBias, void *estimatedMean, void *estimatedVariance, double epsilon) |
Execute forward inference layer for batch normalization. More... | |
miopenStatus_t | miopenBatchNormalizationBackward (miopenHandle_t handle, miopenBatchNormMode_t bn_mode, const void *alphaDataDiff, const void *betaDataDiff, const void *alphaParamDiff, const void *betaParamDiff, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t dxDesc, void *dx, const miopenTensorDescriptor_t bnScaleBiasDiffDesc, const void *bnScale, void *resultBnScaleDiff, void *resultBnBiasDiff, double epsilon, const void *savedMean, const void *savedInvVariance) |
Execute backwards propagation layer for batch normalization. More... | |
miopenStatus_t | miopenBatchNormalizationBackward_V2 (miopenHandle_t handle, miopenBatchNormMode_t bn_mode, const void *alphaDataDiff, const void *betaDataDiff, const void *alphaParamDiff, const void *betaParamDiff, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t dxDesc, void *dx, const miopenTensorDescriptor_t scaleDesc, const miopenTensorDescriptor_t biasDesc, const miopenTensorDescriptor_t savedMeanDesc, const miopenTensorDescriptor_t savedVarDesc, const void *bnScale, void *resultBnScaleDiff, void *resultBnBiasDiff, double epsilon, const void *savedMean, const void *savedInvVariance) |
Execute backwards propagation layer for batch normalization. More... | |
miopenStatus_t | miopenCreateActivationDescriptor (miopenActivationDescriptor_t *activDesc) |
Creates the Activation descriptor object. More... | |
miopenStatus_t | miopenSetActivationDescriptor (const miopenActivationDescriptor_t activDesc, miopenActivationMode_t mode, double activAlpha, double activBeta, double activGamma) |
Sets the activation layer descriptor details. More... | |
miopenStatus_t | miopenGetActivationDescriptor (const miopenActivationDescriptor_t activDesc, miopenActivationMode_t *mode, double *activAlpha, double *activBeta, double *activGamma) |
Gets the activation layer descriptor details. More... | |
miopenStatus_t | miopenActivationForward (miopenHandle_t handle, const miopenActivationDescriptor_t activDesc, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y) |
Execute an activation forward layer. More... | |
miopenStatus_t | miopenActivationBackward (miopenHandle_t handle, const miopenActivationDescriptor_t activDesc, const void *alpha, const miopenTensorDescriptor_t yDesc, const void *y, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx) |
Execute a activation backwards layer. More... | |
miopenStatus_t | miopenDestroyActivationDescriptor (miopenActivationDescriptor_t activDesc) |
Destroys the activation descriptor object. More... | |
miopenStatus_t | miopenGLUForward (miopenHandle_t handle, const miopenTensorDescriptor_t inputDesc, const void *input, const miopenTensorDescriptor_t outputDesc, void *output, const uint32_t dim) |
Execute a GLU forward layer. More... | |
miopenStatus_t | miopenGLUBackward (miopenHandle_t handle, const miopenTensorDescriptor_t inputDesc, const void *input, const miopenTensorDescriptor_t outputGradDesc, const void *outputGrad, const miopenTensorDescriptor_t inputGradDesc, void *inputGrad, const uint32_t dim) |
Execute a GLU backward layer. More... | |
miopenStatus_t | miopenSoftmaxForward (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y) |
Execute a softmax forward layer. More... | |
miopenStatus_t | miopenSoftmaxBackward (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t yDesc, const void *y, const miopenTensorDescriptor_t dyDesc, const void *dy, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx) |
Execute a softmax backwards layer. More... | |
miopenStatus_t | miopenSoftmaxForward_V2 (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y, miopenSoftmaxAlgorithm_t algorithm, miopenSoftmaxMode_t mode) |
Execute a softmax forward layer with expanded modes and algorithms. More... | |
miopenStatus_t | miopenSoftmaxBackward_V2 (miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t yDesc, const void *y, const miopenTensorDescriptor_t dyDesc, const void *dy, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx, miopenSoftmaxAlgorithm_t algorithm, miopenSoftmaxMode_t mode) |
Execute a softmax backwards layer with expanded modes and algorithms. More... | |
MIOPEN_DECLARE_OBJECT (miopenFusionPlanDescriptor) | |
MIOpen fusion interface. More... | |
MIOPEN_DECLARE_OBJECT (miopenOperatorDescriptor) | |
MIOPEN_DECLARE_OBJECT (miopenOperatorArgs) | |
miopenStatus_t | miopenCreateFusionPlan (miopenFusionPlanDescriptor_t *fusePlanDesc, const miopenFusionDirection_t fuseDirection, const miopenTensorDescriptor_t inputDesc) |
Creates the kenrel fusion plan descriptor object. More... | |
miopenStatus_t | miopenDestroyFusionPlan (miopenFusionPlanDescriptor_t fusePlanDesc) |
Destroy the fusion plan descriptor object. More... | |
miopenStatus_t | miopenCompileFusionPlan (miopenHandle_t handle, miopenFusionPlanDescriptor_t fusePlanDesc) |
Compiles the fusion plan. More... | |
miopenStatus_t | miopenFusionPlanGetOp (miopenFusionPlanDescriptor_t fusePlanDesc, const int op_idx, miopenFusionOpDescriptor_t *op) |
Allows access to the operators in a fusion plan. More... | |
miopenStatus_t | miopenFusionPlanGetWorkSpaceSize (miopenHandle_t handle, miopenFusionPlanDescriptor_t fusePlanDesc, size_t *workSpaceSize, miopenConvFwdAlgorithm_t algo) |
Query the workspace size required for the fusion plan. More... | |
miopenStatus_t | miopenFusionPlanConvolutionGetAlgo (miopenFusionPlanDescriptor_t fusePlanDesc, const int requestAlgoCount, int *returnedAlgoCount, miopenConvFwdAlgorithm_t *returnedAlgos) |
Returns the supported algorithms for the convolution operator in the Fusion Plan. More... | |
miopenStatus_t | miopenFusionPlanConvolutionSetAlgo (miopenFusionPlanDescriptor_t fusePlanDesc, miopenConvFwdAlgorithm_t algo) |
Requests the fusion runtime to choose a particular algorithm for the added convolution operation. More... | |
miopenStatus_t | miopenCreateOpConvForward (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *convOp, miopenConvolutionDescriptor_t convDesc, const miopenTensorDescriptor_t wDesc) |
Creates forward convolution operator. More... | |
miopenStatus_t | miopenCreateOpActivationForward (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *activFwdOp, miopenActivationMode_t mode) |
Creates a forward activation operator. More... | |
miopenStatus_t | miopenCreateOpActivationBackward (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *activBwdOp, miopenActivationMode_t mode) |
Creates a backward activation operator. More... | |
miopenStatus_t | miopenCreateOpBiasForward (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *biasOp, const miopenTensorDescriptor_t bDesc) |
Creates a forward bias operator. More... | |
miopenStatus_t | miopenCreateOpBatchNormInference (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *bnOp, const miopenBatchNormMode_t bn_mode, const miopenTensorDescriptor_t bnScaleBiasMeanVarDesc) |
Creates a forward inference batch normalization operator. More... | |
miopenStatus_t | miopenCreateOpBatchNormForward (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *bnFwdOp, const miopenBatchNormMode_t bn_mode, bool runningMeanVariance) |
Creates a forward training batch normalization operator. More... | |
miopenStatus_t | miopenCreateOpBatchNormBackward (miopenFusionPlanDescriptor_t fusePlanDesc, miopenFusionOpDescriptor_t *bnBwdOp, const miopenBatchNormMode_t bn_mode) |
Creates a back propagation batch normalization operator. More... | |
miopenStatus_t | miopenCreateOperatorArgs (miopenOperatorArgs_t *args) |
Creates an operator argument object. More... | |
miopenStatus_t | miopenDestroyOperatorArgs (miopenOperatorArgs_t args) |
Destroys an operator argument object. More... | |
miopenStatus_t | miopenSetOpArgsConvForward (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t convOp, const void *alpha, const void *beta, const void *w) |
Sets the arguments for forward convolution op. More... | |
miopenStatus_t | miopenSetOpArgsActivForward (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t activFwdOp, const void *alpha, const void *beta, double activAlpha, double activBeta, double activGamma) |
Sets the arguments for forward activation op. More... | |
miopenStatus_t | miopenSetOpArgsActivBackward (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t activBwdOp, const void *alpha, const void *beta, const void *y, const void *reserved, double activAlpha, double activBeta, double activGamma) |
Sets the arguments for backward activation op. More... | |
miopenStatus_t | miopenSetOpArgsBatchNormInference (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t bnOp, const void *alpha, const void *beta, const void *bnScale, const void *bnBias, const void *estimatedMean, const void *estimatedVariance, double epsilon) |
Sets the arguments for inference batch normalization op. More... | |
miopenStatus_t | miopenSetOpArgsBatchNormForward (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t bnOp, const void *alpha, const void *beta, const void *bnScale, const void *bnBias, void *savedMean, void *savedInvVariance, void *runningMean, void *runningVariance, double expAvgFactor, double epsilon) |
Sets the arguments for forward batch normalization op. More... | |
miopenStatus_t | miopenSetOpArgsBatchNormBackward (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t bnOp, const void *alpha, const void *beta, const void *x, const void *bnScale, const void *bnBias, void *resultBnScaleDiff, void *resultBnBiasDiff, const void *savedMean, const void *savedInvVariance) |
Sets the arguments for backward batch normalization op. More... | |
miopenStatus_t | miopenSetOpArgsBiasForward (miopenOperatorArgs_t args, const miopenFusionOpDescriptor_t biasOp, const void *alpha, const void *beta, const void *bias) |
Sets the arguments for forward bias op. More... | |
miopenStatus_t | miopenExecuteFusionPlan (const miopenHandle_t handle, const miopenFusionPlanDescriptor_t fusePlanDesc, const miopenTensorDescriptor_t inputDesc, const void *input, const miopenTensorDescriptor_t outputDesc, void *output, miopenOperatorArgs_t args) |
Executes the fusion plan. More... | |
miopenStatus_t | miopenConvolutionBiasActivationForward (miopenHandle_t handle, const void *alpha1, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t wDesc, const void *w, const miopenConvolutionDescriptor_t convDesc, miopenConvFwdAlgorithm_t algo, void *workspace, size_t workspaceSizeInBytes, const void *alpha2, const miopenTensorDescriptor_t zDesc, const void *z, const miopenTensorDescriptor_t biasDesc, const void *bias, const miopenActivationDescriptor_t activationDesc, const miopenTensorDescriptor_t yDesc, void *y) |
Prepares and executes the Convlution+Bias+Activation Fusion. More... | |
miopenStatus_t | miopenCreateRNNDescriptor (miopenRNNDescriptor_t *rnnDesc) |
Create a RNN layer Descriptor. More... | |
miopenStatus_t | miopenGetRNNDescriptor (miopenRNNDescriptor_t rnnDesc, miopenRNNMode_t *rnnMode, miopenRNNAlgo_t *algoMode, miopenRNNInputMode_t *inputMode, miopenRNNDirectionMode_t *dirMode, miopenRNNBiasMode_t *biasMode, int *hiddenSize, int *layer) |
Retrieves a RNN layer descriptor's details. More... | |
miopenStatus_t | miopenGetRNNDescriptor_V2 (miopenRNNDescriptor_t rnnDesc, int *hiddenSize, int *layer, miopenDropoutDescriptor_t *dropoutDesc, miopenRNNInputMode_t *inputMode, miopenRNNDirectionMode_t *dirMode, miopenRNNMode_t *rnnMode, miopenRNNBiasMode_t *biasMode, miopenRNNAlgo_t *algoMode, miopenDataType_t *dataType) |
Retrieves a RNN layer descriptor's details version 2. This version enables retrieving information of the dropout descriptor of the rnn descriptor. More... | |
miopenStatus_t | miopenDestroyRNNDescriptor (miopenRNNDescriptor_t rnnDesc) |
Destroys the tensor descriptor object. More... | |
miopenStatus_t | miopenSetRNNDescriptor (miopenRNNDescriptor_t rnnDesc, const int hsize, const int nlayers, miopenRNNInputMode_t inMode, miopenRNNDirectionMode_t direction, miopenRNNMode_t rnnMode, miopenRNNBiasMode_t biasMode, miopenRNNAlgo_t algo, miopenDataType_t dataType) |
Set the details of the RNN descriptor. More... | |
miopenStatus_t | miopenSetRNNDescriptor_V2 (miopenRNNDescriptor_t rnnDesc, const int hsize, const int nlayers, miopenDropoutDescriptor_t dropoutDesc, miopenRNNInputMode_t inMode, miopenRNNDirectionMode_t direction, miopenRNNMode_t rnnMode, miopenRNNBiasMode_t biasMode, miopenRNNAlgo_t algo, miopenDataType_t dataType) |
Set the details of the RNN descriptor version 2. This version enables the use of dropout in rnn. More... | |
miopenStatus_t | miopenSetRNNDataSeqTensorDescriptor (miopenSeqTensorDescriptor_t seqTensorDesc, miopenDataType_t dataType, miopenRNNBaseLayout_t layout, int maxSequenceLen, int batchSize, int vectorSize, const int *sequenceLenArray, void *paddingMarker) |
Set shape of RNN seqData tensor. More... | |
miopenStatus_t | miopenGetRNNDataSeqTensorDescriptor (miopenSeqTensorDescriptor_t seqTensorDesc, miopenDataType_t *dataType, miopenRNNBaseLayout_t *layout, int *maxSequenceLen, int *batchSize, int *vectorSize, int sequenceLenArrayLimit, int *sequenceLenArray, void *paddingMarker) |
Get shape of RNN seqData tensor. More... | |
miopenStatus_t | miopenGetRNNWorkspaceSize (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, const int sequenceLen, const miopenTensorDescriptor_t *xDesc, size_t *numBytes) |
Query the amount of memory required to execute the RNN layer. More... | |
miopenStatus_t | miopenGetRNNTrainingReserveSize (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int sequenceLen, const miopenTensorDescriptor_t *xDesc, size_t *numBytes) |
Query the amount of memory required for RNN training. More... | |
miopenStatus_t | miopenGetRNNTempSpaceSizes (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, miopenSeqTensorDescriptor_t xDesc, miopenRNNFWDMode_t fwdMode, size_t *workSpaceSize, size_t *reserveSpaceSize) |
Query the amount of additional memory required for this RNN layer execution. More... | |
miopenStatus_t | miopenGetRNNParamsSize (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, miopenTensorDescriptor_t xDesc, size_t *numBytes, miopenDataType_t dtype) |
Query the amount of parameter memory required for RNN training. More... | |
miopenStatus_t | miopenGetRNNParamsDescriptor (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, miopenTensorDescriptor_t xDesc, miopenTensorDescriptor_t wDesc, miopenDataType_t dtype) |
Obtain a weight tensor descriptor for RNNs. More... | |
miopenStatus_t | miopenGetRNNInputTensorSize (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int seqLen, miopenTensorDescriptor_t *xDesc, size_t *numBytes) |
Obtain a the size in bytes of the RNN input tensor. More... | |
miopenStatus_t | miopenGetRNNHiddenTensorSize (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int seqLen, miopenTensorDescriptor_t *xDesc, size_t *numBytes) |
Obtain a the size in bytes of the RNN hidden tensor. More... | |
miopenStatus_t | miopenGetRNNLayerParamSize (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, const int paramID, size_t *numBytes) |
Gets the number of bytes of a parameter matrix. More... | |
miopenStatus_t | miopenGetRNNLayerBiasSize (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int layer, const int biasID, size_t *numBytes) |
Gets the number of bytes of a bias. More... | |
miopenStatus_t | miopenGetRNNLayerParam (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, miopenTensorDescriptor_t wDesc, const void *w, const int paramID, miopenTensorDescriptor_t paramDesc, void *layerParam) |
Gets a weight matrix for a specific layer in an RNN stack. More... | |
miopenStatus_t | miopenGetRNNLayerBias (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, miopenTensorDescriptor_t wDesc, const void *w, const int biasID, miopenTensorDescriptor_t biasDesc, void *layerBias) |
Gets a bias for a specific layer in an RNN stack. More... | |
miopenStatus_t | miopenGetRNNLayerParamOffset (miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, const int paramID, miopenTensorDescriptor_t paramDesc, size_t *layerParamOffset) |
Gets an index offset for a specific weight matrix for a layer in the RNN stack. More... | |
miopenStatus_t | miopenGetRNNLayerBiasOffset (miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, const int biasID, miopenTensorDescriptor_t biasDesc, size_t *layerBiasOffset) |
Gets a bias index offset for a specific layer in an RNN stack. More... | |
miopenStatus_t | miopenSetRNNLayerParam (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, miopenTensorDescriptor_t wDesc, void *w, const int paramID, miopenTensorDescriptor_t paramDesc, const void *layerParam) |
Sets a weight matrix for a specific layer in an RNN stack. More... | |
miopenStatus_t | miopenSetRNNLayerBias (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int layer, miopenTensorDescriptor_t xDesc, miopenTensorDescriptor_t wDesc, void *w, const int biasID, miopenTensorDescriptor_t biasDesc, const void *layerBias) |
Sets a bias for a specific layer in an RNN stack. More... | |
miopenStatus_t | miopenSetRNNPaddingMode (miopenRNNDescriptor_t rnnDesc, miopenRNNPaddingMode_t paddingMode) |
Sets a bias for a specific layer in an RNN stack. More... | |
miopenStatus_t | miopenGetRNNPaddingMode (miopenRNNDescriptor_t rnnDesc, miopenRNNPaddingMode_t *paddingMode) |
This function retrieves the RNN padding mode from the RNN descriptor. More... | |
miopenStatus_t | miopenRNNForward (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, miopenRNNFWDMode_t fwdMode, const miopenSeqTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t hDesc, const void *hx, void *hy, const miopenTensorDescriptor_t cDesc, const void *cx, void *cy, const miopenSeqTensorDescriptor_t yDesc, void *y, const void *w, size_t weightSpaceSize, void *workSpace, size_t workSpaceNumBytes, void *reserveSpace, size_t reserveSpaceNumBytes) |
Execute forward training for recurrent layer. More... | |
miopenStatus_t | miopenRNNBackwardSeqData (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, const miopenSeqTensorDescriptor_t yDesc, const void *y, const void *dy, const miopenTensorDescriptor_t hDesc, const void *hx, const void *dhy, void *dhx, const miopenTensorDescriptor_t cDesc, const void *cx, const void *dcy, void *dcx, const miopenSeqTensorDescriptor_t xDesc, void *dx, const void *w, size_t weightSpaceSize, void *workSpace, size_t workSpaceNumBytes, void *reserveSpace, size_t reserveSpaceNumBytes) |
Execute backward data for recurrent layer. More... | |
miopenStatus_t | miopenRNNBackwardWeightsSeqTensor (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, const miopenSeqTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t hDesc, const void *hx, const miopenSeqTensorDescriptor_t yDesc, const void *y, void *dw, size_t weightSpaceSize, void *workSpace, size_t workSpaceNumBytes, const void *reserveSpace, size_t reserveSpaceNumBytes) |
Execute backward weights for recurrent layer. More... | |
miopenStatus_t | miopenRNNForwardTraining (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, const int sequenceLen, const miopenTensorDescriptor_t *xDesc, const void *x, const miopenTensorDescriptor_t hxDesc, const void *hx, const miopenTensorDescriptor_t cxDesc, const void *cx, const miopenTensorDescriptor_t wDesc, const void *w, const miopenTensorDescriptor_t *yDesc, void *y, const miopenTensorDescriptor_t hyDesc, void *hy, const miopenTensorDescriptor_t cyDesc, void *cy, void *workSpace, size_t workSpaceNumBytes, void *reserveSpace, size_t reserveSpaceNumBytes) |
Execute forward training for recurrent layer. More... | |
miopenStatus_t | miopenRNNBackwardData (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, const int sequenceLen, const miopenTensorDescriptor_t *yDesc, const void *y, const miopenTensorDescriptor_t *dyDesc, const void *dy, const miopenTensorDescriptor_t dhyDesc, const void *dhy, const miopenTensorDescriptor_t dcyDesc, const void *dcy, const miopenTensorDescriptor_t wDesc, const void *w, const miopenTensorDescriptor_t hxDesc, const void *hx, const miopenTensorDescriptor_t cxDesc, const void *cx, const miopenTensorDescriptor_t *dxDesc, void *dx, const miopenTensorDescriptor_t dhxDesc, void *dhx, const miopenTensorDescriptor_t dcxDesc, void *dcx, void *workSpace, size_t workSpaceNumBytes, void *reserveSpace, size_t reserveSpaceNumBytes) |
Execute backward data for recurrent layer. More... | |
miopenStatus_t | miopenRNNBackwardWeights (miopenHandle_t handle, const miopenRNNDescriptor_t rnnDesc, const int sequenceLen, const miopenTensorDescriptor_t *xDesc, const void *x, const miopenTensorDescriptor_t hxDesc, const void *hx, const miopenTensorDescriptor_t *yDesc, const void *y, const miopenTensorDescriptor_t dwDesc, void *dw, void *workSpace, size_t workSpaceNumBytes, const void *reserveSpace, size_t reserveSpaceNumBytes) |
Execute backward weights for recurrent layer. More... | |
miopenStatus_t | miopenRNNForwardInference (miopenHandle_t handle, miopenRNNDescriptor_t rnnDesc, const int sequenceLen, const miopenTensorDescriptor_t *xDesc, const void *x, const miopenTensorDescriptor_t hxDesc, const void *hx, const miopenTensorDescriptor_t cxDesc, const void *cx, const miopenTensorDescriptor_t wDesc, const void *w, const miopenTensorDescriptor_t *yDesc, void *y, const miopenTensorDescriptor_t hyDesc, void *hy, const miopenTensorDescriptor_t cyDesc, void *cy, void *workSpace, size_t workSpaceNumBytes) |
Execute forward inference for RNN layer. More... | |
miopenStatus_t | miopenCreateCTCLossDescriptor (miopenCTCLossDescriptor_t *ctcLossDesc) |
Create a CTC loss function Descriptor. More... | |
miopenStatus_t | miopenGetCTCLossDescriptor (miopenCTCLossDescriptor_t ctcLossDesc, miopenDataType_t *dataType, int *blank_label_id, bool *apply_softmax_layer) |
Retrieves a CTC loss function descriptor's details. More... | |
miopenStatus_t | miopenDestroyCTCLossDescriptor (miopenCTCLossDescriptor_t ctcLossDesc) |
Destroys a CTC loss function descriptor object. More... | |
miopenStatus_t | miopenSetCTCLossDescriptor (miopenCTCLossDescriptor_t ctcLossDesc, miopenDataType_t dataType, const int blank_label_id, bool apply_softmax_layer) |
Set the details of a CTC loss function descriptor. More... | |
miopenStatus_t | miopenGetCTCLossWorkspaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t probsDesc, const miopenTensorDescriptor_t gradientsDesc, const int *labels, const int *labelLengths, const int *inputLengths, miopenCTCLossAlgo_t algo, const miopenCTCLossDescriptor_t ctcLossDesc, size_t *workSpaceSize) |
Query the amount of memory required to execute miopenCTCLoss. More... | |
miopenStatus_t | miopenCTCLoss (miopenHandle_t handle, const miopenTensorDescriptor_t probsDesc, const void *probs, const int *labels, const int *labelLengths, const int *inputLengths, void *losses, const miopenTensorDescriptor_t gradientsDesc, void *gradients, miopenCTCLossAlgo_t algo, const miopenCTCLossDescriptor_t ctcLossDesc, void *workSpace, size_t workSpaceSize) |
Execute forward inference for CTCLoss layer. More... | |
miopenStatus_t | miopenCreateDropoutDescriptor (miopenDropoutDescriptor_t *dropoutDesc) |
Creates the dropout descriptor object. More... | |
miopenStatus_t | miopenDestroyDropoutDescriptor (miopenDropoutDescriptor_t dropoutDesc) |
Destroys the dropout descriptor object. More... | |
miopenStatus_t | miopenDropoutGetReserveSpaceSize (const miopenTensorDescriptor_t xDesc, size_t *reserveSpaceSizeInBytes) |
Query the amount of memory required to run dropout. More... | |
miopenStatus_t | miopenDropoutGetStatesSize (miopenHandle_t handle, size_t *stateSizeInBytes) |
Query the amount of memory required to store the states of the random number generators. More... | |
miopenStatus_t | miopenGetDropoutDescriptor (miopenDropoutDescriptor_t dropoutDesc, miopenHandle_t handle, float *dropout, void **states, unsigned long long *seed, bool *use_mask, bool *state_evo, miopenRNGType_t *rng_mode) |
Get the details of the dropout descriptor. More... | |
miopenStatus_t | miopenRestoreDropoutDescriptor (miopenDropoutDescriptor_t dropoutDesc, miopenHandle_t handle, float dropout, void *states, size_t stateSizeInBytes, unsigned long long seed, bool use_mask, bool state_evo, miopenRNGType_t rng_mode) |
Restore the dropout descriptor to a saved state. More... | |
miopenStatus_t | miopenSetDropoutDescriptor (miopenDropoutDescriptor_t dropoutDesc, miopenHandle_t handle, float dropout, void *states, size_t stateSizeInBytes, unsigned long long seed, bool use_mask, bool state_evo, miopenRNGType_t rng_mode) |
Initialize the dropout descriptor. More... | |
miopenStatus_t | miopenDropoutForward (miopenHandle_t handle, const miopenDropoutDescriptor_t dropoutDesc, const miopenTensorDescriptor_t noise_shape, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t yDesc, void *y, void *reserveSpace, size_t reserveSpaceSizeInBytes) |
Execute forward dropout operation. More... | |
miopenStatus_t | miopenDropoutBackward (miopenHandle_t handle, const miopenDropoutDescriptor_t dropoutDesc, const miopenTensorDescriptor_t noise_shape, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t dxDesc, void *dx, void *reserveSpace, size_t reserveSpaceSizeInBytes) |
Execute backward dropout operation. More... | |
miopenStatus_t | miopenCreateReduceTensorDescriptor (miopenReduceTensorDescriptor_t *reduceTensorDesc) |
Creates the ReduceTensor descriptor object. More... | |
miopenStatus_t | miopenDestroyReduceTensorDescriptor (miopenReduceTensorDescriptor_t reduceTensorDesc) |
Destroy the ReduceTensor descriptor object. More... | |
miopenStatus_t | miopenSetReduceTensorDescriptor (miopenReduceTensorDescriptor_t reduceTensorDesc, miopenReduceTensorOp_t reduceTensorOp, miopenDataType_t reduceTensorCompType, miopenNanPropagation_t reduceTensorNanOpt, miopenReduceTensorIndices_t reduceTensorIndices, miopenIndicesType_t reduceTensorIndicesType) |
Initialize a ReduceTensor descriptor object. More... | |
miopenStatus_t | miopenGetReduceTensorDescriptor (const miopenReduceTensorDescriptor_t reduceTensorDesc, miopenReduceTensorOp_t *reduceTensorOp, miopenDataType_t *reduceTensorCompType, miopenNanPropagation_t *reduceTensorNanOpt, miopenReduceTensorIndices_t *reduceTensorIndices, miopenIndicesType_t *reduceTensorIndicesType) |
Query a ReduceTensor descriptor object. More... | |
miopenStatus_t | miopenGetReductionIndicesSize (miopenHandle_t handle, const miopenReduceTensorDescriptor_t reduceTensorDesc, const miopenTensorDescriptor_t aDesc, const miopenTensorDescriptor_t cDesc, size_t *sizeInBytes) |
Helper function to query the minimum index space size required by the ReduceTensor call. More... | |
miopenStatus_t | miopenGetReductionWorkspaceSize (miopenHandle_t handle, const miopenReduceTensorDescriptor_t reduceTensorDesc, const miopenTensorDescriptor_t aDesc, const miopenTensorDescriptor_t cDesc, size_t *sizeInBytes) |
Helper function to query the minimum workspace size required by the ReduceTensor call. More... | |
miopenStatus_t | miopenReduceTensor (miopenHandle_t handle, const miopenReduceTensorDescriptor_t reduceTensorDesc, void *indices, size_t indicesSizeInBytes, void *workspace, size_t workspaceSizeInBytes, const void *alpha, const miopenTensorDescriptor_t aDesc, const void *A, const void *beta, const miopenTensorDescriptor_t cDesc, void *C) |
TensorReduce function doing reduction on tensor A by implementing C = alpha * reduceOp(A) More... | |
MIOPEN_DECLARE_OBJECT (miopenProblem) | |
Describes a problem for different miopen operations. More... | |
miopenStatus_t | miopenCreateConvProblem (miopenProblem_t *problem, miopenConvolutionDescriptor_t operatorDesc, miopenProblemDirection_t direction) |
Initializes a problem object describing a convolution operation. More... | |
miopenStatus_t | miopenCreateMhaProblem (miopenProblem_t *problem, miopenMhaDescriptor_t operatorDesc, miopenProblemDirection_t direction) |
miopenStatus_t | miopenCreateMhaDescriptor (miopenMhaDescriptor_t *mhaDesc) |
Creates the mha descriptor object. More... | |
miopenStatus_t | miopenSetMhaDescriptor (miopenMhaDescriptor_t mhaDesc, float scale) |
Sets the Mha descriptor details. More... | |
miopenStatus_t | miopenGetMhaDescriptor (miopenMhaDescriptor_t mhaDesc, float *scale) |
Gets the Mha descriptor details. More... | |
miopenStatus_t | miopenCreateSoftmaxDescriptor (miopenSoftmaxDescriptor_t *softmaxDesc) |
Creates the Softmax descriptor object. More... | |
miopenStatus_t | miopenSetSoftmaxDescriptor (miopenSoftmaxDescriptor_t softmaxDesc, float alpha, float beta, miopenSoftmaxAlgorithm_t algorithm, miopenSoftmaxMode_t mode) |
Sets the softmax descriptor details. More... | |
miopenStatus_t | miopenGetSoftmaxDescriptor (const miopenSoftmaxDescriptor_t softmaxDesc, float *alpha, float *beta, miopenSoftmaxAlgorithm_t *algorithm, miopenSoftmaxMode_t *mode) |
Gets the softmax layer descriptor details. More... | |
miopenStatus_t | miopenDestroyProblem (miopenProblem_t problem) |
Destroys a problem object. More... | |
miopenStatus_t | miopenSetProblemTensorDescriptor (miopenProblem_t problem, miopenTensorArgumentId_t id, const miopenTensorDescriptor_t descriptor) |
Sets a tensor descriptor for the specified argument. More... | |
MIOPEN_DECLARE_OBJECT (miopenFindOptions) | |
The miopenFindOptions allows the user to configure how find will be used. More... | |
miopenStatus_t | miopenCreateFindOptions (miopenFindOptions_t *options) |
Initializes miopenFindOptions object. More... | |
miopenStatus_t | miopenDestroyFindOptions (miopenFindOptions_t options) |
Destroys miopenFindOptions object. More... | |
miopenStatus_t | miopenSetFindOptionTuning (miopenFindOptions_t options, int value) |
Sets the tuning find option. Default value is zero. More... | |
miopenStatus_t | miopenSetFindOptionResultsOrder (miopenFindOptions_t options, miopenFindResultsOrder_t value) |
Sets the results order find option. Default value is miopenFindResultsOrderByTime. More... | |
miopenStatus_t | miopenSetFindOptionWorkspaceLimit (miopenFindOptions_t options, size_t value) |
Sets the workspace limit find option. Default value is maximum of size_t. More... | |
miopenStatus_t | miopenSetFindOptionPreallocatedWorkspace (miopenFindOptions_t options, void *buffer, size_t size) |
Attaches the preallocated workspace to find options. Allocated by the library by default. More... | |
miopenStatus_t | miopenSetFindOptionPreallocatedTensor (miopenFindOptions_t options, miopenTensorArgumentId_t id, void *buffer) |
Attaches a preallocated tensor to find options. If not used, buffers are allocated by MIOpen internally, which is not recommended. More... | |
miopenStatus_t | miopenSetFindOptionAttachBinaries (miopenFindOptions_t options, unsigned attach) |
Forces library to attach kernel binaries to solutions for later saving. This allows zero lookup miopenRunSolution calls after miopenLoadSolution. Default value is 0. More... | |
MIOPEN_DECLARE_OBJECT (miopenSolution) | |
The miopenSolution object describes a prepared solution. More... | |
miopenStatus_t | miopenFindSolutions (miopenHandle_t handle, miopenProblem_t problem, miopenFindOptions_t options, miopenSolution_t *solutions, size_t *numSolutions, size_t maxSolutions) |
Finds solutions to a problem by running different applicable solutions. Memory is automatically allocated. More... | |
miopenStatus_t | miopenRunSolution (miopenHandle_t handle, miopenSolution_t solution, size_t nInputs, const miopenTensorArgument_t *tensors, void *workspace, size_t workspaceSize) |
Runs the solution using the passed in buffers. More... | |
miopenStatus_t | miopenDestroySolution (miopenSolution_t solution) |
Destroys solution object. More... | |
miopenStatus_t | miopenLoadSolution (miopenSolution_t *solution, const char *data, size_t size) |
Loads solution object from binary data. More... | |
miopenStatus_t | miopenSaveSolution (miopenSolution_t solution, char *data) |
Saves a solution object as binary data. More... | |
miopenStatus_t | miopenGetSolutionSize (miopenSolution_t solution, size_t *size) |
Reads the expected size of a solution. More... | |
miopenStatus_t | miopenGetSolutionWorkspaceSize (miopenSolution_t solution, size_t *workspaceSize) |
Reads the amount of workspace required to exectute the solution. More... | |
miopenStatus_t | miopenGetSolutionTime (miopenSolution_t solution, float *time) |
Reads the time spent to execute the solution the last it was run. More... | |
miopenStatus_t | miopenGetSolutionSolverId (miopenSolution_t solution, uint64_t *solverId) |
Reads id of the solver referred by the solution. More... | |
miopenStatus_t | miopenGetSolverIdConvAlgorithm (uint64_t solverId, miopenConvAlgorithm_t *result) |
Gets the convolution algorithm implemented by a solver. More... | |
miopenStatus_t | miopenCreateActivationProblem (miopenProblem_t *problem, miopenActivationDescriptor_t operatorDesc, miopenProblemDirection_t direction) |
Initializes a problem object describing an activation operation. More... | |
miopenStatus_t | miopenCreateBatchnormProblem (miopenProblem_t *problem, miopenBatchNormMode_t mode, bool runningMeanVariance, miopenProblemDirection_t direction) |
Initializes a problem object describing an activation operation. More... | |
miopenStatus_t | miopenFuseProblems (miopenProblem_t problem1, miopenProblem_t problem2) |
miopenStatus_t | miopenCreateBiasProblem (miopenProblem_t *problem, miopenProblemDirection_t direction) |
Initializes a problem object describing an bias operation. More... | |
miopenStatus_t | miopenCreateSoftmaxProblem (miopenProblem_t *problem, miopenSoftmaxDescriptor_t operatorDesc, miopenProblemDirection_t direction) |
Initializes a problem object describing a softmax operation. More... | |
miopenStatus_t | miopenGetReduceCalculationWorkspaceSize (miopenHandle_t handle, const miopenTensorDescriptor_t xDesc, const int32_t dim, const miopenReduceCalculationOp_t reduceCalculationOp, const miopenTensorDescriptor_t reduceDesc, size_t *sizeInBytes) |
Helper function to query the minimum workspace size required by the ReduceTensor call. More... | |
miopenStatus_t | miopenReduceCalculationForward (miopenHandle_t handle, miopenReduceCalculationNanPropagation_t nanPropagation, void *workspace, size_t workspaceSizeInBytes, const miopenTensorDescriptor_t xDesc, const void *x, const int32_t dim, const miopenReduceCalculationOp_t reduceCalculationOp, const miopenTensorDescriptor_t reduceDesc, void *y) |
Execute a reducecalculation forward layer. More... | |
miopenStatus_t | miopenReduceExtremeForward (miopenHandle_t handle, const miopenTensorDescriptor_t xDesc, const void *x, const int32_t dim, const miopenReduceExtremeOp_t reduceExtremeOp, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t indiceDesc, void *indice) |
Find the the extreme (minimum, maximum) value and index of a tensor across Dimension. More... | |
miopenStatus_t | miopenGroupNormForward (miopenHandle_t handle, miopenNormMode_t mode, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t weightDesc, const void *weight, const miopenTensorDescriptor_t biasDesc, const void *bias, const uint64_t num_groups, const float epsilon, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t meanDesc, void *mean, const miopenTensorDescriptor_t rstdDesc, void *rstd) |
Execute a groupnorm forward layer. More... | |
miopenStatus_t | miopenAddLayerNormForward (miopenHandle_t handle, miopenNormMode_t mode, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t x2Desc, const void *x2, const miopenTensorDescriptor_t weightDesc, const void *weight, const miopenTensorDescriptor_t biasDesc, const void *bias, const float epsilon, const int32_t normalized_dim, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t meanDesc, void *mean, const miopenTensorDescriptor_t rstdDesc, void *rstd) |
Execute a add and layernorm forward layer. More... | |
miopenStatus_t | miopenT5LayerNormForward (miopenHandle_t handle, miopenNormMode_t mode, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t weightDesc, const void *weight, const float epsilon, const miopenTensorDescriptor_t yDesc, void *y, const miopenTensorDescriptor_t rstdDesc, void *rstd) |
Execute a T5layernorm forward layer. More... | |
miopenStatus_t | miopenGetT5LayerNormBackwardWorkspaceSize (miopenHandle_t handle, miopenNormMode_t mode, const miopenTensorDescriptor_t dyDesc, const miopenTensorDescriptor_t xDesc, const miopenTensorDescriptor_t weightDesc, const miopenTensorDescriptor_t rstdDesc, const miopenTensorDescriptor_t dxDesc, const miopenTensorDescriptor_t dwDesc, size_t *sizeInBytes) |
Helper function to query the minimum workspace size required by the T5layernorm backward call. More... | |
miopenStatus_t | miopenT5LayerNormBackward (miopenHandle_t handle, miopenNormMode_t mode, void *workspace, size_t workspaceSizeInBytes, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t weightDesc, const void *weight, const miopenTensorDescriptor_t rstdDesc, const void *rstd, const miopenTensorDescriptor_t dxDesc, void *dx, const miopenTensorDescriptor_t dwDesc, void *dw) |
Execute a T5layernorm backward layer. More... | |
miopenStatus_t | miopenBackendCreateDescriptor (miopenBackendDescriptorType_t descriptorType, miopenBackendDescriptor_t *descriptor) |
Backend descriptor. More... | |
miopenStatus_t | miopenBackendSetAttribute (miopenBackendDescriptor_t descriptor, miopenBackendAttributeName_t attributeName, miopenBackendAttributeType_t attributeType, int64_t elementCount, void *arrayOfElements) |
Sets an attribute of a descriptor. More... | |
miopenStatus_t | miopenBackendFinalize (miopenBackendDescriptor_t descriptor) |
Finalizes a backend descriptor. More... | |
miopenStatus_t | miopenBackendGetAttribute (miopenBackendDescriptor_t descriptor, miopenBackendAttributeName_t attributeName, miopenBackendAttributeType_t attributeType, int64_t requestedElementCount, int64_t *elementCount, void *arrayOfElements) |
Retrieves backend descriptor's attribute. More... | |
miopenStatus_t | miopenBackendExecute (miopenHandle_t handle, miopenBackendDescriptor_t executionPlan, miopenBackendDescriptor_t variantPack) |
Executes a graph. More... | |
miopenStatus_t | miopenBackendDestroyDescriptor (miopenBackendDescriptor_t descriptor) |
Destroys an instance of miopenBackendDescriptor_t. More... | |
miopenStatus_t | miopenBackendInitialize (miopenBackendDescriptor_t descriptor, miopenBackendDescriptorType_t descriptorType, size_t sizeInBytes) |
Repurposes an instance of miopenBackendDescriptor_t. More... | |
miopenStatus_t | miopenFusedAdam (miopenHandle_t handle, const miopenTensorDescriptor_t paramDesc, void *param, const miopenTensorDescriptor_t gradDesc, const void *grad, const miopenTensorDescriptor_t expAvgDesc, void *expAvg, const miopenTensorDescriptor_t expAvgSqDesc, void *expAvgSq, const miopenTensorDescriptor_t maxExpAvgSqDesc, void *maxExpAvgSq, const miopenTensorDescriptor_t stateStepDesc, void *stateStep, const unsigned int state_step, const float lr, const float beta1, const float beta2, const float weight_decay, const float eps, const bool amsgrad, const bool maximize, const bool adamw, const miopenTensorDescriptor_t gradScaleDesc, const void *gradScale, const miopenTensorDescriptor_t foundInfDesc, const void *foundInf) |
Perform Fused Adam optimization for a single tensor (Adaptive Moment Estimation). More... | |
miopenStatus_t | miopenFusedAdamWithOutput (miopenHandle_t handle, const miopenTensorDescriptor_t paramInDesc, void *paramIn, const miopenTensorDescriptor_t paramOutDesc, void *paramOut, const miopenTensorDescriptor_t paramOutFloat16Desc, void *paramOutFloat16, const miopenTensorDescriptor_t gradInDesc, const void *gradIn, const miopenTensorDescriptor_t expAvgInDesc, void *expAvgIn, const miopenTensorDescriptor_t expAvgOutDesc, void *expAvgOut, const miopenTensorDescriptor_t expAvgSqInDesc, void *expAvgSqIn, const miopenTensorDescriptor_t expAvgSqOutDesc, void *expAvgSqOut, const miopenTensorDescriptor_t maxExpAvgSqInDesc, void *maxExpAvgSqIn, const miopenTensorDescriptor_t maxExpAvgSqOutDesc, void *maxExpAvgSqOut, const miopenTensorDescriptor_t stateStepInDesc, void *stateStepIn, const miopenTensorDescriptor_t stateStepOutDesc, void *stateStepOut, const unsigned int state_step, const float lr, const float beta1, const float beta2, const float weight_decay, const float eps, const bool amsgrad, const bool maximize, const bool adamw, const miopenTensorDescriptor_t gradScaleDesc, const void *gradScale, const miopenTensorDescriptor_t foundInfDesc, const void *foundInf) |
Execute single tensor Adam optimization and receive the result in a separate output tensor. More... | |
miopenStatus_t | miopenTransformersAdamW (miopenHandle_t handle, const miopenTensorDescriptor_t paramDesc, void *param, const miopenTensorDescriptor_t gradDesc, const void *grad, const miopenTensorDescriptor_t expAvgDesc, void *expAvg, const miopenTensorDescriptor_t expAvgSqDesc, void *expAvgSq, const miopenTensorDescriptor_t stateStepDesc, void *stateStep, const unsigned int state_step, const float lr, const float beta1, const float beta2, const float weight_decay, const float eps, const bool correct_bias, const miopenTensorDescriptor_t gradScaleDesc, const void *gradScale, const miopenTensorDescriptor_t foundInfDesc, const void *foundInf) |
Implements Adam algorithm with weight decay fix as introduced in Decoupled Weight Decay Regularization. This is the fused kernel version of AdamW included in the Hugging Face Transformers module. More... | |
miopenStatus_t | miopenTransformersAdamWWithOutput (miopenHandle_t handle, const miopenTensorDescriptor_t paramInDesc, void *paramIn, const miopenTensorDescriptor_t paramOutDesc, void *paramOut, const miopenTensorDescriptor_t paramOutFloat16Desc, void *paramOutFloat16, const miopenTensorDescriptor_t gradInDesc, const void *gradIn, const miopenTensorDescriptor_t expAvgInDesc, void *expAvgIn, const miopenTensorDescriptor_t expAvgOutDesc, void *expAvgOut, const miopenTensorDescriptor_t expAvgSqInDesc, void *expAvgSqIn, const miopenTensorDescriptor_t expAvgSqOutDesc, void *expAvgSqOut, const miopenTensorDescriptor_t stateStepInDesc, void *stateStepIn, const miopenTensorDescriptor_t stateStepOutDesc, void *stateStepOut, const unsigned int state_step, const float lr, const float beta1, const float beta2, const float weight_decay, const float eps, const float step_size, const bool correct_bias, const miopenTensorDescriptor_t gradScaleDesc, const void *gradScale, const miopenTensorDescriptor_t foundInfDesc, const void *foundInf) |
Execute single tensor Adam optimization and receive the result in a separate output tensor. More... | |
miopenStatus_t | miopenGetGetitemWorkspaceSize (miopenHandle_t handle, uint32_t indexCount, const miopenTensorDescriptor_t *indexDescs, size_t *sizeInBytes) |
Helper function to query the minimum workspace size required by the getitem call. More... | |
miopenStatus_t | miopenGetitemBackward (miopenHandle_t handle, void *workspace, size_t workspaceSizeInBytes, const miopenTensorDescriptor_t dyDesc, const void *dy, uint32_t indexCount, const miopenTensorDescriptor_t *indexDescs, const void *const *indexs, const miopenTensorDescriptor_t dxDesc, void *dx, const miopenTensorDescriptor_t errorDesc, void *error, uint32_t dimCount, const int32_t *dims, uint32_t sliceCount, const int32_t *slices, uint32_t offset) |
Execute a getitem backward layer. More... | |
miopenStatus_t | miopenRoPEForward (miopenHandle_t handle, const miopenTensorDescriptor_t xDesc, const void *x, const miopenTensorDescriptor_t cosDesc, const void *cos, const miopenTensorDescriptor_t sinDesc, const void *sin, const miopenTensorDescriptor_t yDesc, void *y) |
Execute a rope forward layer. More... | |
miopenStatus_t | miopenRoPEBackward (miopenHandle_t handle, const miopenTensorDescriptor_t dyDesc, const void *dy, const miopenTensorDescriptor_t cosDesc, const void *cos, const miopenTensorDescriptor_t sinDesc, const void *sin, const miopenTensorDescriptor_t dxDesc, void *dx) |
Execute a rope backward layer. More... | |
miopenStatus_t | miopenKthvalueForward (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, const void *input, miopenTensorDescriptor_t outputDesc, void *output, miopenTensorDescriptor_t indicesDesc, size_t *indices, size_t k, int32_t dim=-1, bool keepDim=false) |
Execute a Kthvalue forward layer. More... | |
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... | |
miopenStatus_t | miopenGetSoftMarginLossForwardWorkspaceSize (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, miopenTensorDescriptor_t targetDesc, miopenTensorDescriptor_t outputDesc, miopenLossReductionMode_t reduction, size_t *sizeInBytes) |
Helper function to query the minimum workspace size required by the SoftMarginLossForward call. More... | |
miopenStatus_t | miopenSoftMarginLossForward (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, const void *input, miopenTensorDescriptor_t targetDesc, const void *target, miopenTensorDescriptor_t outputDesc, void *output, miopenLossReductionMode_t reduction, void *workspace=nullptr, size_t workspaceSizeInBytes=0) |
Execute a SoftMarginLoss forward layer. More... | |
miopenStatus_t | miopenSoftMarginLossBackward (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, const void *input, miopenTensorDescriptor_t targetDesc, const void *target, miopenTensorDescriptor_t doutputDesc, const void *doutput, miopenTensorDescriptor_t dinputDesc, void *dinput, miopenLossReductionMode_t reduction) |
Execute a SoftMarginLoss backward layer. More... | |
miopenStatus_t | miopenGetMultiMarginLossForwardWorkspaceSize (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, miopenTensorDescriptor_t targetDesc, miopenTensorDescriptor_t weightDesc, miopenTensorDescriptor_t outputDesc, long p, float margin, miopenLossReductionMode_t reduction, size_t *sizeInBytes) |
Helper function to query the minimum workspace size required by the MultiMarginLossForward call. More... | |
miopenStatus_t | miopenMultiMarginLossForward (miopenHandle_t handle, miopenTensorDescriptor_t inputDesc, const void *input, miopenTensorDescriptor_t targetDesc, const void *target, miopenTensorDescriptor_t weightDesc, const void *weight, miopenTensorDescriptor_t outputDesc, void *output, long p, float margin, miopenLossReductionMode_t reduction, void *workspace, size_t workspaceSizeInBytes) |
Execute a MultiMarginLoss forward layer. More... | |
Macro Definition Documentation
◆ MIOPEN_DECLARE_OBJECT
#define MIOPEN_DECLARE_OBJECT | ( | name | ) |
Value:
struct name \
{ \
}; \
typedef struct name* name##_t;
Constructs type name from a struct
Enumeration Type Documentation
◆ miopenNanPropagation_t
◆ miopenPaddingMode_t
enum miopenPaddingMode_t |
◆ miopenReduceCalculationNanPropagation_t
Function Documentation
◆ MIOPEN_DECLARE_OBJECT() [1/5]
MIOPEN_DECLARE_OBJECT | ( | miopenDropoutDescriptor | ) |
Creates the miopenDropoutDescriptor_t type.
◆ MIOPEN_DECLARE_OBJECT() [2/5]
MIOPEN_DECLARE_OBJECT | ( | miopenFusionOpDescriptor | ) |
Creates the miopenFusionOpDescriptor_t type.
Fusion Operator Descriptor contains the meta-data associated with an operator to be fused in a compute graph
◆ MIOPEN_DECLARE_OBJECT() [3/5]
MIOPEN_DECLARE_OBJECT | ( | miopenMhaDescriptor | ) |
Creates the miopenMhaDescriptor_t type.
◆ MIOPEN_DECLARE_OBJECT() [4/5]
MIOPEN_DECLARE_OBJECT | ( | miopenOperatorArgs | ) |
◆ MIOPEN_DECLARE_OBJECT() [5/5]
MIOPEN_DECLARE_OBJECT | ( | miopenOperatorDescriptor | ) |