Skip to main content

build_trinary_key_raw

atomr_accel_cuda::kernel::tensor::elementwise

Function build_trinary_key_raw 

Source 
pub fn build_trinary_key_raw(
    dtype_tag: &'static str,
    modes_a: &[i32],
    modes_b: &[i32],
    modes_c: &[i32],
    modes_d: &[i32],
    extent_a: &[i64],
    extent_b: &[i64],
    extent_c: &[i64],
    extent_d: &[i64],
    alignment: u32,
    compute: ComputeDesc,
    op_a: cutensorOperator_t,
    op_b: cutensorOperator_t,
    op_c: cutensorOperator_t,
    op_ab: cutensorOperator_t,
    op_abc: cutensorOperator_t,
) -> PlanKey