1# Library for generating feature vectors from audio data
2package(
3    default_visibility = ["//visibility:public"],
4    features = ["-layering_check"],
5    licenses = ["notice"],
6)
7
8cc_library(
9    name = "micro_model_settings",
10    srcs = [
11        "micro_model_settings.cc",
12    ],
13    hdrs = [
14        "micro_model_settings.h",
15    ],
16)
17
18cc_library(
19    name = "model",
20    srcs = [
21        "model.cc",
22    ],
23    hdrs = [
24        "model.h",
25    ],
26)
27
28cc_library(
29    name = "micro_features_test_data",
30    srcs = [
31        "no_micro_features_data.cc",
32        "yes_micro_features_data.cc",
33    ],
34    hdrs = [
35        "no_micro_features_data.h",
36        "yes_micro_features_data.h",
37    ],
38)
39
40cc_library(
41    name = "static_alloc",
42    hdrs = ["static_alloc.h"],
43)
44
45cc_library(
46    name = "micro_features_generator",
47    srcs = [
48        "micro_features_generator.cc",
49    ],
50    hdrs = [
51        "micro_features_generator.h",
52    ],
53    deps = [
54        ":micro_model_settings",
55        "//tensorflow/lite/c:common",
56        "//tensorflow/lite/experimental/microfrontend/lib:frontend",
57        "//tensorflow/lite/micro:micro_error_reporter",
58    ],
59)
60
61cc_library(
62    name = "micro_features_generator_test_data",
63    srcs = [
64        "no_feature_data_slice.cc",
65        "yes_feature_data_slice.cc",
66    ],
67    hdrs = [
68        "no_feature_data_slice.h",
69        "yes_feature_data_slice.h",
70    ],
71)
72
73cc_test(
74    name = "micro_features_generator_test",
75    size = "small",
76    srcs = [
77        "micro_features_generator_test.cc",
78    ],
79    tags = [
80        "noasan",  # TODO(b/179930607): Fix with asan.
81    ],
82    deps = [
83        ":micro_features_generator",
84        ":micro_features_generator_test_data",
85        ":micro_model_settings",
86        "//tensorflow/lite/c:common",
87        "//tensorflow/lite/micro:micro_error_reporter",
88        "//tensorflow/lite/micro:micro_framework",
89        "//tensorflow/lite/micro/examples/micro_speech:audio_sample_test_data",
90        "//tensorflow/lite/micro/testing:micro_test",
91    ],
92)
93