Struct tract_hir::ops::nn::Reduce

pub struct Reduce { /* fields omitted */ }

Implementations

impl Reduce

pub fn new(axes: Option<Vec<i64>>, keep_dims: bool, reducer: Reducer) -> Self

Constructs a new Reduce.

impl Reduce

pub fn must_reduce(&self, ax: usize, rank: usize) -> bool

Trait Implementations

impl Clone for Reduce

fn clone(&self) -> Reduce

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Reduce

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

Formats the value using the given formatter.

impl DynHash for Reduce

fn dyn_hash(&self, state: &mut dyn Hasher)

impl Expansion for Reduce

fn name(&self) -> Cow<str>

fn info(&self) -> TractResult<Vec<String>>

fn op_families(&self) -> &'static [&'static str]

fn rules<'r, 'p: 'r, 's: 'r>(
    &'s self,
    s: &mut Solver<'r>,
    inputs: &'p [TensorProxy],
    outputs: &'p [TensorProxy]
) -> InferenceResult

fn wire(
    &self,
    name: &str,
    target: &mut TypedModel,
    inputs: &[OutletId]
) -> TractResult<TVec<OutletId>>

fn validation(&self) -> Validation

fn nboutputs(&self) -> TractResult<usize>

impl Hash for Reduce

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, 

Feeds a slice of this type into the given [Hasher].