diff options
author | 2018-09-11 21:18:05 -0700 | |
---|---|---|
committer | 2018-09-11 21:22:09 -0700 | |
commit | 6a21e1386e3e68cf752af861b9b1b950bda8a130 (patch) | |
tree | 0ccc9df1c589ffb059b707c58d8c7c094f6f9de0 /tensorflow/contrib/lite/testing | |
parent | cadd6b42bf6b01c2668420463b0986acd7fd9009 (diff) |
Implementation of square.
PiperOrigin-RevId: 212577288
Diffstat (limited to 'tensorflow/contrib/lite/testing')
-rw-r--r-- | tensorflow/contrib/lite/testing/generate_examples.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/testing/generate_examples.py b/tensorflow/contrib/lite/testing/generate_examples.py index 812385e706..5d0895c72f 100644 --- a/tensorflow/contrib/lite/testing/generate_examples.py +++ b/tensorflow/contrib/lite/testing/generate_examples.py @@ -2882,6 +2882,11 @@ def make_rsqrt_tests(zip_path): return _make_elementwise_tests(tf.rsqrt)(zip_path) +def make_square_tests(zip_path): + """Make a set of tests to do square.""" + return _make_elementwise_tests(tf.square)(zip_path) + + def make_where_tests(zip_path): """Make a set of tests to do where.""" |