diff --git a/Android.bp b/Android.bp
index 31bc14b..04f9d93 100644
--- a/Android.bp
+++ b/Android.bp
@@ -300,7 +300,6 @@
         "src/core/NEON/kernels/NESobel3x3Kernel.cpp",
         "src/core/NEON/kernels/NESobel5x5Kernel.cpp",
         "src/core/NEON/kernels/NESobel7x7Kernel.cpp",
-        "src/core/NEON/kernels/NESoftmaxLayerKernel.cpp",
         "src/core/NEON/kernels/NESpaceToBatchLayerKernel.cpp",
         "src/core/NEON/kernels/NESpaceToDepthLayerKernel.cpp",
         "src/core/NEON/kernels/NEStackLayerKernel.cpp",
@@ -405,6 +404,7 @@
         "src/core/cpu/kernels/CpuPoolingAssemblyWrapperKernel.cpp",
         "src/core/cpu/kernels/CpuPoolingKernel.cpp",
         "src/core/cpu/kernels/CpuReshapeKernel.cpp",
+        "src/core/cpu/kernels/CpuSoftmaxKernel.cpp",
         "src/core/cpu/kernels/CpuSubKernel.cpp",
         "src/core/cpu/kernels/activation/NEON/fp16.cpp",
         "src/core/cpu/kernels/activation/NEON/fp32.cpp",
@@ -801,6 +801,7 @@
         "src/runtime/cpu/operators/CpuPooling.cpp",
         "src/runtime/cpu/operators/CpuPoolingAssemblyDispatch.cpp",
         "src/runtime/cpu/operators/CpuReshape.cpp",
+        "src/runtime/cpu/operators/CpuSoftmax.cpp",
         "src/runtime/cpu/operators/CpuSub.cpp",
         "src/runtime/gpu/cl/operators/ClActivation.cpp",
         "src/runtime/gpu/cl/operators/ClAdd.cpp",