▼oneDNN API | |
▼Primitives | Compute primitives |
Common | Common operations to create, destroy and inspect primitives |
Attributes | A container for parameters that extend primitives behavior |
Reorder | A primitive to copy data between two memory objects |
Concat | A primitive to concatenate data by arbitrary dimension |
Sum | A primitive to sum multiple tensors |
Binary | A primitive to perform tensor operations over two tensors |
Convolution | A primitive to perform 1D, 2D or 3D convolution |
Deconvolution | A primitive to perform 1D, 2D or 3D deconvolution |
Shuffle | A primitive to shuffle tensor data along an axis |
Eltwise | A primitive to perform elementwise operations such as the rectifier linear unit (ReLU) |
Softmax | A primitive to perform softmax |
LogSoftmax | A primitive to perform logsoftmax |
Pooling | A primitive to perform max or average pooling |
LRN | A primitive to perform local response normalization (LRN) across or within channels |
Batch Normalization | A primitive to perform batch normalization |
Layer Normalization | A primitive to perform layer normalization |
Inner Product | A primitive to compute an inner product |
RNN | A primitive to compute recurrent neural network layers |
Matrix Multiplication | A primitive to perform matrix-matrix multiplication |
Resampling | A primitive to compute resampling operation on 1D, 2D or 3D data tensor using Nearest Neighbor, or Linear (Bilinear, Trilinear) interpolation method |
Reduction | A primitive to compute reduction operation on data tensor using min, max, mul, sum, mean and norm_lp operations |
Dnnl_api_pooling_v2 | |
Memory | A container that describes and stores data |
Engine | An abstraction of a computational device: a CPU, a specific GPU card in the system, etc |
Stream | An encapsulation of execution context tied to a particular engine |
Primitive Cache | A set of functions that provide primitive cache control |
Service | A set of functions that aid in oneDNN debugging and profiling |
BLAS functions | A subset of Basic Linear Algebra (BLAS) functions that perform matrix-matrix multiplication |
Utilities | Utility types and definitions |