Struct tract_core::ops::nn::LeakyRelu

pub struct LeakyRelu { /* fields omitted */ }

Methods

impl LeakyRelu

pub fn new(alpha: f32) -> Self

Constructs a new LeakyRelu.

Trait Implementations

impl InferenceRulesOp for LeakyRelu

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

Infers properties about the input and output tensors.

impl StatelessOp for LeakyRelu

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

impl Op for LeakyRelu

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 Default for LeakyRelu

fn default() -> LeakyRelu

Returns the "default value" for a type.

impl Clone for LeakyRelu

fn clone(&self) -> LeakyRelu

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for LeakyRelu

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

Formats the value using the given formatter.