69ebc96068
* - raspberry Pi build and ArmCompute library support - initial ArmCompute platform implementations (Maxpool2d AvgPool2d for float32) Signed-off-by: AbdelRauf <rauf@konduit.ai> * - Build script for pi - small changes Signed-off-by: AbdelRauf <rauf@konduit.ai>
23 lines
384 B
C
23 lines
384 B
C
#ifndef LIBND4J_CONFIG_H
|
|
#define LIBND4J_CONFIG_H
|
|
|
|
#cmakedefine HAVE_MKLDNN
|
|
|
|
#cmakedefine HAVE_ARMCOMPUTE
|
|
|
|
#cmakedefine MKLDNN_PATH "@MKLDNN_PATH@"
|
|
|
|
#cmakedefine HAVE_OPENBLAS
|
|
|
|
#cmakedefine OPENBLAS_PATH "@OPENBLAS_PATH@"
|
|
|
|
#cmakedefine HAVE_FLATBUFFERS
|
|
|
|
#cmakedefine FLATBUFFERS_PATH "@FLATBUFFERS_PATH@"
|
|
|
|
#cmakedefine HAVE_CUDNN
|
|
|
|
#cmakedefine DEFAULT_ENGINE @DEFAULT_ENGINE@
|
|
|
|
#endif
|