Enum ModuleOperationIr 

pub enum ModuleOperationIr {
    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),
    MaxPool1d(MaxPool1dOpIr),
    MaxPool1dWithIndices(MaxPool1dWithIndicesOpIr),
    MaxPool1dWithIndicesBackward(MaxPool1dWithIndicesBackwardOpIr),
    MaxPool2d(MaxPool2dOpIr),
    MaxPool2dWithIndices(MaxPool2dWithIndicesOpIr),
    MaxPool2dWithIndicesBackward(MaxPool2dWithIndicesBackwardOpIr),
    Interpolate(InterpolateOpIr),
    InterpolateBackward(InterpolateBackwardOpIr),
    Rfft(RfftOpIr),
    IRfft(IRfftOpIr),
    Attention(AttentionOpIr),
    CtcLoss(CtcLossOpIr),
    CtcLossBackward(CtcLossBackwardOpIr),
}

Operation intermediate representation specific to module.

Variants

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.

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.

Rfft(RfftOpIr)

Operation corresponding to rfft

IRfft(IRfftOpIr)

Operation corresponding to irfft

Attention(AttentionOpIr)

Operation corresponding to attention.

CtcLoss(CtcLossOpIr)

Operation corresponding to ctc_loss.

CtcLossBackward(CtcLossBackwardOpIr)

Operation corresponding to ctc_loss_backward.

Trait Implementations

impl Clone for ModuleOperationIr

fn clone(&self) -> ModuleOperationIr

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModuleOperationIr

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

Formats the value using the given formatter.

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

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Hash for ModuleOperationIr

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

Feeds this value into the given Hasher.

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

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for ModuleOperationIr

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

Tests for self and other values to be equal, and is used by ==.

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::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for ModuleOperationIr