Struct tract_tensorflow::ops::nn::Softmax

pub struct Softmax {}

Methods

impl Softmax

pub fn build(_pb: &NodeDef) -> TractResult<Box<dyn Op>>

Trait Implementations

impl Clone for Softmax

fn clone(&self) -> Softmax

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Softmax

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

Formats the value using the given formatter.

impl StatelessOp for Softmax

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

Evaluates the operation given the input tensors.

impl Op for Softmax

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

fn rounding_errors(&self) -> bool

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 noutputs(&self) -> usize

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

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

impl InferenceRulesOp for Softmax

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.