Function melior::dialect::ods::tosa::argmax

source ·
pub fn argmax<'c>(
    context: &'c Context,
    output: Type<'c>,
    input: Value<'c, '_>,
    axis: IntegerAttribute<'c>,
    location: Location<'c>
) -> ArgMaxOperation<'c>
Expand description

Creates an argmax operation.