ymwangg commented on pull request #7441: URL: https://github.com/apache/tvm/pull/7441#issuecomment-782585022
@masahi I added the `return_counts` option for the `topi.unique` operator. I also added pytorch frontend. Interestingly, it looks like pytorch returns unique elements in random order when `sorted=False`. I'll work on the GPU version of `unique` next week. ---------------------------------------------------------------- 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. For queries about this service, please contact Infrastructure at: [email protected]
