7a7ee4b021
* - implementation of cudnn batchnorm_bp op Signed-off-by: Yurii <iuriish@yahoo.com> * - testing and fixing bugs in batchnorm_bp based on cudnn api Signed-off-by: Yurii <iuriish@yahoo.com> * - move pooling mkl code and delete some unnecessary files Signed-off-by: Yurii <iuriish@yahoo.com> * - implementation and testing cudnn pooling2d ops (avg/max, ff/bp) Signed-off-by: Yurii <iuriish@yahoo.com> * - implementation and testing cudnn pooling 3d (ff/bp) ops Signed-off-by: Yurii <iuriish@yahoo.com> * - provide ff step in case of cudnn maxpool3d_bp op Signed-off-by: Yurii <iuriish@yahoo.com> * - remove half type from set of supported types in mkl dpethwise conv op Signed-off-by: Yurii <iuriish@yahoo.com> * - bring back cudaStreamSynchronize in batchnorm and pooling cudnn ops Signed-off-by: Yurii <iuriish@yahoo.com> Co-authored-by: raver119 <raver119@gmail.com> |
||
|---|---|---|
| .. | ||
| activations | ||
| bitwise | ||
| blas | ||
| boolean | ||
| broadcastable | ||
| compat | ||
| datatypes | ||
| flow | ||
| grad | ||
| helpers | ||
| images | ||
| kernels | ||
| list | ||
| loss | ||
| nlp | ||
| nn | ||
| parity_ops | ||
| random | ||
| recurrent | ||
| shape | ||
| strings | ||
| tests | ||
| thrid_party | ||
| transforms | ||
| tsne | ||
| util | ||
| CustomOperations.cpp | ||
| README.md | ||
| parity_ops.cpp | ||
README.md
Generic ops
Put any cross platform operations here. Anything platform specific should be put in: in ops/helpers.