File tree Expand file tree Collapse file tree 3 files changed +22
-3
lines changed
tensorflow_lite_support/cc/task/core Expand file tree Collapse file tree 3 files changed +22
-3
lines changed Original file line number Diff line number Diff line change @@ -20,6 +20,7 @@ cc_library_with_tflite(
20
20
# to get included in the executable.
21
21
"@org_tensorflow//tensorflow/lite/kernels:builtin_ops" ,
22
22
"@org_tensorflow//tensorflow/lite:framework_stable" ,
23
+ "@org_tensorflow//tensorflow/lite/c:c_api" ,
23
24
"@org_tensorflow//tensorflow/lite/tools:verifier" ,
24
25
"//tensorflow_lite_support/cc/port:tflite_wrapper" ,
25
26
],
@@ -38,6 +39,7 @@ cc_library_with_tflite(
38
39
"@com_google_absl//absl/status" ,
39
40
"@com_google_absl//absl/strings" ,
40
41
"@org_tensorflow//tensorflow/lite:kernel_api" ,
42
+ "@org_tensorflow//tensorflow/lite/core/api:error_reporter" ,
41
43
"@org_tensorflow//tensorflow/lite/core/api:op_resolver" ,
42
44
],
43
45
)
Original file line number Diff line number Diff line change @@ -19,18 +19,28 @@ limitations under the License.
19
19
#include < unistd.h>
20
20
#endif
21
21
22
+ #include < stddef.h>
23
+
22
24
#include < memory>
25
+ #include < string>
26
+ #include < utility>
27
+ #include < vector>
23
28
29
+ #include " absl/status/status.h" // from @com_google_absl
24
30
#include " absl/strings/match.h" // from @com_google_absl
25
31
#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"
29
37
#include " tensorflow/lite/tools/verifier.h"
30
38
#include " tensorflow_lite_support/cc/common.h"
31
39
#include " tensorflow_lite_support/cc/port/configuration_proto_inc.h"
32
40
#include " tensorflow_lite_support/cc/port/status_macros.h"
33
41
#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"
34
44
35
45
namespace tflite {
36
46
namespace task {
Original file line number Diff line number Diff line change @@ -16,16 +16,23 @@ limitations under the License.
16
16
#ifndef TENSORFLOW_LITE_SUPPORT_CC_TASK_CORE_TFLITE_ENGINE_H_
17
17
#define TENSORFLOW_LITE_SUPPORT_CC_TASK_CORE_TFLITE_ENGINE_H_
18
18
19
+ #include < stddef.h>
20
+ #include < stdint.h>
21
+
19
22
#include < memory>
23
+ #include < string>
24
+ #include < vector>
20
25
21
26
#include " absl/memory/memory.h" // from @com_google_absl
22
27
#include " absl/status/status.h" // from @com_google_absl
23
28
#include " absl/strings/string_view.h" // from @com_google_absl
24
29
#include " tensorflow/lite/c/common.h"
30
+ #include " tensorflow/lite/core/api/error_reporter.h"
25
31
#include " tensorflow/lite/core/api/op_resolver.h"
26
32
#include " tensorflow/lite/interpreter.h"
27
33
#include " tensorflow/lite/kernels/register.h"
28
34
#include " tensorflow/lite/model.h"
35
+ #include " tensorflow/lite/model_builder.h"
29
36
#include " tensorflow_lite_support/cc/port/configuration_proto_inc.h"
30
37
#include " tensorflow_lite_support/cc/port/tflite_wrapper.h"
31
38
#include " tensorflow_lite_support/cc/task/core/error_reporter.h"
You can’t perform that action at this time.
0 commit comments