Trait candle_core::op::UnaryOpT

source ·

pub trait UnaryOpT {
    const NAME: &'static str;
    const KERNEL: &'static str;
    const V: Self;
    const BF16_VEC: bool = false;
    const F16_VEC: bool = false;
    const F32_VEC: bool = false;
    const F64_VEC: bool = false;

    // Required methods
    fn bf16(v1: bf16) -> bf16;
    fn f16(v1: f16) -> f16;
    fn f32(v1: f32) -> f32;
    fn f64(v1: f64) -> f64;
    fn u8(v1: u8) -> u8;
    fn u32(v1: u32) -> u32;
    fn i64(v1: i64) -> i64;

    // Provided methods
    fn bf16_vec(_xs: &[bf16], _ys: &mut [bf16]) { ... }
    fn f16_vec(_xs: &[f16], _ys: &mut [f16]) { ... }
    fn f32_vec(_xs: &[f32], _ys: &mut [f32]) { ... }
    fn f64_vec(_xs: &[f64], _ys: &mut [f64]) { ... }
}