Skip to main content

build_reduction_key_raw

atomr_accel_cuda::kernel::tensor::reduce

Function build_reduction_key_raw 

Source 
pub fn build_reduction_key_raw(
    dtype_tag: &'static str,
    modes_a: &[i32],
    modes_c: &[i32],
    extent_a: &[i64],
    extent_c: &[i64],
    alignment: u32,
    compute: ComputeDesc,
    op_reduce: cutensorOperator_t,
) -> PlanKey