aboutsummaryrefslogtreecommitdiff
path: root/src/BUILD.bazel
diff options
context:
space:
mode:
Diffstat (limited to 'src/BUILD.bazel')
-rw-r--r--src/BUILD.bazel12
1 files changed, 12 insertions, 0 deletions
diff --git a/src/BUILD.bazel b/src/BUILD.bazel
index e3cac07de1..f270824ab4 100644
--- a/src/BUILD.bazel
+++ b/src/BUILD.bazel
@@ -119,6 +119,8 @@ filegroup(
"cpu/kernels/lut/generic/sve2/u8.cpp",
"cpu/kernels/softmax/generic/sme2/fp16.cpp",
"cpu/kernels/softmax/generic/sme2/fp32.cpp",
+ "cpu/kernels/softmax/generic/sme2/qasymm8.cpp",
+ "cpu/kernels/softmax/generic/sme2/qasymm8_signed.cpp",
"cpu/kernels/softmax/generic/sve2/impl.cpp"] +
glob(["**/*.h",
"**/*.hpp",
@@ -751,6 +753,8 @@ filegroup(
"cpu/kernels/depthwiseconv2d/generic/neon/impl.cpp",
"cpu/kernels/depthwiseconv2d/generic/neon/qasymm8.cpp",
"cpu/kernels/depthwiseconv2d/generic/neon/qasymm8_signed.cpp",
+ "cpu/kernels/dequantize/generic/neon/fp16.cpp",
+ "cpu/kernels/dequantize/generic/neon/fp32.cpp",
"cpu/kernels/directconv2d/nchw/all.cpp",
"cpu/kernels/directconv2d/nchw/fp16.cpp",
"cpu/kernels/directconv2d/nhwc/neon/fp16.cpp",
@@ -816,9 +820,17 @@ filegroup(
"cpu/kernels/pool3d/neon/fp32.cpp",
"cpu/kernels/pool3d/neon/qasymm8.cpp",
"cpu/kernels/pool3d/neon/qasymm8_signed.cpp",
+ "cpu/kernels/quantize/generic/neon/fp16.cpp",
+ "cpu/kernels/quantize/generic/neon/fp32.cpp",
+ "cpu/kernels/quantize/generic/neon/integer.cpp",
"cpu/kernels/range/generic/neon/fp16.cpp",
"cpu/kernels/range/generic/neon/fp32.cpp",
"cpu/kernels/range/generic/neon/integer.cpp",
+ "cpu/kernels/reduction_layer/generic/neon/fp16.cpp",
+ "cpu/kernels/reduction_layer/generic/neon/fp32.cpp",
+ "cpu/kernels/reduction_layer/generic/neon/integer.cpp",
+ "cpu/kernels/reduction_layer/generic/neon/qasymm8.cpp",
+ "cpu/kernels/reduction_layer/generic/neon/qasymm8_signed.cpp",
"cpu/kernels/roialign/generic/neon/fp16.cpp",
"cpu/kernels/roialign/generic/neon/fp32.cpp",
"cpu/kernels/roialign/generic/neon/qasymm8.cpp",