Skip to main content

ModuleOperationIr

burn::backend::ir

Enum ModuleOperationIr 

pub enum ModuleOperationIr {

Show 53 variants    BatchNorm(BatchNormOpIr),
    Embedding(EmbeddingOpIr),
    EmbeddingBackward(EmbeddingBackwardOpIr),
    Linear(LinearOpIr),
    LinearXBackward(LinearXBackwardOpIr),
    LinearWeightBackward(LinearWeightBackwardOpIr),
    LinearBiasBackward(LinearBiasBackwardOpIr),
    Conv1d(Conv1dOpIr),
    Conv1dXBackward(Conv1dXBackwardOpIr),
    Conv1dWeightBackward(Conv1dWeightBackwardOpIr),
    Conv1dBiasBackward(Conv1dBiasBackwardOpIr),
    Conv2d(Conv2dOpIr),
    Conv2dXBackward(Conv2dXBackwardOpIr),
    Conv2dWeightBackward(Conv2dWeightBackwardOpIr),
    Conv2dBiasBackward(Conv2dBiasBackwardOpIr),
    Conv3d(Conv3dOpIr),
    Conv3dXBackward(Conv3dXBackwardOpIr),
    Conv3dWeightBackward(Conv3dWeightBackwardOpIr),
    Conv3dBiasBackward(Conv3dBiasBackwardOpIr),
    DeformableConv2d(Box<DeformConv2dOpIr>),
    DeformableConv2dBackward(Box<DeformConv2dBackwardOpIr>),
    ConvTranspose1d(ConvTranspose1dOpIr),
    ConvTranspose2d(ConvTranspose2dOpIr),
    ConvTranspose3d(ConvTranspose3dOpIr),
    AvgPool1d(AvgPool1dOpIr),
    AvgPool2d(AvgPool2dOpIr),
    AvgPool1dBackward(AvgPool1dBackwardOpIr),
    AvgPool2dBackward(AvgPool2dBackwardOpIr),
    AdaptiveAvgPool1d(AdaptiveAvgPool1dOpIr),
    AdaptiveAvgPool2d(AdaptiveAvgPool2dOpIr),
    AdaptiveAvgPool1dBackward(AdaptiveAvgPool1dBackwardOpIr),
    AdaptiveAvgPool2dBackward(AdaptiveAvgPool2dBackwardOpIr),
    AdaptiveAvgPool3d(AdaptiveAvgPool3dOpIr),
    AdaptiveAvgPool3dBackward(AdaptiveAvgPool3dBackwardOpIr),
    MaxPool1d(MaxPool1dOpIr),
    MaxPool1dWithIndices(MaxPool1dWithIndicesOpIr),
    MaxPool1dWithIndicesBackward(MaxPool1dWithIndicesBackwardOpIr),
    MaxPool2d(MaxPool2dOpIr),
    MaxPool2dWithIndices(MaxPool2dWithIndicesOpIr),
    MaxPool2dWithIndicesBackward(MaxPool2dWithIndicesBackwardOpIr),
    Interpolate(InterpolateOpIr),
    InterpolateBackward(InterpolateBackwardOpIr),
    Attention(AttentionOpIr),
    CtcLoss(CtcLossOpIr),
    CtcLossBackward(CtcLossBackwardOpIr),
    LayerNorm(LayerNormOpIr),
    Unfold4d(Unfold4dOpIr),
    ConvTranspose1dWeightBackward(ConvTranspose1dWeightBackwardOpIr),
    ConvTranspose1dBiasBackward(ConvTranspose1dBiasBackwardOpIr),
    ConvTranspose2dWeightBackward(ConvTranspose2dWeightBackwardOpIr),
    ConvTranspose2dBiasBackward(ConvTranspose2dBiasBackwardOpIr),
    ConvTranspose3dWeightBackward(ConvTranspose3dWeightBackwardOpIr),
    ConvTranspose3dBiasBackward(ConvTranspose3dBiasBackwardOpIr),

}
Expand description

Operation intermediate representation specific to module.

Variants§

§

BatchNorm(BatchNormOpIr)

Batch normalization with explicitly supplied statistics, corresponding to batch_norm.

§

Embedding(EmbeddingOpIr)

Operation corresponding to embedding.

§

EmbeddingBackward(EmbeddingBackwardOpIr)

Operation corresponding to embedding_backward.

§

Linear(LinearOpIr)

Operation corresponding to linear.

§

LinearXBackward(LinearXBackwardOpIr)

Operation corresponding to linear_x_backward.

§

LinearWeightBackward(LinearWeightBackwardOpIr)

Operation corresponding to linear_weight_backward.

§

LinearBiasBackward(LinearBiasBackwardOpIr)

Operation corresponding to linear_bias_backward.

§

Conv1d(Conv1dOpIr)

Operation corresponding to conv1d.

§

Conv1dXBackward(Conv1dXBackwardOpIr)

Operation corresponding to conv1d_x_backward.

§

Conv1dWeightBackward(Conv1dWeightBackwardOpIr)

Operation corresponding to conv1d_weight_backward.

§

Conv1dBiasBackward(Conv1dBiasBackwardOpIr)

Operation corresponding to conv1d_bias_backward.

§

Conv2d(Conv2dOpIr)

Operation corresponding to conv2d.

§

Conv2dXBackward(Conv2dXBackwardOpIr)

Operation corresponding to conv2d_x_backward.

§

Conv2dWeightBackward(Conv2dWeightBackwardOpIr)

Operation corresponding to conv2d_weight_backward.

§

Conv2dBiasBackward(Conv2dBiasBackwardOpIr)

Operation corresponding to conv2d_bias_backward.

§

Conv3d(Conv3dOpIr)

Operation corresponding to conv3d.

§

Conv3dXBackward(Conv3dXBackwardOpIr)

Operation corresponding to conv3d_x_backward.

§

Conv3dWeightBackward(Conv3dWeightBackwardOpIr)

Operation corresponding to conv3d_weight_backward.

§

Conv3dBiasBackward(Conv3dBiasBackwardOpIr)

Operation corresponding to conv3d_bias_backward.

§

DeformableConv2d(Box<DeformConv2dOpIr>)

Operation corresponding to deform_conv2d

§

DeformableConv2dBackward(Box<DeformConv2dBackwardOpIr>)

Operation corresponding to deform_conv2d_backward

§

ConvTranspose1d(ConvTranspose1dOpIr)

Operation corresponding to conv transpose 1d.

§

ConvTranspose2d(ConvTranspose2dOpIr)

Operation corresponding to conv transpose 2d.

§

ConvTranspose3d(ConvTranspose3dOpIr)

Operation corresponding to conv transpose 3d.

§

AvgPool1d(AvgPool1dOpIr)

Operation corresponding to avg pool 1d.

§

AvgPool2d(AvgPool2dOpIr)

Operation corresponding to avg pool 2d.

§

AvgPool1dBackward(AvgPool1dBackwardOpIr)

Operation corresponding to avg pool 1d backward.

§

AvgPool2dBackward(AvgPool2dBackwardOpIr)

Operation corresponding to avg pool 2d backward.

§

AdaptiveAvgPool1d(AdaptiveAvgPool1dOpIr)

Operation corresponding to adaptive avg pool 1d.

§

AdaptiveAvgPool2d(AdaptiveAvgPool2dOpIr)

Operation corresponding to adaptive avg pool 2d.

§

AdaptiveAvgPool1dBackward(AdaptiveAvgPool1dBackwardOpIr)

Operation corresponding to adaptive avg pool 1d backward.

§

AdaptiveAvgPool2dBackward(AdaptiveAvgPool2dBackwardOpIr)

Operation corresponding to adaptive avg pool 2d backward.

§

AdaptiveAvgPool3d(AdaptiveAvgPool3dOpIr)

Operation corresponding to adaptive avg pool 3d.

§

AdaptiveAvgPool3dBackward(AdaptiveAvgPool3dBackwardOpIr)

Operation corresponding to adaptive avg pool 3d backward.

§

MaxPool1d(MaxPool1dOpIr)

Operation corresponding to max pool 1d.

§

MaxPool1dWithIndices(MaxPool1dWithIndicesOpIr)

Operation corresponding to max pool 1d with indices.

§

MaxPool1dWithIndicesBackward(MaxPool1dWithIndicesBackwardOpIr)

Operation corresponding to max pool 1d with indices backward.

§

MaxPool2d(MaxPool2dOpIr)

Operation corresponding to max pool 2d.

§

MaxPool2dWithIndices(MaxPool2dWithIndicesOpIr)

Operation corresponding to max pool 2d with indices.

§

MaxPool2dWithIndicesBackward(MaxPool2dWithIndicesBackwardOpIr)

Operation corresponding to max pool 2d with indices backward.

§

Interpolate(InterpolateOpIr)

Operation corresponding to interpolate.

§

InterpolateBackward(InterpolateBackwardOpIr)

Operation corresponding to interpolate backward.

§

Attention(AttentionOpIr)

Operation corresponding to attention.

§

CtcLoss(CtcLossOpIr)

Operation corresponding to ctc_loss.

§

CtcLossBackward(CtcLossBackwardOpIr)

Operation corresponding to ctc_loss_backward.

§

LayerNorm(LayerNormOpIr)

Operation corresponding to layer_norm.

§

Unfold4d(Unfold4dOpIr)

Operation corresponding to unfold4d.

§

ConvTranspose1dWeightBackward(ConvTranspose1dWeightBackwardOpIr)

Operation corresponding to conv_transpose1d_weight_backward.

§

ConvTranspose1dBiasBackward(ConvTranspose1dBiasBackwardOpIr)

Operation corresponding to conv_transpose1d_bias_backward.

§

ConvTranspose2dWeightBackward(ConvTranspose2dWeightBackwardOpIr)

Operation corresponding to conv_transpose2d_weight_backward.

§

ConvTranspose2dBiasBackward(ConvTranspose2dBiasBackwardOpIr)

Operation corresponding to conv_transpose2d_bias_backward.

§

ConvTranspose3dWeightBackward(ConvTranspose3dWeightBackwardOpIr)

Operation corresponding to conv_transpose3d_weight_backward.

§

ConvTranspose3dBiasBackward(ConvTranspose3dBiasBackwardOpIr)

Operation corresponding to conv_transpose3d_bias_backward.

Trait Implementations§

§

impl Clone for ModuleOperationIr

§

fn clone(&self) -> ModuleOperationIr

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

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

Performs copy-assignment from source. Read more
§

impl Debug for ModuleOperationIr

§

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter. Read more
§

impl<'de> Deserialize<'de> for ModuleOperationIr

§

fn deserialize<__D>( __deserializer: __D, ) -> Result<ModuleOperationIr, <__D as Deserializer<'de>>::Error>
where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer. Read more
§

impl Hash for ModuleOperationIr

§

fn hash<__H>(&self, state: &mut __H)
where __H: Hasher,

Feeds this value into the given Hasher. Read more
1.3.0 · Source§

fn hash_slice<H>(data: &[Self], state: &mut H)
where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher. Read more
§

impl PartialEq for ModuleOperationIr

§

fn eq(&self, other: &ModuleOperationIr) -> bool

Tests for self and other values to be equal, and is used by ==.
1.0.0 (const: unstable) · Source§

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.
§

impl Serialize for ModuleOperationIr

§

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>
where __S: Serializer,

Serialize this value into the given Serde serializer. Read more
§

impl StructuralPartialEq for ModuleOperationIr

Auto Trait Implementations§

Blanket Implementations§

§

impl<T> Adaptor<()> for T

§

fn adapt(&self)

Adapt the type to be passed to a metric.
Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
§

impl<K, Q> Comparable<Q> for K
where K: Borrow<Q> + ?Sized, Q: Ord + ?Sized,

§

fn compare(&self, key: &Q) -> Ordering

Compare self to key and return their ordering.
Source§

impl<T> DeserializeOwned for T
where T: for<'de> Deserialize<'de>,

§

impl<K, Q> Equivalent<Q> for K
where K: Borrow<Q> + ?Sized, Q: Eq + ?Sized,

§

fn equivalent(&self, key: &Q) -> bool

Compare self to key and return true if they are equal.
§

impl<T> ErasedDestructor for T
where T: 'static,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

§

impl<T> Instrument for T

§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more
§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

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 more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

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
§

impl<T> Pointable for T

§

const ALIGN: usize

The alignment of pointer.
§

type Init = T

The type for initializers.
§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
§

impl<T> PolicyExt for T
where T: ?Sized,

§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns [Action::Follow] only if self and other return Action::Follow. Read more
§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns [Action::Follow] if either self or other returns Action::Follow. Read more
§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

§

fn vzip(self) -> V

§

impl<T> WithSubscriber for T

§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more
§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more