• Q
    Fix top_k op (#14034) · c7379a73
    qingqing01 提交于
    1. Fix CUDA kernel when height is large than 2048.
    2. Support input with more than 2D.
    3. Fix unit test when k is large than 1.
    4. Enhence unit testing.
    
    test=develop
    c7379a73
top_k_op.cu 11.1 KB