• Adam Becker's avatar
    mixed changes · 95f6eda6
    Adam Becker 提交于
     - add multidim support for top_k
     - use unified TopKOp, can implement topk, argtopk, and both
    95f6eda6
__init__.py 1.8 KB