Struct ReduceStrategy

pub struct ReduceStrategy {
    pub use_planes: bool,
    pub shared: bool,
}

Fields

use_planes: bool

If true and the compute client support plane instructions, then try using them in the kernel. It could still be impossible to use plane instructions depending on the memory layout of the tensors.

shared: bool

If true, all units within a single cube cooperate to reduce a single item in the output. Else, each unit or plane (if planes is true) reduce a single item by itself.

Implementations

impl ReduceStrategy

pub fn validate<R: Runtime>( self, client: &ComputeClient<R::Server, R::Channel>, ) -> Result<Self, ReduceError>

pub fn new<R: Runtime>( client: &ComputeClient<R::Server, R::Channel>, shared: bool, ) -> Self

Trait Implementations

impl Clone for ReduceStrategy

fn clone(&self) -> ReduceStrategy

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ReduceStrategy

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ReduceStrategy

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Hash for ReduceStrategy

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 ReduceStrategy

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

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for ReduceStrategy

impl Eq for ReduceStrategy

impl StructuralPartialEq for ReduceStrategy