Add test for TopK

apache · Feb 14, 2019 · c7952cd · c7952cd
1 parent 7d40c38
commit c7952cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python-pytest/onnx/test_cases.py b/tests/python-pytest/onnx/test_cases.py
@@ -78,7 +78,8 @@
              'test_max_',
              'test_softplus',
              'test_reduce_',
-             'test_split_equal'
+             'test_split_equal',
+             'test_top_k'
              ],
     'import': ['test_gather',
                'test_softsign',