Skip to main content

build_weights_gpu

xlog_prob::compilation::gpu_weights

Function build_weights_gpu 

Source 
pub fn build_weights_gpu(
    vars: &GpuCnfVarTables,
    leaf_probs: &TrackedCudaSlice<f64>,
    choice_true: &TrackedCudaSlice<f64>,
    choice_false: &TrackedCudaSlice<f64>,
    evidence_by_var: &TrackedCudaSlice<u8>,
    provider: &Arc<CudaKernelProvider>,
) -> Result<GpuWeights>