pub struct Const(/* private fields */);Implementations§
Source§impl Const
impl Const
pub fn new(tensor: Arc<Tensor>) -> Result<Const, Error>
pub fn new_with_opaque_fact( tensor: Arc<Tensor>, fact: Box<dyn OpaqueFact>, ) -> Result<Const, Error>
pub fn new_with_opt_opaque_fact( tensor: Arc<Tensor>, fact: Option<Box<dyn OpaqueFact>>, ) -> Result<Const, Error>
pub fn val(&self) -> &Arc<Tensor>
pub fn opaque_fact(&self) -> Option<&(dyn OpaqueFact + 'static)>
Trait Implementations§
Source§impl EvalOp for Const
impl EvalOp for Const
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 Const
impl InferenceRulesOp for Const
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 Const
impl Op for Const
Source§impl TypedOp for Const
impl TypedOp for Const
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.
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
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.
fn change_axes( &self, _model: &Graph<TypedFact, Box<dyn TypedOp>>, _node: &Node<TypedFact, Box<dyn TypedOp>>, io: InOut, change: &AxisOp, ) -> Result<Option<AxisChangeConsequence>, Error>
fn axes_mapping( &self, inputs: &[&TypedFact], outputs: &[&TypedFact], ) -> Result<AxesMapping, Error>
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 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.
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
impl Eq for Const
impl StructuralPartialEq for Const
Auto Trait Implementations§
impl Freeze for Const
impl !RefUnwindSafe for Const
impl Send for Const
impl Sync for Const
impl Unpin for Const
impl !UnwindSafe for Const
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<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
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