pub struct Pad {
pub pads: Vec<(usize, usize)>,
pub mode: PadMode,
}Fields§
§pads: Vec<(usize, usize)>§mode: PadModeImplementations§
Trait Implementations§
Source§impl EvalOp for Pad
impl EvalOp for Pad
fn is_stateless(&self) -> bool
fn eval( &self, inputs: SmallVec<[TValue; 4]>, ) -> Result<SmallVec<[TValue; 4]>, Error>
fn eval_with_session( &self, node_id: usize, session: &SessionState, inputs: SmallVec<[TValue; 4]>, ) -> Result<SmallVec<[TValue; 4]>, Error>
fn state( &self, session: &mut SessionState, node_id: usize, ) -> Result<Option<Box<dyn OpState>>, Error>
Source§impl InferenceRulesOp for Pad
impl InferenceRulesOp for Pad
Source§fn rules<'r, 'p: 'r, 's: 'r>(
&'s self,
s: &mut Solver<'r>,
inputs: &'p [TensorProxy],
outputs: &'p [TensorProxy],
) -> InferenceResult
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.
fn as_op(&self) -> &dyn Op
fn as_op_mut(&mut self) -> &mut dyn Op
fn to_typed( &self, _source: &InferenceModel, node: &InferenceNode, target: &mut TypedModel, mapping: &HashMap<OutletId, OutletId>, ) -> TractResult<TVec<OutletId>>
fn nboutputs(&self) -> TractResult<usize>
fn incorporate( &self, model: &InferenceModel, node: &InferenceNode, ) -> TractResult<Option<InferenceModelPatch>>
Source§impl Op for Pad
impl Op for Pad
fn name(&self) -> Cow<'static, str>
Source§fn info(&self) -> Result<Vec<String>, Error>
fn info(&self) -> Result<Vec<String>, Error>
Short (one-line) strings giving hints on internal implementation or
important configuration details to be displayed in dumps.
fn as_typed(&self) -> Option<&(dyn TypedOp + 'static)>
Source§fn validation(&self) -> Validation
fn validation(&self) -> Validation
The kind of accuracy check that should be performed on operation when
testing them.
Source§impl TypedOp for Pad
impl TypedOp for Pad
Source§fn output_facts(
&self,
inputs: &[&TypedFact],
) -> Result<SmallVec<[TypedFact; 4]>, Error>
fn output_facts( &self, inputs: &[&TypedFact], ) -> Result<SmallVec<[TypedFact; 4]>, Error>
Deduce output facts from input facts.
fn axes_mapping( &self, inputs: &[&TypedFact], outputs: &[&TypedFact], ) -> Result<AxesMapping, Error>
fn change_axes( &self, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, io: InOut, change: &AxisOp, ) -> Result<Option<AxisChangeConsequence>, Error>
Source§fn declutter(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>,
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn declutter( &self, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
Declutter the op to the tract_core operator set as much as possible.
Source§fn fuse(
&self,
_model: &Graph<TypedFact, Box<dyn TypedOp>>,
_node: &Node<TypedFact, Box<dyn TypedOp>>,
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn fuse( &self, _model: &Graph<TypedFact, Box<dyn TypedOp>>, _node: &Node<TypedFact, Box<dyn TypedOp>>, ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
Fuse op after codegen to deal with local optimisations.
Source§fn declutter_with_session(
&self,
session: &mut OptimizerSession<'_>,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>,
) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
fn declutter_with_session( &self, session: &mut OptimizerSession<'_>, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, ) -> Result<Option<ModelPatch<TypedFact, Box<dyn TypedOp>>>, Error>
Declutter the op to the tract_core operator set as much as possible.
Source§fn cost(
&self,
_inputs: &[&TypedFact],
) -> Result<SmallVec<[(Cost, TDim); 4]>, Error>
fn cost( &self, _inputs: &[&TypedFact], ) -> Result<SmallVec<[(Cost, TDim); 4]>, Error>
Computes a cost hint of the operation. Read more
fn suggested_axis_changes( &self, ) -> Result<SmallVec<[(InOut, AxisOp); 4]>, Error>
fn slice( &self, patch: &mut ModelPatch<TypedFact, Box<dyn TypedOp>>, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, prefix: &str, inputs: &[OutletId], output_axis: usize, start: &TDim, end: &TDim, ) -> Result<Option<SmallVec<[OutletId; 4]>>, Error>
Source§fn quantize(
&self,
model: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>,
dt: DatumType,
scale: f32,
zero_point: i32,
) -> Result<Option<Box<dyn TypedOp>>, Error>
fn quantize( &self, model: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, dt: DatumType, scale: f32, zero_point: i32, ) -> Result<Option<Box<dyn TypedOp>>, Error>
Transforms the op in an equivalent one, operating on dt (i8 or u8). Read more
Source§fn concretize_dims(
&self,
source: &Graph<TypedFact, Box<dyn TypedOp>>,
node: &Node<TypedFact, Box<dyn TypedOp>>,
target: &mut Graph<TypedFact, Box<dyn TypedOp>>,
mapping: &HashMap<OutletId, OutletId>,
values: &SymbolValues,
) -> Result<SmallVec<[OutletId; 4]>, Error>
fn concretize_dims( &self, source: &Graph<TypedFact, Box<dyn TypedOp>>, node: &Node<TypedFact, Box<dyn TypedOp>>, target: &mut Graph<TypedFact, Box<dyn TypedOp>>, mapping: &HashMap<OutletId, OutletId>, values: &SymbolValues, ) -> Result<SmallVec<[OutletId; 4]>, Error>
Transform the op into by providing a value to one or more symbols.
Auto Trait Implementations§
impl Freeze for Pad
impl RefUnwindSafe for Pad
impl Send for Pad
impl Sync for Pad
impl Unpin for Pad
impl UnwindSafe for Pad
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Downcast for Twhere
T: Any,
impl<T> Downcast for Twhere
T: Any,
Source§fn into_any(self: Box<T>) -> Box<dyn Any>
fn into_any(self: Box<T>) -> Box<dyn Any>
Convert
Box<dyn Trait> (where Trait: Downcast) to Box<dyn Any>. Box<dyn Any> can
then be further downcast into Box<ConcreteType> where ConcreteType implements Trait.Source§fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
fn into_any_rc(self: Rc<T>) -> Rc<dyn Any>
Convert
Rc<Trait> (where Trait: Downcast) to Rc<Any>. Rc<Any> can then be
further downcast into Rc<ConcreteType> where ConcreteType implements Trait.Source§fn as_any(&self) -> &(dyn Any + 'static)
fn as_any(&self) -> &(dyn Any + 'static)
Convert
&Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &Any’s vtable from &Trait’s.Source§fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
fn as_any_mut(&mut self) -> &mut (dyn Any + 'static)
Convert
&mut Trait (where Trait: Downcast) to &Any. This is needed since Rust cannot
generate &mut Any’s vtable from &mut Trait’s.Source§impl<T> DowncastSync for T
impl<T> DowncastSync for T
Source§impl<O> InferenceOp for Owhere
O: InferenceRulesOp + Op,
impl<O> InferenceOp for Owhere
O: InferenceRulesOp + Op,
Source§fn infer_facts(
&mut self,
inputs: SmallVec<[&InferenceFact; 4]>,
outputs: SmallVec<[&InferenceFact; 4]>,
observed: SmallVec<[&InferenceFact; 4]>,
) -> Result<(SmallVec<[InferenceFact; 4]>, SmallVec<[InferenceFact; 4]>, SmallVec<[InferenceFact; 4]>), Error>
fn infer_facts( &mut self, inputs: SmallVec<[&InferenceFact; 4]>, outputs: SmallVec<[&InferenceFact; 4]>, observed: SmallVec<[&InferenceFact; 4]>, ) -> Result<(SmallVec<[InferenceFact; 4]>, SmallVec<[InferenceFact; 4]>, SmallVec<[InferenceFact; 4]>), Error>
Infer properties about inputs and output tensors. This method does not
need to deal with the “trivial” stateless op with fully determined
inputs cases. Read more
fn nboutputs(&self) -> Result<usize, Error>
Source§fn observe_outlets(
&self,
_model: &Graph<InferenceFact, Box<dyn InferenceOp>>,
_node: &Node<InferenceFact, Box<dyn InferenceOp>>,
) -> Result<Vec<OutletId>, Error>
fn observe_outlets( &self, _model: &Graph<InferenceFact, Box<dyn InferenceOp>>, _node: &Node<InferenceFact, Box<dyn InferenceOp>>, ) -> Result<Vec<OutletId>, Error>
Allow an op to specify a supplementary list of outlets facts that
will trigger inference again.
Source§fn as_op_mut(&mut self) -> &mut (dyn Op + 'static)
fn as_op_mut(&mut self) -> &mut (dyn Op + 'static)
Reinterpret the InferenceOp as an Op, mutably.
Source§fn to_typed(
&self,
source: &Graph<InferenceFact, Box<dyn InferenceOp>>,
node: &Node<InferenceFact, Box<dyn InferenceOp>>,
target: &mut Graph<TypedFact, Box<dyn TypedOp>>,
mapping: &HashMap<OutletId, OutletId>,
) -> Result<SmallVec<[OutletId; 4]>, Error>
fn to_typed( &self, source: &Graph<InferenceFact, Box<dyn InferenceOp>>, node: &Node<InferenceFact, Box<dyn InferenceOp>>, target: &mut Graph<TypedFact, Box<dyn TypedOp>>, mapping: &HashMap<OutletId, OutletId>, ) -> Result<SmallVec<[OutletId; 4]>, Error>
Called during translation to TypedModel.
Source§fn incorporate(
&self,
model: &Graph<InferenceFact, Box<dyn InferenceOp>>,
node: &Node<InferenceFact, Box<dyn InferenceOp>>,
) -> Result<Option<ModelPatch<InferenceFact, Box<dyn InferenceOp>>>, Error>
fn incorporate( &self, model: &Graph<InferenceFact, Box<dyn InferenceOp>>, node: &Node<InferenceFact, Box<dyn InferenceOp>>, ) -> Result<Option<ModelPatch<InferenceFact, Box<dyn InferenceOp>>>, Error>
Early pass on inference model, after analyse, but before translation to
typed network. Meant to deal with some framework idiosyncrasies that
manifest with temporaries nodes that can run some form of inference but
require refactoring the network before it can be evaluated. Read more
Source§fn infer(
&mut self,
inputs: TVec<&InferenceFact>,
outputs: TVec<&InferenceFact>,
observed: TVec<&InferenceFact>,
) -> TractResult<(TVec<InferenceFact>, TVec<InferenceFact>, TVec<InferenceFact>)>
fn infer( &mut self, inputs: TVec<&InferenceFact>, outputs: TVec<&InferenceFact>, observed: TVec<&InferenceFact>, ) -> TractResult<(TVec<InferenceFact>, TVec<InferenceFact>, TVec<InferenceFact>)>
Infers properties about the input and output tensors. Read more
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more