Trait TensorMetadataPrims

Source

pub trait TensorMetadataPrims {
    type Plan;
    type Context;

    // Required methods
    fn plan(
        ctx: &mut Self::Context,
        desc: &MetadataPrimsDescriptor,
        inputs: &[MetadataTensorRef<'_>],
        output: MetadataTensorMut<'_>,
    ) -> Result<Self::Plan>;
    fn execute(
        ctx: &mut Self::Context,
        plan: &Self::Plan,
        inputs: &[MetadataTensorRef<'_>],
        output: MetadataTensorMut<'_>,
    ) -> Result<()>;
    fn has_metadata_support(desc: MetadataPrimsDescriptor) -> bool;
}

Expand description

Metadata tensor planning and execution protocol.

Metadata execution is overwrite-based and uses erased integer/bool metadata tensor handles instead of scalar-family alpha / beta scaling.

§Examples

use tenferro_prims::{
    MetadataGenerateOp, MetadataPrimsDescriptor, MetadataTensorMut,
    MetadataTensorRef, TensorMetadataPrims,
};

fn accepts_family<F: TensorMetadataPrims>(_: &F) {}

let _ = MetadataPrimsDescriptor::Generate {
    op: MetadataGenerateOp::IotaStartZero,
    output_dtype: tenferro_prims::MetadataDType::I32,
};