From 5f26a30df10dc2db36a66692e6b5dd904468c06c Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 25 Oct 2019 14:34:48 -0700 Subject: [PATCH] Add quantize tests for activation op: sigmoid(aka logistic). PiperOrigin-RevId: 276762422 Change-Id: I79303dbdb19178e144a673fa0ae87ba58147ee2c --- tensorflow/lite/testing/BUILD | 1 + tensorflow/lite/testing/op_tests/sigmoid.py | 6 +++++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/testing/BUILD b/tensorflow/lite/testing/BUILD index a9bc36b30df..e9889ac6586 100644 --- a/tensorflow/lite/testing/BUILD +++ b/tensorflow/lite/testing/BUILD @@ -495,6 +495,7 @@ edgetpu_ops = [ "pad", # high error "reshape", "resize_bilinear", + "sigmoid", "slice", "softmax", "space_to_depth", diff --git a/tensorflow/lite/testing/op_tests/sigmoid.py b/tensorflow/lite/testing/op_tests/sigmoid.py index 5e50e279ddd..d13bb4ad8e4 100644 --- a/tensorflow/lite/testing/op_tests/sigmoid.py +++ b/tensorflow/lite/testing/op_tests/sigmoid.py @@ -30,6 +30,8 @@ def make_sigmoid_tests(options): test_parameters = [{ "dtype": [tf.float32], "input_shape": [[1, 3, 4, 3], [4], [], [1, 2, 3, 4, 5, 6]], + "fully_quantize": [True, False], + "input_range": [(-10, 10)], }] def build_graph(parameters): @@ -41,8 +43,10 @@ def make_sigmoid_tests(options): return [input_tensor], [out] def build_inputs(parameters, sess, inputs, outputs): + min_value, max_value = parameters["input_range"] input_values = create_tensor_data(parameters["dtype"], - parameters["input_shape"]) + parameters["input_shape"], min_value, + max_value) return [input_values], sess.run( outputs, feed_dict=dict(zip(inputs, [input_values])))