Struct InputSpecializationRequirements 

pub struct InputSpecializationRequirements { /* private fields */ }

Per-input specialization requirements.

Examples

use tenferro_runtime::InputSpecializationRequirements;

let requirements = InputSpecializationRequirements::builder().build()?;
assert!(!requirements.specializes_dtype());

Implementations

impl InputSpecializationRequirements

pub fn builder() -> InputSpecializationRequirementsBuilder

Return a new requirements builder.

Examples

use tenferro_runtime::InputSpecializationRequirements;

let mut builder = InputSpecializationRequirements::builder();
builder.dtype(true);
assert!(builder.build()?.specializes_dtype());

pub fn specializes_dtype(&self) -> bool

Return whether dtype is specialized.

Examples

use tenferro_runtime::InputSpecializationRequirements;

let mut builder = InputSpecializationRequirements::builder();
builder.dtype(true);
assert!(builder.build()?.specializes_dtype());

pub fn specializes_rank(&self) -> bool

Return whether rank is specialized.

Examples

use tenferro_runtime::InputSpecializationRequirements;

let mut builder = InputSpecializationRequirements::builder();
builder.rank(true);
assert!(builder.build()?.specializes_rank());

pub fn concrete_dimensions(&self) -> &[u32]

Return concrete-dimension axes.

Examples

use tenferro_runtime::InputSpecializationRequirements;

let mut builder = InputSpecializationRequirements::builder();
builder.rank(true).concrete_dimensions(vec![1]);
assert_eq!(builder.build()?.concrete_dimensions(), &[1]);

pub fn placement(&self) -> PlacementSpecialization

Return the placement specialization mode.

Examples

use tenferro_runtime::{InputSpecializationRequirements, PlacementSpecialization};

let mut builder = InputSpecializationRequirements::builder();
builder.placement(PlacementSpecialization::Device);
assert_eq!(builder.build()?.placement(), PlacementSpecialization::Device);

pub fn layout(&self) -> LayoutSpecialization

Return the layout specialization mode.

Examples

use tenferro_runtime::{InputSpecializationRequirements, LayoutSpecialization};

let mut builder = InputSpecializationRequirements::builder();
builder.layout(LayoutSpecialization::Class);
assert_eq!(builder.build()?.layout(), LayoutSpecialization::Class);

pub fn alignment_log2(&self) -> Option<u8>

Return the required alignment class, if any.

Examples

use tenferro_runtime::InputSpecializationRequirements;

let mut builder = InputSpecializationRequirements::builder();
builder.alignment_log2(Some(2));
assert_eq!(builder.build()?.alignment_log2(), Some(2));

Trait Implementations

impl Clone for InputSpecializationRequirements

fn clone(&self) -> InputSpecializationRequirements

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InputSpecializationRequirements

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

Formats the value using the given formatter.

impl Hash for InputSpecializationRequirements

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 InputSpecializationRequirements

fn eq(&self, other: &InputSpecializationRequirements) -> 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 Eq for InputSpecializationRequirements

impl StructuralPartialEq for InputSpecializationRequirements