Unpad2D on GPU:

This commit is contained in:
George Hotz
2020-12-29 13:16:14 -05:00
parent 02655c07d5
commit 837aaacfbf
4 changed files with 32 additions and 24 deletions

View File

@@ -113,7 +113,7 @@ Relu, Log, Exp # unary ops
Sum, Max # reduce ops (with axis argument)
Dot, Conv2D # matrix multiplication and conv
Reshape, Transpose # moving things around ops
Unpad2D, Pad2D # stupid slices
Pad2D, Unpad2D # stupid slices
```
## ImageNet inference