mxnet
Classes | Enumerations | Functions
mxnet::features Namespace Reference

Classes

struct  EnumNames
 
struct  LibInfo
 

Enumerations

enum  : unsigned {
  CUDA = 0, CUDNN, NCCL, TENSORRT,
  CUTENSOR, CPU_SSE, CPU_SSE2, CPU_SSE3,
  CPU_SSE4_1, CPU_SSE4_2, CPU_SSE4A, CPU_AVX,
  CPU_AVX2, OPENMP, SSE, F16C,
  JEMALLOC, BLAS_OPEN, BLAS_ATLAS, BLAS_MKL,
  BLAS_APPLE, LAPACK, ONEDNN, OPENCV,
  DIST_KVSTORE, INT64_TENSOR_SIZE, SIGNAL_HANDLER, DEBUG,
  TVM_OP, MAX_FEATURES
}
 Compile time features. More...
 

Functions

bool is_enabled (unsigned feat)
 

Enumeration Type Documentation

◆ anonymous enum

anonymous enum : unsigned

Compile time features.

Enumerator
CUDA 
CUDNN 
NCCL 
TENSORRT 
CUTENSOR 
CPU_SSE 
CPU_SSE2 
CPU_SSE3 
CPU_SSE4_1 
CPU_SSE4_2 
CPU_SSE4A 
CPU_AVX 
CPU_AVX2 
OPENMP 
SSE 
F16C 
JEMALLOC 
BLAS_OPEN 
BLAS_ATLAS 
BLAS_MKL 
BLAS_APPLE 
LAPACK 
ONEDNN 
OPENCV 
DIST_KVSTORE 
INT64_TENSOR_SIZE 
SIGNAL_HANDLER 
DEBUG 
TVM_OP 
MAX_FEATURES 

Function Documentation

◆ is_enabled()

bool mxnet::features::is_enabled ( unsigned  feat)
Returns
true if the given feature is supported