burn_backend/tensor/ops/mod.rs
1mod autodiff;
2mod base;
3mod bool;
4mod float;
5mod int;
6mod numeric;
7mod ordered;
8
9pub use autodiff::*;
10pub use base::*;
11pub use float::FloatMathOps;
12pub use numeric::*;
13pub use ordered::*;
14
15/// Computation to be used to update the existing values in indexed assignment operations (scatter/select).
16#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
17pub enum IndexingUpdateOp {
18 /// Overwrite existing values.
19 Assign,
20 /// Performs an addition.
21 Add,
22 /// Multiply existing values.
23 Mul,
24 /// Take element-wise minimum.
25 Min,
26 /// Take element-wise maximum.
27 Max,
28}