Skip to main content

Module dispatch

Module dispatch 

Source
Expand description

Once-per-batch kernel dispatch.

SPDX-License-Identifier: MIT OR Apache-2.0

Enums§

KernelImpl
Selected implementation class for a batch.

Functions§

accumulate_contingency
Public semantic entry: contingency table accumulation.
accumulate_contingency_rows
Public semantic entry: contingency accumulation over a row subset.
arch_simd_available
Whether an arch-SIMD path is compiled in and available on this CPU.
copy_vec
Public semantic entry: copy.
gather
Public semantic entry: gather.
masked_covariance
Public semantic entry: masked population covariance.
masked_mean
Public semantic entry: masked mean.
masked_sum
Public semantic entry: masked sum.
masked_variance
Public semantic entry: masked population variance.
pairwise_l1_fill
Public semantic entry: pairwise L1 distance matrix fill.
partial_correlation
Public semantic entry: partial correlation of x and y given z_cols.
select_impl
Resolve the implementation for a batch from policy.
standardize_inplace
Public semantic entry: in-place standardization (sample SD).
weighted_dot
Public semantic entry: weighted dot product.
weighted_mean
Public semantic entry: weighted mean.
weighted_sum
Public semantic entry: weighted sum (bootstrap weight accumulation).