Struct OperationCapability 

pub struct OperationCapability {
    pub backend: BackendId,
    pub op: PrimitiveOpKind,
    pub dtype: DType,
    pub output_dtype: DType,
    pub result: SupportLevel,
    pub read_inputs: SupportLevel,
    pub write_output: SupportLevel,
    pub strided_output: SupportLevel,
    pub accumulation: SupportLevel,
}

One backend capability entry for a primitive op and input dtype.

Examples

use tenferro_core_ops::PrimitiveOpKind;
use tenferro_tensor::{
    BackendId, CapabilityAxis, DType, OperationCapability, SupportLevel,
};

let entry = OperationCapability {
    backend: BackendId::Cpu,
    op: PrimitiveOpKind::Add,
    dtype: DType::F64,
    output_dtype: DType::F64,
    result: SupportLevel::Native,
    read_inputs: SupportLevel::Native,
    write_output: SupportLevel::Native,
    strided_output: SupportLevel::Native,
    accumulation: SupportLevel::Unsupported,
};
assert_eq!(entry.axis(CapabilityAxis::OwnedResult), SupportLevel::Native);

Fields

backend: BackendId

op: PrimitiveOpKind

dtype: DType

output_dtype: DType

result: SupportLevel

read_inputs: SupportLevel

write_output: SupportLevel

strided_output: SupportLevel

accumulation: SupportLevel

Implementations

impl OperationCapability

pub const fn axis(&self, axis: CapabilityAxis) -> SupportLevel

Return the support level for one capability axis.

Examples

use tenferro_core_ops::PrimitiveOpKind;
use tenferro_tensor::{
    BackendId, CapabilityAxis, DType, OperationCapability, SupportLevel,
};

let entry = OperationCapability {
    backend: BackendId::Cuda,
    op: PrimitiveOpKind::ReduceSum,
    dtype: DType::I32,
    output_dtype: DType::I32,
    result: SupportLevel::Native,
    read_inputs: SupportLevel::FallbackCopy,
    write_output: SupportLevel::Unsupported,
    strided_output: SupportLevel::Unsupported,
    accumulation: SupportLevel::Unsupported,
};
assert_eq!(entry.axis(CapabilityAxis::ReadInputs), SupportLevel::FallbackCopy);

Trait Implementations

impl Clone for OperationCapability

fn clone(&self) -> OperationCapability

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for OperationCapability

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Hash for OperationCapability

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for OperationCapability

fn eq(&self, other: &OperationCapability) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for OperationCapability

impl Eq for OperationCapability

impl StructuralPartialEq for OperationCapability