experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make portable test suite in kernels/internal package combined

Browse Source 
PiperOrigin-RevId: 272325828
This commit is contained in:
Juho Ha 2019-10-01 16:54:02 -07:00 committed by TensorFlower Gardener
parent e4db003a22
commit db6be9c716
3 changed files with 4 additions and 16 deletions
tensorflow/lite/kernels/internal
BUILDquantization_util_test.cctensor_test.cc

8
tensorflow/lite/kernels/internal/BUILD
View File

@ -1,6 +1,6 @@
load("//tensorflow:tensorflow.bzl", "transitive_hdrs") load("//tensorflow:tensorflow.bzl", "transitive_hdrs")
load("//tensorflow/lite:build_def.bzl", "tflite_copts") load("//tensorflow/lite:build_def.bzl", "tflite_copts")
load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite") load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite_combined")
package( package(
default_visibility = [ default_visibility = [
@ -318,7 +318,7 @@ cc_test(
srcs = ["tensor_test.cc"], srcs = ["tensor_test.cc"],
deps = [ deps = [
":tensor", ":tensor",
"@com_google_googletest//:gtest", "@com_google_googletest//:gtest_main",
], ],
) )
@ -345,7 +345,7 @@ cc_test(
srcs = ["quantization_util_test.cc"], srcs = ["quantization_util_test.cc"],
deps = [ deps = [
":quantization_util", ":quantization_util",
"@com_google_googletest//:gtest", "@com_google_googletest//:gtest_main",
], ],
) )
@ -1010,4 +1010,4 @@ genrule(
visibility = ["//visibility:private"], visibility = ["//visibility:private"],
) )
tflite_portable_test_suite() tflite_portable_test_suite_combined(combine_conditions = {"deps": ["//testing/base/public:gunit_main"]})

6
tensorflow/lite/kernels/internal/quantization_util_test.cc
View File

@ -471,9 +471,3 @@ TEST(QuantizationUtilTest, QuantizeMultiplierArray) {
} // namespace } // namespace
} // namespace tflite } // namespace tflite
int main(int argc, char** argv) {
// On Linux, add: tflite::LogToStderr();
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}

6
tensorflow/lite/kernels/internal/tensor_test.cc
View File

@ -51,9 +51,3 @@ TEST(TensorTest, GetTensorShape1D) {
} // namespace } // namespace
} // namespace tflite } // namespace tflite
int main(int argc, char** argv) {
// On Linux, add: tflite::LogToStderr();
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}