Skip to content

Commit 82cb0db

Browse files
committed
Merge branch 'master' into ios-image-searcher-api-impl
2 parents 62e44b0 + 508139c commit 82cb0db

File tree

18 files changed

+57
-59
lines changed

18 files changed

+57
-59
lines changed

tensorflow_lite_support/ios/task/text/nlclassifier/BUILD

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
1-
load("//tools/build_defs/swift:swift_library.bzl", "swift_library")
1+
load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
22
load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
33
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
44
load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")
55

66
package(
7+
default_visibility = ["//tensorflow_lite_support:internal"],
78
licenses = ["notice"], # Apache 2.0
89
)
910

tensorflow_lite_support/ios/task/text/qa/BUILD

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
1-
load("//tools/build_defs/swift:swift_library.bzl", "swift_library")
1+
load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
22
load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
33
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
44
load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")
55

66
package(
7+
default_visibility = ["//tensorflow_lite_support:internal"],
78
licenses = ["notice"], # Apache 2.0
89
)
910

tensorflow_lite_support/ios/test/task/audio/audio_classifier/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
22
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
33
load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")
4-
load("//tools/build_defs/swift:swift_library.bzl", "swift_library")
4+
load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
55

66
package(
77
default_visibility = ["//visibility:private"],

tensorflow_lite_support/ios/test/task/audio/core/TFLRingBufferTests.m

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ - (void)testLoadSucceedsWithFullLengthBuffer {
5050
float expectedData[] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f};
5151

5252
for (int i = 0; i < inDataLength; i++) {
53-
XCTAssertEqual(outBuffer.data[i], inData[i]);
53+
XCTAssertEqual(outBuffer.data[i], expectedData[i]);
5454
}
5555
}
5656

@@ -241,7 +241,6 @@ - (void)testLoadFailsWithIndexOutofBounds {
241241
size:initialDataSize
242242
error:nil]);
243243

244-
NSInteger totalInSize = 4;
245244
float inArray[] = {6.0f, 7.0f, 8.0f, 9.0f};
246245

247246
NSInteger offset = 2;

tensorflow_lite_support/ios/test/task/audio/core/audio_record/TFLAudioRecordTests.m

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -189,9 +189,7 @@ - (void)testReadFailsWithoutLoad {
189189
XCTAssertNotNil(audioRecord);
190190

191191
NSError *error = nil;
192-
TFLFloatBuffer *floatBuffer = [audioRecord readAtOffset:0
193-
withSize:audioRecord.bufferSize
194-
error:&error];
192+
[audioRecord readAtOffset:0 withSize:audioRecord.bufferSize error:&error];
195193

196194
VerifyError(error, @"org.tensorflow.lite.audio.record",
197195
TFLAudioRecordErrorCodeWaitingForNewMicInputError,

tensorflow_lite_support/ios/test/task/audio/core/audio_tensor/TFLAudioTensorTests.m

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ - (void)testLoadBufferSucceeds {
209209
float expectedData[] = {1.0f, 2.0f, 3.0f, 4.0f, 5.0f};
210210

211211
for (int i = 0; i < sampleCount; i++) {
212-
XCTAssertEqual(outBuffer.data[i], inData[i]);
212+
XCTAssertEqual(outBuffer.data[i], expectedData[i]);
213213
}
214214
}
215215

tensorflow_lite_support/ios/test/task/vision/image_classifier/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
load("//tools/build_defs/swift:swift_library.bzl", "swift_library")
1+
load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
22
load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS")
33
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
44
load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")

tensorflow_lite_support/ios/test/task/vision/image_classifier/TFLImageClassifierTests.m

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,8 +103,8 @@ - (void)testInferenceWithBoundingBox {
103103
XCTAssertTrue(classificationResults.classifications.count > 0);
104104
XCTAssertTrue(classificationResults.classifications[0].categories.count > 0);
105105

106-
TFLCategory *category = classificationResults.classifications[0].categories[0];
107106
// TODO: match the label and score as image_classifier_test.cc
107+
// TFLCategory *__unused category = classificationResults.classifications[0].categories[0];
108108
// XCTAssertTrue([category.label isEqual:@"soccer ball"]);
109109
// XCTAssertEqualWithAccuracy(category.score, 0.256512, 0.001);
110110
}

tensorflow_lite_support/ios/test/task/vision/image_segmenter/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
load("//tools/build_defs/swift:swift_library.bzl", "swift_library")
1+
load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
22
load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
33
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
44
load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")

tensorflow_lite_support/ios/test/task/vision/object_detector/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
load("//tools/build_defs/swift:swift_library.bzl", "swift_library")
1+
load("@build_bazel_rules_swift//swift:swift.bzl", "swift_library")
22
load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
33
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
44
load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")

0 commit comments

Comments
 (0)