Struct tract_tensorflow::ops::nn::s2b::unary::SpaceToBatchUnary

pub struct SpaceToBatchUnary {
    pub datum_type: DatumType,
    pub space_shape: TVec<TDim>,
    pub batch_shape: TVec<TDim>,
    pub block_shape: Array1<i32>,
    pub pad: TVec<PaddingStrat>,
}

Fields

datum_type: DatumType

space_shape: TVec<TDim>

batch_shape: TVec<TDim>

block_shape: Array1<i32>

pad: TVec<PaddingStrat>

Methods

impl SpaceToBatchUnary

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

Constructs a new SpaceToBatchUnary.

Trait Implementations

impl Clone for SpaceToBatchUnary

fn clone(&self) -> SpaceToBatchUnary

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for SpaceToBatchUnary

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

Formats the value using the given formatter.

impl StatelessOp for SpaceToBatchUnary

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

impl Op for SpaceToBatchUnary

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 SpaceToBatchUnary

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.