tract_onnx/ops/
rec.rs

1use crate::model::OnnxOpRegister;
2
3pub mod common;
4pub mod gru;
5pub mod lstm;
6pub mod rnn;
7pub mod scan;
8
9pub fn register_all_ops(reg: &mut OnnxOpRegister) {
10    reg.insert("GRU", gru::gru);
11    reg.insert("LSTM", lstm::lstm);
12    reg.insert("RNN", rnn::rnn);
13    reg.insert("Scan", scan::scan);
14}