diff --git a/device.mk b/device.mk
index f9eaa291..7913c12c 100644
--- a/device.mk
+++ b/device.mk
@@ -148,9 +148,10 @@ PRODUCT_PACKAGES += \
        libGLES_mali-gs201-prebuilt \
        vulkan.gs201-prebuilt \
        libOpenCL \
-       android.hardware.neuralnetworks@1.3-service-armnn
 #        libgpudataproducer
 
+include device/google/gs101/neuralnetwork/neuralnetwork.mk
+
 ifeq ($(USE_SWIFTSHADER),true)
 PRODUCT_PACKAGES += \
        libGLESv1_CM_swiftshader \