tensor_forge/op.rs
1//! Defines support ML operations.
2
3/// This enum represents graph-level operation identity.
4///
5/// # Examples
6/// # TODO add examples
7#[derive(Clone, Debug, PartialEq, Eq, Hash)]
8pub enum OpKind {
9 /// Defines node as an input.
10 Input,
11 /// Matrix Multiplication operation.
12 MatMul,
13 /// Matrix Addition operation.
14 Add,
15 /// Matrix `ReLU` operation.
16 ReLU,
17}