Add test for TopK

apache · Jan 14, 2019 · e3d4343 · e3d4343
1 parent a4921cd
commit e3d4343
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
@@ -77,7 +77,8 @@
              'test_elu',
              'test_max_',
              'test_softplus',
-             'test_reduce_'
+             'test_reduce_',
+             'test_top_k'
              ],
     'import': ['test_gather',
                'test_softsign',