search

amazon-archives / amazon-dsstne

Deep Scalable Sparse Tensor Network Engine (DSSTNE) is an Amazon developed library for building Deep Learning (DL) machine learning (ML) models
Apache License 2.0
4.41k stars 731 forks source link

Added brute force gpu knn and its JNI bindings #220

Closed kiukA9 closed 5 years ago

kiukA9 commented 5 years ago

Dsstne builds. All new files for knn are isolated in their own directories and does not affect existing dsstne function, capability or build (e.g. knn is its separate module).

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.