Skip to content

Commit 1407175

Browse files
Merge pull request #893 from khanhlvg:ios-image-searcher-api-impl
PiperOrigin-RevId: 482888648
2 parents 949e3e4 + 2f2cf7a commit 1407175

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

tensorflow_lite_support/ios/BUILD

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,14 +45,19 @@ strip_api_include_path_prefix(
4545
"//tensorflow_lite_support/ios/task/vision:sources/TFLImageClassifier.h",
4646
"//tensorflow_lite_support/ios/task/vision:sources/TFLObjectDetector.h",
4747
"//tensorflow_lite_support/ios/task/vision:sources/TFLImageSegmenter.h",
48+
"//tensorflow_lite_support/ios/task/vision:sources/TFLImageSearcher.h",
4849
"//tensorflow_lite_support/ios:sources/TFLCommon.h",
4950
"//tensorflow_lite_support/ios/task/core:sources/TFLBaseOptions.h",
51+
"//tensorflow_lite_support/ios/task/core:sources/TFLExternalFile.h",
52+
"//tensorflow_lite_support/ios/task/processor:sources/TFLEmbeddingOptions.h",
53+
"//tensorflow_lite_support/ios/task/processor:sources/TFLSearchOptions.h",
5054
"//tensorflow_lite_support/ios/task/processor:sources/TFLClassificationOptions.h",
5155
"//tensorflow_lite_support/ios/task/processor:sources/TFLCategory.h",
5256
"//tensorflow_lite_support/ios/task/processor:sources/TFLClassificationResult.h",
5357
"//tensorflow_lite_support/ios/task/processor:sources/TFLDetectionResult.h",
5458
"//tensorflow_lite_support/ios/task/processor:sources/TFLSegmentationResult.h",
5559
"//tensorflow_lite_support/odml/ios/image:apis/GMLImage.h",
60+
"//tensorflow_lite_support/ios/task/processor:sources/TFLSearchResult.h",
5661
"//tensorflow_lite_support/ios/task/audio/core:sources/TFLAudioFormat.h",
5762
"//tensorflow_lite_support/ios/task/audio/core:sources/TFLFloatBuffer.h",
5863
"//tensorflow_lite_support/ios/task/audio/core/audio_record:sources/TFLAudioRecord.h",
@@ -179,15 +184,21 @@ ios_static_framework(
179184
":TFLClassificationResult.h",
180185
":TFLCommon.h",
181186
":TFLDetectionResult.h",
187+
":TFLEmbeddingOptions.h",
188+
":TFLExternalFile.h",
182189
":TFLImageClassifier.h",
190+
":TFLImageSearcher.h",
183191
":TFLImageSegmenter.h",
184192
":TFLObjectDetector.h",
193+
":TFLSearchOptions.h",
194+
":TFLSearchResult.h",
185195
":TFLSegmentationResult.h",
186196
],
187197
bundle_name = "TensorFlowLiteTaskVision",
188198
minimum_os_version = TFL_TASK_MINIMUM_OS_VERSION,
189199
deps = [
190200
"//tensorflow_lite_support/ios/task/vision:TFLImageClassifier",
201+
"//tensorflow_lite_support/ios/task/vision:TFLImageSearcher",
191202
"//tensorflow_lite_support/ios/task/vision:TFLImageSegmenter",
192203
"//tensorflow_lite_support/ios/task/vision:TFLObjectDetector",
193204
],
@@ -221,6 +232,7 @@ ios_static_framework(
221232
":TFLClassificationOptions.h",
222233
":TFLClassificationResult.h",
223234
":TFLCommon.h",
235+
":TFLExternalFile.h",
224236
":TFLFloatBuffer.h",
225237
],
226238
bundle_name = "TensorFlowLiteTaskAudio",

tensorflow_lite_support/ios/task/core/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ objc_library(
5151
hdrs = [
5252
"sources/TFLBaseOptions+CppHelpers.h",
5353
],
54-
module_name = "TFLBaseOptionsHelpers",
54+
module_name = "TFLBaseOptionsCppHelpers",
5555
deps = [
5656
"//tensorflow_lite_support/cc/task/core/proto:base_options_proto_inc",
5757
"//tensorflow_lite_support/ios/task/core:TFLBaseOptions",

0 commit comments

Comments
 (0)