diff options
Diffstat (limited to 'src/BUILD.bazel')
-rw-r--r-- | src/BUILD.bazel | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/BUILD.bazel b/src/BUILD.bazel index 42841fe28d..f281b6a4d5 100644 --- a/src/BUILD.bazel +++ b/src/BUILD.bazel @@ -766,6 +766,8 @@ filegroup( "cpu/kernels/fuse_batch_normalization/generic/fp16.cpp", "cpu/kernels/fuse_batch_normalization/generic/fp32.cpp", "cpu/kernels/fuse_batch_normalization/nchw/all.cpp", + "cpu/kernels/fuse_batch_normalization/nchw/neon/fp16.cpp", + "cpu/kernels/fuse_batch_normalization/nchw/neon/fp32.cpp", "cpu/kernels/fuse_batch_normalization/nhwc/neon/fp16.cpp", "cpu/kernels/fuse_batch_normalization/nhwc/neon/fp32.cpp", "cpu/kernels/gemm_matrix_add/generic/neon/fp16.cpp", |