Functions — github.com/XiangpengHao/pq-vector

hub / github.com/XiangpengHao/pq-vector / functions

Functions167 in github.com/XiangpengHao/pq-vector

⨍ Functions 167◇ Types & classes 36

↓ 47 callersMethodas_any

(&self)

src/df_vector/exec.rs:339

↓ 16 callersMethodas_usize

(self)

src/ivf/index.rs:28

↓ 7 callersMethodas_str

(&self)

src/ivf/mod.rs:30

↓ 7 callersMethodas_usize

(self)

src/ivf/mod.rs:63

↓ 7 callersMethodchildren

(&self)

src/df_vector/exec.rs:354

↓ 7 callersFunctionsquared_l2_distance

(a: &[f32], b: &[f32])

↓ 5 callersMethodbuild_new

(self, output: impl AsRef<Path>)

src/ivf/parquet.rs:71

↓ 5 callersMethoddim

(&self)

src/ivf/index.rs:53

↓ 5 callersFunctionstrip_wrappers

(expr: &Arc<dyn PhysicalExpr>)

src/df_vector/physical.rs:216

↓ 5 callersMethodwith_new_children

( self: Arc<Self>, children: Vec<Arc<dyn ExecutionPlan>>, )

src/df_vector/exec.rs:358

↓ 3 callersMethodbuild_inplace

(self)

src/ivf/parquet.rs:57

↓ 3 callersMethodbuild_topk_from_sort

( &self, sort: &SortExec, limit_fetch: Option<usize>, )

src/df_vector/physical.rs:115

↓ 3 callersFunctioncompute_distance_values

(values: &ArrayRef, query: &[f32])

src/df_vector/exec.rs:524

↓ 3 callersMethoddata

(&self)

src/ivf/mod.rs:91

↓ 3 callersFunctionextract_vector_values

( values: &arrow::array::ArrayRef, )

benches/query.rs:449

↓ 3 callersFunctionfloat_array_to_vec

(values: &ArrayRef)

src/df_vector/expr.rs:44

↓ 3 callersFunctiongather_single_parquet_scan

( plan: &Arc<dyn ExecutionPlan>, )

src/df_vector/access.rs:35

↓ 3 callersFunctionhash_vector_values

(values: &arrow::array::ArrayRef)

benches/query.rs:540

↓ 3 callersFunctionparallel_chunks_mut

(data: &mut [T], f: F)

src/ivf/index.rs:296

↓ 3 callersMethodpartial_cmp

(&self, other: &Self)

src/ivf/search.rs:27

↓ 3 callersFunctionread_index_metadata_from_file_metadata

( metadata: &FileMetaData, )

src/ivf/parquet.rs:145

↓ 3 callersMethodrow_count

(&self)

src/ivf/mod.rs:99

↓ 3 callersMethodto_bytes

(&self)

src/ivf/index.rs:65

↓ 2 callersMethodas_u32

(self)

src/ivf/index.rs:32

↓ 2 callersMethodbuild_config

(&self)

src/ivf/parquet.rs:88

↓ 2 callersFunctionbuild_ivf_index

( embeddings: &Embeddings, config: IvfBuildConfig, )

src/ivf/index.rs:152

↓ 2 callersMethodbuild_topk_from_sort_inner

( &self, sort: &SortExec, limit_fetch: Option<usize>, )

src/df_vector/physical.rs:134

↓ 2 callersMethodcandidate_rows

(&self, query: &[f32], nprobe: usize)

src/ivf/index.rs:57

↓ 2 callersFunctioncolumn_from_expr

(expr: &Arc<dyn PhysicalExpr>)

src/df_vector/physical.rs:231

↓ 2 callersFunctioncolumn_write_options

(column: &ColumnChunkMetaData)

src/ivf/parquet.rs:466

↓ 2 callersFunctionensure_indexed

(source: &str, indexed: &str)

examples/common/mod.rs:39

↓ 2 callersFunctionextract_keys

( batches: &[arrow::record_batch::RecordBatch], use_id: bool, )

benches/query.rs:498

↓ 2 callersFunctiongenerate_parquet

( path: &Path, rows: usize, dim: usize, batch_rows: usize, )

benches/bench_util.rs:12

↓ 2 callersFunctionis_dictionary_encoding

(encoding: Encoding)

src/ivf/parquet.rs:529

↓ 2 callersFunctionis_level_encoding

(encoding: Encoding)

src/ivf/parquet.rs:525

↓ 2 callersFunctionk_means

K-means clustering implementation with k-means++ initialization.

src/ivf/index.rs:323

↓ 2 callersFunctionliteral_from_expr

(expr: &Arc<dyn PhysicalExpr>)

src/df_vector/physical.rs:238

↓ 2 callersMethodn_clusters

(mut self, n_clusters: usize)

src/ivf/parquet.rs:42

↓ 2 callersMethodname

(&self)

src/df_vector/exec.rs:335

↓ 2 callersMethodname

(&self)

src/df_vector/physical.rs:189

↓ 2 callersFunctionparquet_scan_files

(scan: &ParquetScanInfo)

src/df_vector/access.rs:178

↓ 2 callersMethodpartial_cmp

(&self, other: &Self)

src/df_vector/exec.rs:444

↓ 2 callersFunctionprint_plan_metrics

(plan: &Arc<dyn ExecutionPlan>, indent: usize)

benches/query.rs:461

↓ 2 callersFunctionread_embedding_at_row

( path: &Path, column_name: &str, row: usize, )

examples/common/mod.rs:7

↓ 2 callersFunctionread_index_from_parquet

( path: &Path, )

src/ivf/parquet.rs:191

↓ 2 callersFunctionread_index_from_payload

( payload: &[u8], embedding_column: EmbeddingColumn, )

src/ivf/parquet.rs:151

↓ 2 callersFunctionread_parquet_with_embeddings

( path: &Path, embedding_column: &EmbeddingColumn, )

src/ivf/parquet.rs:216

↓ 2 callersMethodrewrite_plan

( &self, plan: Arc<dyn ExecutionPlan>, has_offset: bool, )

src/df_vector/physical.rs:32

↓ 2 callersMethodwith_pq_vector

(mut self)

src/df_vector/session.rs:17

↓ 2 callersFunctionworker_count

(len: usize)

src/ivf/index.rs:259

↓ 1 callersFunctionaccess_plan_for_rows

(row_groups: &[u64], rows: &[u32])

src/df_vector/access.rs:123

↓ 1 callersMethodadd_candidates

(&mut self, idx: usize, candidates: Vec<u32>)

src/df_vector/access.rs:208

↓ 1 callersFunctionappend_index_inplace

(plan: ParquetIndexAppend<'_>)

src/ivf/parquet.rs:542

↓ 1 callersFunctionarray_literal

(values: &[f32])

benches/bench_util.rs:67

↓ 1 callersFunctionbuild_access_plans

( files: &[FileEntry], selections: &HashMap<String, Vec<u32>>, )

src/df_vector/access.rs:107

↓ 1 callersFunctionbuild_batch_from_rows

(schema: SchemaRef, rows: Vec<Vec<ScalarValue>>)

src/df_vector/exec.rs:552

↓ 1 callersFunctionbuild_context

(options: VectorTopKOptions)

src/df_vector/tests.rs:243

↓ 1 callersFunctionbuild_row_selection

(row_count: usize, rows: &[u32])

src/df_vector/access.rs:161

↓ 1 callersMethodbuild_topk_from_sort_merge

( &self, sort: &SortExec, limit_fetch: Option<usize>, )

src/df_vector/physical.rs:126

↓ 1 callersMethodcmp

(&self, other: &Self)

src/ivf/search.rs:33

↓ 1 callersMethodcmp

(&self, other: &Self)

src/df_vector/exec.rs:450

↓ 1 callersMethodcollect_candidates

( &self, context: Arc<TaskContext>, )

src/df_vector/exec.rs:108

↓ 1 callersFunctioncollect_column_write_options

( source: &Path, expected_columns: usize, )

src/ivf/parquet.rs:417

↓ 1 callersFunctioncollect_parquet_scans

( plan: &Arc<dyn ExecutionPlan>, scans: &mut Vec<ParquetScanInfo>, )

src/df_vector/access.rs:46

↓ 1 callersFunctioncolumn_statistics_level

(column: &ColumnChunkMetaData)

src/ivf/parquet.rs:479

↓ 1 callersFunctioncolumn_uses_dictionary

(column: &ColumnChunkMetaData)

src/ivf/parquet.rs:475

↓ 1 callersFunctioncompute_distance

(array: &ArrayRef, row: usize, query: &[f32])

src/df_vector/exec.rs:494

↓ 1 callersFunctiondata_page_encoding

(column: &ColumnChunkMetaData)

src/ivf/parquet.rs:489

↓ 1 callersFunctionembedding_column_path

( columns: &[parquet::schema::types::ColumnDescPtr], embedding_column: &EmbeddingColumn, )

src/ivf/parquet.rs:379

↓ 1 callersMethodexecute_index_scan

(&self, context: Arc<TaskContext>)

src/df_vector/index_exec.rs:83

↓ 1 callersMethodexecute_topk

( &self, context: Arc<TaskContext>, metrics: &VectorTopKMetricHandles, )

src/df_vector/exec.rs:279

↓ 1 callersMethodexecute_with_candidates

( &self, file_entries: &[FileEntry], context: Arc<TaskContext>, metrics: &Vect

src/df_vector/exec.rs:207

↓ 1 callersFunctionextract_array_distance_physical

(expr: &Arc<dyn PhysicalExpr>)

src/df_vector/physical.rs:198

↓ 1 callersFunctionextract_vector_row

( array: &dyn Array, row: usize, )

benches/query.rs:433

↓ 1 callersMethodfiles_with_candidates

( &self, scan: &ParquetScanInfo, context: Arc<TaskContext>, mut candidate_rows

src/df_vector/exec.rs:157

↓ 1 callersMethodfind_closest_centroids

(&self, query: &[f32], nprobe: usize)

src/ivf/index.rs:130

↓ 1 callersFunctionfirst_large_list_length

(list: &LargeListArray)

benches/query.rs:489

↓ 1 callersFunctionfirst_list_length

(list: &ListArray)

benches/query.rs:480

↓ 1 callersFunctionfixed_list_array_to_f32

(array: &FixedSizeListArray)

src/df_vector/expr.rs:36

↓ 1 callersFunctionget_embedding_at_row

( path: &std::path::Path, column_name: &str, row: usize, )

src/df_vector/tests.rs:253

↓ 1 callersFunctionhas_pq_vector_index

Returns true if the parquet file contains pq-vector index metadata.

src/ivf/parquet.rs:187

↓ 1 callersFunctionhash_vector

(array: &dyn Array, row: usize)

benches/query.rs:524

↓ 1 callersFunctionindex_dim

(index: &IvfIndex)

src/ivf/search.rs:150

↓ 1 callersFunctioninfer_vector_column

(schema: &SchemaRef)

benches/query.rs:364

↓ 1 callersFunctioninfer_vector_dim

(path: &Path, vector_column: &str)

benches/query.rs:386

↓ 1 callersFunctionis_vector_field

↓ 1 callersFunctionlarge_list_array_to_f32

(array: &LargeListArray)

src/df_vector/expr.rs:28

↓ 1 callersFunctionlist_array_to_f32

(array: &ListArray)

src/df_vector/expr.rs:17

↓ 1 callersMethodmetrics

(&self)

src/df_vector/exec.rs:396

↓ 1 callersFunctionnearest_centroid

(vec: &[f32], centroids: &[f32], dim: usize)

src/ivf/index.rs:244

↓ 1 callersMethodnext_batch

(&mut self, batch_size: usize)

src/df_vector/access.rs:214

↓ 1 callersMethodnprobe

(mut self, nprobe: usize)

src/ivf/search.rs:71

↓ 1 callersFunctionparallel_ranges

(len: usize, f: F)

src/ivf/index.rs:267

↓ 1 callersFunctionparse_index_metadata

( metadata: &FileMetaData, )

src/ivf/parquet.rs:120

↓ 1 callersFunctionrandom_query

(dim: usize, seed: u64)

benches/bench_util.rs:61

↓ 1 callersFunctionread_embeddings_for_rows

Read embeddings for specific rows using direct page reads.

src/ivf/search.rs:155

↓ 1 callersFunctionread_index_metadata

( path: &Path, )

src/ivf/parquet.rs:176

↓ 1 callersFunctionread_schema

(path: &Path)

benches/query.rs:358

next →1–100 of 167, ranked by callers