Modules§
- apply_
rope - binary
- broadcast
- cast
- causal_
conv1d_ update - change_
axes - concat
- copy_
based - Translators for ops that only need the generic copy_nd dispatch. These are fully backend-agnostic and can be constructed without any backend-specific arguments.
- diag_
gather - dyn_
kv_ cache - element_
wise - fused_
output - Writing a copy-based op’s output in the layout its consumer wants.
- gather
- gdn_
recurrent - gelu_
approximate - iff
- leaky_
relu - pad
- pulse
- reduce
- resize
- rms_
norm - rotate_
half - scaled_
masked_ softmax - slice
- softmax
- stft