Struct tract_core::ops::array::Shape

pub struct Shape { /* fields omitted */ }

Methods

impl Shape

pub fn new(dt: DatumType) -> Self

Constructs a new Shape.

impl Shape

pub fn coerce_to<T>(shape: &[usize]) -> TractResult<Arc<Tensor>> where
    T: Copy + Datum,
    usize: AsPrimitive<T>, 

Trait Implementations

impl InferenceRulesOp for Shape

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 StatelessOp for Shape

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

Evaluates the operation given the input tensors.

impl Op for Shape

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

fn infer(
    &self,
    inputs: TVec<&TensorFact>,
    outputs: TVec<&TensorFact>
) -> TractResult<(TVec<TensorFact>, TVec<TensorFact>)>

Infers properties about the input and output tensors.

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

fn pulsify(
    &self,
    _source: &NormalizedModel,
    _node: &NormalizedNode,
    _target: &mut PulsedModel,
    _mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>

fn codegen(
    &self,
    _model: &TypedModel,
    _node: &TypedNode
) -> TractResult<Option<TypedModelPatch>>

fn cost(&self, _inputs: &[&TypedTensorInfo]) -> TractResult<TVec<(Cost, TDim)>>

fn rounding_errors(&self) -> bool

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

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

impl Clone for Shape

fn clone(&self) -> Shape

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Shape

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

Formats the value using the given formatter.