Skip to content

Commit 0faffb0

Browse files
"Include what you use" fixes.
PiperOrigin-RevId: 667881420
1 parent 2a6e9a8 commit 0faffb0

File tree

3 files changed

+22
-3
lines changed

3 files changed

+22
-3
lines changed

‎tensorflow_lite_support/cc/task/core/BUILD

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ cc_library_with_tflite(
2020
# to get included in the executable.
2121
"@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
2222
"@org_tensorflow//tensorflow/lite:framework_stable",
23+
"@org_tensorflow//tensorflow/lite/c:c_api",
2324
"@org_tensorflow//tensorflow/lite/tools:verifier",
2425
"//tensorflow_lite_support/cc/port:tflite_wrapper",
2526
],
@@ -38,6 +39,7 @@ cc_library_with_tflite(
3839
"@com_google_absl//absl/status",
3940
"@com_google_absl//absl/strings",
4041
"@org_tensorflow//tensorflow/lite:kernel_api",
42+
"@org_tensorflow//tensorflow/lite/core/api:error_reporter",
4143
"@org_tensorflow//tensorflow/lite/core/api:op_resolver",
4244
],
4345
)

‎tensorflow_lite_support/cc/task/core/tflite_engine.cc

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,18 +19,28 @@ limitations under the License.
1919
#include <unistd.h>
2020
#endif
2121

22+
#include <stddef.h>
23+
2224
#include <memory>
25+
#include <string>
26+
#include <utility>
27+
#include <vector>
2328

29+
#include "absl/status/status.h" // from @com_google_absl
2430
#include "absl/strings/match.h" // from @com_google_absl
2531
#include "absl/strings/str_cat.h" // from @com_google_absl
26-
#include "tensorflow/lite/builtin_ops.h"
27-
#include "tensorflow/lite/kernels/register.h"
28-
#include "tensorflow/lite/stderr_reporter.h"
32+
#include "tensorflow/lite/c/c_api.h"
33+
#include "tensorflow/lite/core/api/error_reporter.h"
34+
#include "tensorflow/lite/core/api/op_resolver.h"
35+
#include "tensorflow/lite/interpreter_builder.h"
36+
#include "tensorflow/lite/model_builder.h"
2937
#include "tensorflow/lite/tools/verifier.h"
3038
#include "tensorflow_lite_support/cc/common.h"
3139
#include "tensorflow_lite_support/cc/port/configuration_proto_inc.h"
3240
#include "tensorflow_lite_support/cc/port/status_macros.h"
3341
#include "tensorflow_lite_support/cc/task/core/external_file_handler.h"
42+
#include "tensorflow_lite_support/cc/task/core/proto/external_file_proto_inc.h"
43+
#include "tensorflow_lite_support/metadata/cc/metadata_extractor.h"
3444

3545
namespace tflite {
3646
namespace task {

‎tensorflow_lite_support/cc/task/core/tflite_engine.h

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,16 +16,23 @@ limitations under the License.
1616
#ifndef TENSORFLOW_LITE_SUPPORT_CC_TASK_CORE_TFLITE_ENGINE_H_
1717
#define TENSORFLOW_LITE_SUPPORT_CC_TASK_CORE_TFLITE_ENGINE_H_
1818

19+
#include <stddef.h>
20+
#include <stdint.h>
21+
1922
#include <memory>
23+
#include <string>
24+
#include <vector>
2025

2126
#include "absl/memory/memory.h" // from @com_google_absl
2227
#include "absl/status/status.h" // from @com_google_absl
2328
#include "absl/strings/string_view.h" // from @com_google_absl
2429
#include "tensorflow/lite/c/common.h"
30+
#include "tensorflow/lite/core/api/error_reporter.h"
2531
#include "tensorflow/lite/core/api/op_resolver.h"
2632
#include "tensorflow/lite/interpreter.h"
2733
#include "tensorflow/lite/kernels/register.h"
2834
#include "tensorflow/lite/model.h"
35+
#include "tensorflow/lite/model_builder.h"
2936
#include "tensorflow_lite_support/cc/port/configuration_proto_inc.h"
3037
#include "tensorflow_lite_support/cc/port/tflite_wrapper.h"
3138
#include "tensorflow_lite_support/cc/task/core/error_reporter.h"

0 commit comments

Comments
 (0)