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 InferenceRulesOp for SpaceToBatchUnary

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

Registers the inference rules of the operator.

impl Op for SpaceToBatchUnary

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

fn declutter(
    &self,
    model: &TypedModel,
    node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

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 pulsify(
    &self,
    _source: &Model<NormalizedTensorInfo>,
    _node: &Node<NormalizedTensorInfo>,
    _target: &mut Model<PulsedTensorFact>,
    _mapping: &HashMap<OutletId, OutletId, RandomState>
) -> Result<SmallVec<[OutletId; 4]>, TractError>

fn codegen(
    &self,
    _model: &Model<TypedTensorInfo>,
    _node: &Node<TypedTensorInfo>
) -> Result<Option<ModelPatch<TypedTensorInfo>>, TractError>

fn cost(
    &self,
    _inputs: &[&TypedTensorInfo]
) -> Result<SmallVec<[(Cost, TDim); 4]>, TractError>

fn rounding_errors(&self) -> bool

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

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

impl StatelessOp for SpaceToBatchUnary

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