5d7e80f413
[Feature] Add bruteforce implementation for KNN with O(Nk) space complexity ( #2892 )
...
* add bruteforce impl
* add support for bruteforce-sharemem
* modify python API
* add tests
* change file path
* change python API
* fix lint
* fix test
* also check worst_dist in the last few dim
* use heap and early-stop on CPU
* fix lint
* fix lint
* add device check
* use cuda function to determine max shared mem
* use cuda to determine block info
* add memory free for tmp var
* update doc-string and add dist option
* fix lint
* add more tests
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com >
2021-05-19 18:27:43 +08:00
e83d0a8007
[Feature] Add kd-tree implementation (CPU) for kNN ( #2767 )
...
* add submodule nanoflann
* finish python API for knn
* finish ndarray adaptor
* finish cpu-kdtree version of knn
* use openmp
* add endline
* upt
* upt
* fix format and code style
* upt
* add warning for gpu-cpu copy
* avoid contiguous copy
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com >
Co-authored-by: Tong He <hetong007@gmail.com >
2021-04-09 17:39:34 +08:00
2fa7f71ad5
[Doc] new nn api doc ( #2019 )
...
* Add dotproduct attention
* [Feature] Add dotproduct attention
* [Feature] Add dotproduct attention
* [Feature] Add dotproduct attention
* [New] Update landing page
* [New] Update landing page
* [New] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Doc] Update landing page
* [Improvement] use dgl build-in in dotgatconv
* [Doc] review API doc string bottom up
* [Doc] Add doc of input and output features
* [Doc] Update doc string for pooling and transformer.
* [Doc] Reformat doc string and change some wordings.
* [Doc] Doc string refactoring.
* [Doc] Doc string refactoring.
* [Doc] Doc string refactoring.
* [Doc] Doc string refactoring.
* [Doc] Doc string refactoring.
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com >
2020-08-14 18:00:38 +08:00
36ec71a268
[Doc] Quick Fix doc ( #820 )
...
* fix doc
* poke ci
* poke ci
* upd
* fix doc
* upd
* add title for all nn
* upd
* fix warning
* fix
2019-09-01 22:30:50 +08:00
14bffe9728
[NN] Renaming NearestNeighborGraph to KNNGraph ( #802 )
...
* initial commit
* second commit
* another commit
* change docstring
* migrating to dgl.nn
* fixes
* docs
* lint
* multiple fixes
* doc
* renaming nearest neighbor graph
2019-08-28 12:20:35 +08:00
dc19cd5687
[Example] Dynamic Graph CNN on Point Cloud ( #789 )
...
* initial commit
* second commit
* another commit
* change docstring
* migrating to dgl.nn
* fixes
* docs
* lint
* multiple fixes
* doc
2019-08-28 09:21:57 +08:00
Tianqi Zhang (张天启)