You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2023/01/05 12:05:05 UTC

[GitHub] [tvm] heliqi commented on a diff in pull request #13701: [Frontend][PaddlePaddle] Add topk op and Fix bug, when the output is a dimension, it …

heliqi commented on code in PR #13701:
URL: https://github.com/apache/tvm/pull/13701#discussion_r1062402600


##########
tests/python/frontend/paddlepaddle/test_forward.py:
##########
@@ -1681,5 +1687,15 @@ def forward(self, inputs, prev_h):
         )
 
 
+@tvm.testing.uses_gpu
+def test_forward_topk():
+    @paddle.jit.to_static
+    def topk1(inputs):
+        return paddle.topk(inputs, k=1)
+
+    input_data = paddle.to_tensor([1, 4, 5, 7])
+    verify_model(topk1, input_data=input_data)

Review Comment:
   Yes, the paddle.topk api will definitely have two outputs.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org