LICENSE
MANIFEST.in
README.md
setup.py
torchpq/CustomModule.py
torchpq/__init__.py
torchpq/metric.py
torchpq/util.py
torchpq.egg-info/PKG-INFO
torchpq.egg-info/SOURCES.txt
torchpq.egg-info/dependency_links.txt
torchpq.egg-info/requires.txt
torchpq.egg-info/top_level.txt
torchpq/clustering/KMeans.py
torchpq/clustering/KMeansOld.py
torchpq/clustering/MinibatchKMeans.py
torchpq/clustering/MultiKMeans.py
torchpq/clustering/MultiKMeansOld.py
torchpq/clustering/__init__.py
torchpq/codec/BaseCodec.py
torchpq/codec/PQCodec.py
torchpq/codec/SQCodec.py
torchpq/codec/VQCodec.py
torchpq/codec/__init__.py
torchpq/container/BaseContainer.py
torchpq/container/CellContainer.py
torchpq/container/FlatContainer.py
torchpq/container/FlatContainerGroup.py
torchpq/container/__init__.py
torchpq/experimental/IVFMPQ.py
torchpq/experimental/MPQ.py
torchpq/experimental/__init__.py
torchpq/fn/IVFPQTopk.py
torchpq/fn/Topk.py
torchpq/fn/__init__.py
torchpq/index/BaseIndex.py
torchpq/index/IVFPQIndex.py
torchpq/index/__init__.py
torchpq/kernels/BMMCuda.py
torchpq/kernels/ComputeCentroidsCuda.py
torchpq/kernels/ComputeProductCuda.py
torchpq/kernels/CustomKernel.py
torchpq/kernels/GetAddressByIDCuda.py
torchpq/kernels/GetDivByAddressCuda.py
torchpq/kernels/GetDivByAddressV2Cuda.py
torchpq/kernels/GetIOACuda.py
torchpq/kernels/GetWriteAddressCuda.py
torchpq/kernels/GetWriteAddressV2Cuda.py
torchpq/kernels/IVFPQTop1Cuda.py
torchpq/kernels/IVFPQTopkCuda.py
torchpq/kernels/MBMMCuda.py
torchpq/kernels/MaxSimCuda.py
torchpq/kernels/MinBMMCuda.py
torchpq/kernels/PQDecodeCuda.py
torchpq/kernels/TopkBMMCuda.py
torchpq/kernels/TopkSelectCuda.py
torchpq/kernels/__init__.py
torchpq/kernels/cuda/bmm.cu
torchpq/kernels/cuda/bmm_helpers.cu
torchpq/kernels/cuda/compute_centroids.cu
torchpq/kernels/cuda/compute_product.cu
torchpq/kernels/cuda/get_address_by_id.cu
torchpq/kernels/cuda/get_div_by_address.cu
torchpq/kernels/cuda/get_div_by_address_v2.cu
torchpq/kernels/cuda/get_ioa.cu
torchpq/kernels/cuda/get_write_address.cu
torchpq/kernels/cuda/get_write_address_v2.cu
torchpq/kernels/cuda/ivfpq_top1.cu
torchpq/kernels/cuda/ivfpq_topk.cu
torchpq/kernels/cuda/max_sim.cu
torchpq/kernels/cuda/mbmm.cu
torchpq/kernels/cuda/minbmm.cu
torchpq/kernels/cuda/pq_decode.cu
torchpq/kernels/cuda/topk_select.cu
torchpq/kernels/cuda/topkbmm.cu
torchpq/legacy/IVFPQ.py
torchpq/legacy/IVFPQBase.py
torchpq/legacy/IVFPQR.py
torchpq/legacy/IVFPQTopk.py
torchpq/legacy/PQ.py
torchpq/legacy/SQ.py
torchpq/legacy/__init__.py
torchpq/transform/PCA.py
torchpq/transform/__init__.py