Skip to main content

cpu_capabilities

Function cpu_capabilities 

Source
pub const fn cpu_capabilities() -> &'static [OperationCapability]
Expand description

Return the CPU backend operation capability descriptor table.

ยงExamples

use tenferro_core_ops::PrimitiveOpKind;
use tenferro_cpu::cpu_capabilities;
use tenferro_tensor::{DType, SupportLevel};

let add = cpu_capabilities()
    .iter()
    .find(|entry| entry.op == PrimitiveOpKind::Add && entry.dtype == DType::F64)
    .unwrap();
assert_eq!(add.result, SupportLevel::Native);