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

Add test cases of softmax with 1d input.

Browse Source 
PiperOrigin-RevId: 289771939
Change-Id: Ifade8eedd5c8f3df1dd6c4cf4487fa6d46b25675
This commit is contained in:
A. Unique TensorFlower 2020-01-14 18:14:09 -08:00 committed by TensorFlower Gardener
parent 4573dbdb90
commit 90cde42918
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/testing/op_tests/softmax.py
View File

@ -29,7 +29,8 @@ def make_softmax_tests(options):
test_parameters = [{ test_parameters = [{
"dtype": [tf.float32], "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], "dim": [-1, 0],
"fully_quantize": [False, True], "fully_quantize": [False, True],
}, { }, {