relu UnaryOp is back

This commit is contained in:
George Hotz
2023-04-14 07:12:53 -07:00
parent 584ee6f616
commit 133521e730
4 changed files with 21 additions and 5 deletions

View File

@@ -145,7 +145,7 @@ hlops are syntactic sugar around mlops. They support most things torch does.
mlops are mid level ops. They understand derivatives. They are very simple.
```
Log, Exp # unary ops
Relu, Log, Exp # unary ops
Sum, Max # reduce ops (with axis argument)
Maximum, Add, Sub, Mul, Pow, Div, Equal # binary ops (no broadcasting, use expand)
Expand, Reshape, Permute, Pad, Shrink, Flip # movement ops