Skip to main content

elementwise_trinary

Function elementwise_trinary 

Source
pub fn elementwise_trinary<R: Runtime>(
    inputs: [(&RudaTensor<R>, &[Mode]); 3],
    output_modes: &[Mode],
    combine_ab: BinaryOp,
    combine_abc: BinaryOp,
) -> Result<RudaTensor<R>>
Expand description

Combine three broadcast operands as (A op_ab B) op_abc C.