Add test cases of softmax with 1d input.

PiperOrigin-RevId: 289771939 Change-Id: Ifade8eedd5c8f3df1dd6c4cf4487fa6d46b25675
2020-01-14 18:14:09 -08:00 · 2020-01-14 18:14:09 -08:00 · 90cde42918
commit 90cde42918
parent 4573dbdb90
1 changed files with 2 additions and 1 deletions
--- a/tensorflow/lite/testing/op_tests/softmax.py
+++ b/tensorflow/lite/testing/op_tests/softmax.py
@ -29,7 +29,8 @@ def make_softmax_tests(options):

  test_parameters = [{
      "dtype": [tf.float32],
-      "input_shape": [[1, 3, 4, 3], [2, 3]],
+      "input_shape": [[1, 3, 4, 3], [2, 3], [3], [1, 4], [1, 1, 5],
+                      [1, 1, 1, 6]],
      "dim": [-1, 0],
      "fully_quantize": [False, True],
  }, {