• Justin Johnson's avatar
    Add CPU implementation for nearest neighbor · e290f87c
    Justin Johnson authored
    Summary:
    Adds a CPU implementation for `pytorch3d.ops.nn_points_idx`.
    
    Also renames the associated C++ and CUDA functions to use `AllCaps` names used in other C++ / CUDA code.
    
    Reviewed By: gkioxari
    
    Differential Revision: D19670491
    
    fbshipit-source-id: 1b6409404025bf05e6a93f5d847e35afc9062f05
    e290f87c
test_nearest_neighbor_points.py 2.8 KB