aboutsummaryrefslogtreecommitdiff
path: root/src/BUILD.bazel
diff options
context:
space:
mode:
Diffstat (limited to 'src/BUILD.bazel')
-rw-r--r--src/BUILD.bazel4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/BUILD.bazel b/src/BUILD.bazel
index 279c52e151..26acc14a68 100644
--- a/src/BUILD.bazel
+++ b/src/BUILD.bazel
@@ -765,10 +765,10 @@ filegroup(
"cpu/kernels/instancenorm/generic/neon/impl.cpp",
"cpu/kernels/internal/CpuDepthwiseConv2dAssemblyWrapperKernel.cpp",
"cpu/kernels/internal/CpuPool2dAssemblyWrapperKernel.cpp",
- "cpu/kernels/lut/generic/neon/u8.cpp",
"cpu/kernels/l2normlayer/generic/neon/fp16.cpp",
"cpu/kernels/l2normlayer/generic/neon/fp32.cpp",
"cpu/kernels/l2normlayer/generic/neon/impl.cpp",
+ "cpu/kernels/lut/generic/neon/u8.cpp",
"cpu/kernels/maxunpool/generic/neon/fp16.cpp",
"cpu/kernels/maxunpool/generic/neon/fp32.cpp",
"cpu/kernels/maxunpool/generic/neon/impl.cpp",
@@ -837,6 +837,7 @@ filegroup(
"cpu/operators/CpuGemmDirectConv2d.cpp",
"cpu/operators/CpuGemmLowpMatrixMultiplyCore.cpp",
"cpu/operators/CpuGemmLowpOutputStage.cpp",
+ "cpu/operators/CpuMatMul.cpp",
"cpu/operators/CpuMaxUnpooling.cpp",
"cpu/operators/CpuMul.cpp",
"cpu/operators/CpuPermute.cpp",
@@ -921,6 +922,7 @@ filegroup(
"runtime/NEON/functions/NELSTMLayer.cpp",
"runtime/NEON/functions/NELSTMLayerQuantized.cpp",
"runtime/NEON/functions/NELogical.cpp",
+ "runtime/NEON/functions/NEMatMul.cpp",
"runtime/NEON/functions/NEMaxUnpoolingLayer.cpp",
"runtime/NEON/functions/NEMeanStdDevNormalizationLayer.cpp",
"runtime/NEON/functions/NENormalizationLayer.cpp",