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

Classes

struct  EnumNames
 
struct  LibInfo
 

Enumerations

enum  : unsigned {
  CUDA = 0, CUDNN, NCCL, CUDA_RTC,
  TENSORRT, 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, MKLDNN, OPENCV,
  CAFFE, PROFILER, DIST_KVSTORE, CXX14,
  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 : unsigned

Compile time features.

Enumerator
CUDA 
CUDNN 
NCCL 
CUDA_RTC 
TENSORRT 
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 
MKLDNN 
OPENCV 
CAFFE 
PROFILER 
DIST_KVSTORE 
CXX14 
INT64_TENSOR_SIZE 
SIGNAL_HANDLER 
DEBUG 
TVM_OP 
MAX_FEATURES 

Function Documentation

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