ulysseB / telamon

A framework to find good combinations of optimizations for computational kernels on GPUs.
https://ulysseb.github.io/telamon/telamon
Apache License 2.0
23 stars 6 forks source link

Add kernel for ResNet cell #267

Closed andidr closed 5 years ago

andidr commented 5 years ago

Add a new kernel ResNetCell implementing one cell of ResNet [1]. The kernel computes:

O = activation(activation(A.B).C) + A

where A, B and C are matrices and activation an activation function (ReLU, Sigmoid or identity).

[1] https://arxiv.org/abs/1512.03385