Skip to main content

build_permutation_key_raw

atomr_accel_cuda::kernel::tensor::permute

Function build_permutation_key_raw 

Source 
pub fn build_permutation_key_raw(
    dtype_tag: &'static str,
    modes_a: &[i32],
    modes_b: &[i32],
    extent_a: &[i64],
    extent_b: &[i64],
    alignment: u32,
    compute: ComputeDesc,
    op_a: cutensorOperator_t,
) -> PlanKey