DNNL namespace. More...
Classes | |
struct | batch_normalization_backward |
Batch normalization backward propagation primitive. More... | |
struct | batch_normalization_forward |
Batch normalization forward propagation primitive. More... | |
struct | binary |
Elementwise binary operator primitive. More... | |
struct | concat |
Tensor concatenation (concat) primitive. More... | |
struct | convolution_backward_data |
Convolution backward propagation primitive. More... | |
struct | convolution_backward_weights |
Convolution weights gradient primitive. More... | |
struct | convolution_forward |
Convolution forward propagation primitive. More... | |
struct | deconvolution_backward_data |
Deconvolution backward propagation primitive. More... | |
struct | deconvolution_backward_weights |
Deconvolution weights gradient primitive. More... | |
struct | deconvolution_forward |
Deconvolution forward propagation primitive. More... | |
struct | eltwise_backward |
Elementwise unary operation backward propagation primitive. More... | |
struct | eltwise_forward |
Elementwise unary operation forward propagation primitive. More... | |
struct | engine |
An execution engine. More... | |
struct | error |
DNNL exception class. More... | |
struct | gru_backward |
GRU backward propagation primitive. More... | |
struct | gru_forward |
GRU forward propagation primitive. More... | |
struct | handle |
DNNL C API handle wrapper class. More... | |
struct | handle_traits |
A class that provides the destructor for a DNNL C API handle. More... | |
struct | inner_product_backward_data |
Inner product backward propagation primitive. More... | |
struct | inner_product_backward_weights |
Inner product weights gradient primitive. More... | |
struct | inner_product_forward |
Inner product forward propagation primitive. More... | |
struct | layer_normalization_backward |
Layer normalization backward propagation primitive. More... | |
struct | layer_normalization_forward |
Layer normalization forward propagation primitive. More... | |
struct | lbr_gru_backward |
LBR GRU backward propagation primitive. More... | |
struct | lbr_gru_forward |
LBR GRU forward propagation primitive. More... | |
struct | logsoftmax_backward |
Logsoftmax backward propagation primitive. More... | |
struct | logsoftmax_forward |
Logsoftmax forward propagation primitive. More... | |
struct | lrn_backward |
Local response normalization (LRN) backward propagation primitive. More... | |
struct | lrn_forward |
Local response normalization (LRN) forward propagation primitive. More... | |
struct | lstm_backward |
LSTM backward propagation primitive. More... | |
struct | lstm_forward |
LSTM forward propagation primitive. More... | |
struct | matmul |
Matrix multiplication (matmul) primitive. More... | |
struct | memory |
Memory object. More... | |
struct | pooling_backward |
Pooling backward propagation primitive. More... | |
struct | pooling_forward |
Pooling forward propagation primitive. More... | |
struct | post_ops |
Post-ops. More... | |
struct | primitive |
Base class for all computational primitives. More... | |
struct | primitive_attr |
Primitive attributes. More... | |
struct | primitive_desc |
A base class for descriptors of all primitives that have an operation descriptor and that support iteration over multiple implementations. More... | |
struct | primitive_desc_base |
Base class for all primitive descriptors. More... | |
struct | reorder |
Reorder primitive. More... | |
struct | resampling_backward |
Resampling backward propagation primitive. More... | |
struct | resampling_forward |
Resampling forward propagation. More... | |
struct | rnn_primitive_desc_base |
Base class for primitive descriptors for RNN primitives. More... | |
struct | shuffle_backward |
Shuffle backward propagation primitive. More... | |
struct | shuffle_forward |
Shuffle forward propagation primitive. More... | |
struct | softmax_backward |
Softmax backward propagation primitive. More... | |
struct | softmax_forward |
Softmax forward propagation primitive. More... | |
struct | stream |
An execution stream. More... | |
struct | sum |
Out-of-place summation (sum) primitive. More... | |
struct | vanilla_rnn_backward |
Vanilla RNN backward propagation primitive. More... | |
struct | vanilla_rnn_forward |
Vanilla RNN forward propagation primitive. More... | |
Typedefs | |
using | version_t = dnnl_version_t |
Structure containing version information as per Semantic Versioning More... | |
Enumerations | |
enum | scratchpad_mode |
Scratchpad mode. More... | |
enum | prop_kind |
Propagation kind. More... | |
enum | algorithm |
Kinds of algorithms. More... | |
enum | normalization_flags : unsigned |
Flags for normalization primitives. More... | |
enum | rnn_flags : unsigned |
RNN cell flags. More... | |
enum | rnn_direction |
A direction of RNN primitive execution. More... | |
enum | query |
Primitive descriptor query specification. More... | |
enum | status |
Status values returned by the library functions. More... | |
enum | cpu_isa |
CPU instruction set flags. More... | |
Functions | |
dnnl_primitive_kind_t | convert_to_c (primitive::kind kind) |
Converts primitive kind enum value from C++ API to C API type. More... | |
dnnl_scratchpad_mode_t | convert_to_c (scratchpad_mode mode) |
Converts a scratchpad mode enum value from C++ API to C API type. More... | |
dnnl_prop_kind_t | convert_to_c (prop_kind kind) |
Converts propagation kind enum value from C++ API to C API type. More... | |
dnnl_alg_kind_t | convert_to_c (algorithm algorithm) |
Converts algorithm kind enum value from C++ API to C API type. More... | |
dnnl_normalization_flags_t | convert_to_c (normalization_flags flags) |
Converts normalization flags enum value from C++ API to C API type. More... | |
dnnl_rnn_flags_t | convert_to_c (rnn_flags flags) |
Converts RNN cell flags enum value from C++ API to C API type. More... | |
dnnl_rnn_direction_t | convert_to_c (rnn_direction dir) |
Converts RNN direction enum value from C++ API to C API type. More... | |
dnnl_query_t | convert_to_c (query query) |
Converts query enum value from C++ API to C API type. More... | |
status | set_verbose (int level) |
Configures verbose output to stdout. More... | |
const version_t * | version () |
Returns library version information. More... | |
status | set_jit_dump (int enable) |
Configures dumping of JIT-generated code. More... | |
status | set_jit_profiling_flags (unsigned flags) |
Sets library profiling flags. More... | |
status | set_jit_profiling_jitdumpdir (const std::string &dir) |
Sets JIT dump output path. More... | |
status | set_max_cpu_isa (cpu_isa isa) |
Sets the maximal ISA DNNL can dispatch to on the CPU. More... | |
status | sgemm (char transa, char transb, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const float *A, dnnl_dim_t lda, const float *B, dnnl_dim_t ldb, float beta, float *C, dnnl_dim_t ldc) |
Performs single-precision matrix-matrix multiply. More... | |
status | gemm_u8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const uint8_t *A, dnnl_dim_t lda, uint8_t ao, const int8_t *B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t *C, dnnl_dim_t ldc, const int32_t *co) |
Performs integer matrix-matrix multiply on 8-bit unsigned matrix A, 8-bit signed matrix B, and 32-bit signed resulting matrix C. More... | |
status | gemm_s8s8s32 (char transa, char transb, char offsetc, dnnl_dim_t M, dnnl_dim_t N, dnnl_dim_t K, float alpha, const int8_t *A, dnnl_dim_t lda, int8_t ao, const int8_t *B, dnnl_dim_t ldb, int8_t bo, float beta, int32_t *C, dnnl_dim_t ldc, const int32_t *co) |
Performs integer matrix-matrix multiply on 8-bit signed matrix A, 8-bit signed matrix B, and 32-bit signed resulting matrix C. More... | |
DNNL namespace.