onnx-ir 0.21.0

ONNX-IR is a pure Rust library for parsing ONNX models into an intermediate representation that can be used to generate code for various ML/DL frameworks
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174

//! Node module contains implementations of ONNX operations.
//!
//! Each submodule implements a specific ONNX operation, providing:
//! - Operation configuration and parameters
//! - Rank inference functionality
//! - NodeProcessor implementation for node-centric processing
//!
//! This modular structure allows for clean separation of operation implementations
//! and facilitates easier maintenance and extension of the ONNX operation set.

#[cfg(test)]
pub mod test_utils;

pub mod elementwise;
pub mod elu;

// Unary element-wise operations
pub mod abs;
pub mod acos;
pub mod acosh;
pub mod asin;
pub mod asinh;
pub mod atan;
pub mod atanh;
pub mod bitwisenot;
pub mod ceil;
pub mod celu;
pub mod cos;
pub mod cosh;
pub mod erf;
pub mod exp;
pub mod floor;
pub mod gelu;
pub mod log;
pub mod neg;
pub mod not;
pub mod reciprocal;
pub mod round;
pub mod shrink;
pub mod sigmoid;
pub mod sign;
pub mod sin;
pub mod sinh;
pub mod softplus;
pub mod softsign;
pub mod sqrt;
pub mod swish;
pub mod tan;
pub mod tanh;
pub mod thresholded_relu;

// Binary element-wise operations
pub mod and;
pub mod argmax;
pub mod argmin;
pub mod arithmetic;
pub mod attention;
pub mod avg_pool1d;
pub mod avg_pool2d;
pub mod avg_pool3d;
pub mod batch_norm;
pub mod bernoulli;
pub mod bitshift;
pub mod bitwiseand;
pub mod bitwiseor;
pub mod bitwisexor;
pub mod blackman_window;
pub mod cast;
pub mod cast_like;
pub mod clip;
pub mod col2im;
pub mod comparison;
pub mod concat;
pub mod constant;
pub mod constant_of_shape;
pub mod conv1d;
pub mod conv2d;
pub mod conv3d;
pub mod conv_transpose1d;
pub mod conv_transpose2d;
pub mod conv_transpose3d;
pub mod cumsum;
pub mod deform_conv;
pub mod depth_to_space;
pub mod dequantize_linear;
pub mod det;
pub mod dft;
pub mod dropout;
pub mod einsum;
pub mod expand;
pub mod eye_like;
pub mod flatten;
pub mod gather;
pub mod gather_elements;
pub mod gathernd;
pub mod gemm;
pub mod global_avg_pool;
pub mod grid_sample;
pub mod group_norm;
pub mod gru;
pub mod hamming_window;
pub mod hann_window;
pub mod hard_sigmoid;
pub mod hard_swish;
pub mod hardmax;
pub mod identity;
pub mod if_node;
pub mod imputer;
pub mod instance_norm;
pub mod is_inf;
pub mod is_nan;
pub mod layer_norm;
pub mod leaky_relu;
pub mod linear;
pub mod log_softmax;
pub mod loop_node;
pub mod lp_normalization;
pub mod lp_pool1d;
pub mod lp_pool2d;
pub mod lrn;
pub mod lstm;
pub mod matmul;
pub mod matmulinteger;
pub mod max;
pub mod max_pool1d;
pub mod max_pool2d;
pub mod max_pool3d;
pub mod mean;
pub mod mean_variance_normalization;
pub mod mel_weight_matrix;
pub mod min;
pub mod mish;
pub mod modulo;
pub mod nonzero;
pub mod one_hot;
pub mod or;
pub mod pad;
pub mod padding;
pub mod pow;
pub mod prelu;
pub mod qlinear_matmul;
pub mod quantize_linear;
pub mod random;
pub mod random_like;
pub mod range;
pub mod reduce;
pub mod relu;
pub mod reshape;
pub mod resize;
pub mod rnn;
pub mod scaler;
pub mod scan_node;
pub mod scatter_elements;
pub mod scatter_nd;
pub mod selu;
pub mod shape;
pub mod size;
pub mod slice;
pub mod softmax;
pub mod space_to_depth;
pub mod split;
pub mod squeeze;
pub mod stft;
pub mod sum;
pub mod svmregressor;
pub mod tile;
pub mod topk;
pub mod transpose;
pub mod trilu;
pub mod unsqueeze;
pub mod unsupported;
pub mod where_op;
pub mod window_common;
pub mod xor;