eea3062ccf
* - start working on space_to_batch_nd Signed-off-by: Yurii <yurii@skymind.io> * - provide cpu helper for space_to_batch_nd op Signed-off-by: Yurii <yurii@skymind.io> * few typos fixed Signed-off-by: raver119 <raver119@gmail.com> * - add tests for space_to_batch and correct bugs Signed-off-by: Yurii <yurii@skymind.io> * - write cuda kernel for space_to_batch op Signed-off-by: Yurii <yurii@skymind.io> * - add order argument to shape::index2coords method in convolution cuda ops Signed-off-by: Yurii <yurii@skymind.io> * - restore some previous code Signed-off-by: Yurii <yurii@skymind.io> * old col2im kernel activated Signed-off-by: raver119 <raver119@gmail.com> * - change coords calculation in col2im kernel Signed-off-by: Yurii <yurii@skymind.io> * - restore old col2im kernel Signed-off-by: Yurii <yurii@skymind.io> * - add custom op for batch_to_space Signed-off-by: Yurii <yurii@skymind.io> * - provide cpu version for batch_to_space_nd op Signed-off-by: Yurii <yurii@skymind.io> * - provide cuda kernel for batch_to_space_nd op Signed-off-by: Yurii <yurii@skymind.io> |
||
|---|---|---|
| .. | ||
| activations | ||
| bitwise | ||
| blas | ||
| boolean | ||
| broadcastable | ||
| convo | ||
| datatypes | ||
| flow | ||
| grad | ||
| helpers | ||
| list | ||
| loss | ||
| nlp | ||
| nn | ||
| parity_ops | ||
| random | ||
| recurrent | ||
| shape | ||
| tests | ||
| thrid_party | ||
| transforms | ||
| tsne | ||
| 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.