Home
last modified time | relevance | path

Searched refs:residualBase (Results 1 – 6 of 6) sorted by relevance

/dports/math/faiss/faiss-1.7.1/faiss/gpu/impl/scan/
H A DIVFInterleavedImpl.cuh25 Tensor<float, 3, true>& residualBase, \
49 Tensor<float, 3, true>& residualBase, \
66 residualBase, \
/dports/math/faiss/faiss-1.7.1/faiss/gpu/impl/
H A DIVFFlat.cu209 DeviceTensor<float, 3, true> residualBase( in query() local
216 quantizer_->reconstruct(coarseIndices, residualBase); in query()
230 residualBase, in query()
247 residualBase, in query()
H A DIVFFlatScan.cu139 Tensor<float, 3, true> residualBase, in ivfFlatScan() argument
168 auto residualBaseSlice = residualBase[queryId][probeId].data(); in ivfFlatScan()
201 Tensor<float, 3, true>& residualBase, in runIVFFlatScanTile() argument
238 residualBase, \ in runIVFFlatScanTile()
353 Tensor<float, 3, true>& residualBase, in runIVFFlatScan() argument
481 residualBase.narrowOutermost(query, numQueriesInTile); in runIVFFlatScan()
H A DIVFInterleaved.cuh42 Tensor<float, 3, true> residualBase, in ivfInterleavedScan() argument
77 auto residualBaseSlice = residualBase[queryId][probeId].data(); in ivfInterleavedScan()
228 residualBase, \
246 residualBase, \
420 Tensor<float, 3, true>& residualBase,
H A DIVFFlatScan.cuh32 Tensor<float, 3, true>& residualBase,
H A DIVFInterleaved.cu179 Tensor<float, 3, true>& residualBase, in runIVFInterleavedScan() argument