From 241e828794162436d1eb08c42e072249388f171f Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Tue, 8 May 2018 15:43:34 -0700 Subject: [PATCH] Add test to test suite. PiperOrigin-RevId: 195888932 --- tensorflow/contrib/lite/kernels/internal/BUILD | 4 ++++ tensorflow/contrib/lite/kernels/internal/quantization_util_test.cc | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/tensorflow/contrib/lite/kernels/internal/BUILD b/tensorflow/contrib/lite/kernels/internal/BUILD index 5418821..d8340d4 100644 --- a/tensorflow/contrib/lite/kernels/internal/BUILD +++ b/tensorflow/contrib/lite/kernels/internal/BUILD @@ -5,6 +5,7 @@ package(default_visibility = [ licenses(["notice"]) # Apache 2.0 load("//tensorflow/contrib/lite:build_def.bzl", "tflite_copts") +load("//tensorflow/contrib/lite:special_rules.bzl", "tflite_portable_test_suite") tflite_deps_intel = [ "@arm_neon_2_x86_sse", @@ -428,6 +429,7 @@ cc_test( "//conditions:default": [], }), linkstatic = 1, + tags = ["tflite_not_portable_ios"], deps = [ ":tensor_utils", "//tensorflow/contrib/lite:builtin_op_data", @@ -462,3 +464,5 @@ cc_test( ) exports_files(["optimized/eigen_tensor_reduced_instantiations_oss.h"]) + +tflite_portable_test_suite() diff --git a/tensorflow/contrib/lite/kernels/internal/quantization_util_test.cc b/tensorflow/contrib/lite/kernels/internal/quantization_util_test.cc index 3e9a3c2..2d74b3d 100644 --- a/tensorflow/contrib/lite/kernels/internal/quantization_util_test.cc +++ b/tensorflow/contrib/lite/kernels/internal/quantization_util_test.cc @@ -167,6 +167,7 @@ TEST(QuantizationUtilTest, ChooseQuantizationParamsZeroPointOnMinBoundary) { EXPECT_EQ(qp.zero_point, 0); } +#ifdef GTEST_HAS_DEATH_TEST TEST(QuantizationUtilTest, ChooseQuantizationParamsZeroNotInRange) { // Assumption is that zero is within the range. EXPECT_DEATH(ChooseQuantizationParams(10.0, 30.0), ""); @@ -176,6 +177,7 @@ TEST(QuantizationUtilTest, ChooseQuantizationParamsEmptyRangePositive) { // Assumption is that zero is within the range. EXPECT_DEATH(ChooseQuantizationParams(30.0, 30.0), ""); } +#endif // GTEST_HAS_DEATH_TEST TEST(QuantizationUtilTest, ChooseQuantizationParamsEmptyRangeZero) { QuantizationParams qp = ChooseQuantizationParams(0.0, 0.0); @@ -189,6 +191,7 @@ TEST(QuantizationUtilTest, ChooseQuantizationParamsZeroPointOnMaxBoundary) { EXPECT_EQ(qp.zero_point, 255); } +#ifdef GTEST_HAS_DEATH_TEST TEST(QuantizationUtilTest, ChooseQuantizationParamsInvalidRange) { EXPECT_DEATH(ChooseQuantizationParams(10.0, -30.0), ""); } @@ -261,6 +264,7 @@ TEST(QuantizationUtilTest, PreprocessSoftmaxScaling) { EXPECT_THAT(quantize(2.0, 16.0, 5), Pair(2147483647, 31)); EXPECT_THAT(quantize(2.0, 8.0, 5), Pair(1073741824, 31)); } +#endif // GTEST_HAS_DEATH_TEST TEST(QuantizationUtilTest, CalculateInputRadius) { EXPECT_EQ(CalculateInputRadius(4, 27), 15); -- 2.7.4