Add test for TopK

apache · Jan 3, 2019 · aad3e84 · aad3e84
1 parent 3e45d5d
commit aad3e84
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
@@ -76,7 +76,8 @@
              'test_selu_default',
              'test_elu',
              'test_max_',
-             'test_softplus'
+             'test_softplus',
+             'test_top_k'
              ],
     'import': ['test_gather',
                'test_global_lppooling',