Struct tract_tensorflow::ops::nn::s2b::unary::BatchToSpaceUnary

pub struct BatchToSpaceUnary { /* fields omitted */ }

Methods

impl BatchToSpaceUnary

pub fn new(     datum_type: DatumType,     batch_shape: TVec<TDim>,     space_shape: TVec<TDim>,     block_shape: Array1<i32>,     pad: Vec<PaddingStrat> ) -> Self

Constructs a new BatchToSpaceUnary.

Trait Implementations

impl Clone for BatchToSpaceUnary

fn clone(&self) -> BatchToSpaceUnary

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for BatchToSpaceUnary

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

Formats the value using the given formatter.

impl StatelessOp for BatchToSpaceUnary

fn eval(&self, inputs: TVec<SharedTensor>) -> TractResult<TVec<SharedTensor>>

impl Op for BatchToSpaceUnary

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

fn infer(     &self,     inputs: SmallVec<[&TensorFact; 4]>,     outputs: SmallVec<[&TensorFact; 4]> ) -> Result<(SmallVec<[TensorFact; 4]>, SmallVec<[TensorFact; 4]>), TractError>

Infers properties about the input and output tensors.

fn reduce(     &self,     _inputs: SmallVec<[&TensorFact; 4]>,     _outputs: SmallVec<[&TensorFact; 4]>,     _phase: ReductionPhase ) -> Result<Option<ReducedOpRewire>, TractError>

fn pulsify(     &self,     _inputs: SmallVec<[&PulsedTensorFact; 4]> ) -> Result<Vec<PulsifiedOp>, TractError>

fn const_value(&self) -> Option<SharedTensor>

fn rounding_errors(&self) -> bool

fn noutputs(&self) -> usize

fn same_as(&self, _other: &(dyn Op + 'static)) -> bool

fn info(&self) -> Result<Option<String>, TractError>

impl InferenceRulesOp for BatchToSpaceUnary

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

Registers the inference rules of the operator.