Compute primitives. More...
Modules | |
| 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. | |
Compute primitives.