1#![cfg_attr(not(feature = "kernel-ir"), no_std)]
2
3#[cfg(feature = "cann")]
4pub mod cann;
5
6pub mod attention;
7
8#[cfg(feature = "tensor-normalization")]
9pub mod normalization;
10
11#[cfg(feature = "tensor-gated-delta")]
12pub mod gated_delta;
13
14#[cfg(feature = "tensor-moe")]
15pub mod moe;
16
17#[cfg(feature = "kernel-ir")]
18pub mod convolution;
19
20#[cfg(feature = "pooling")]
21pub mod pooling;
22
23#[cfg(feature = "interpolation")]
24pub mod interpolation;
25
26#[cfg(feature = "grid-sample")]
27pub mod grid_sample;
28
29#[cfg(feature = "ctc")]
30pub mod ctc;
31
32
33
34
35
36#[cfg(feature = "tensor-paged-attention")]
37pub mod paged_attention;