76f3553679
* - provide correct possible output types in mergeMaxIndex op Signed-off-by: Yurii <iuriish@yahoo.com> * - cleaning up the unneeded backprop arg in reverse_bp op Signed-off-by: Yurii <iuriish@yahoo.com> * - improve clipByNorm both ff and bp Signed-off-by: Yurii <iuriish@yahoo.com> * - implementation and testing clipByAvgNorm_bp op Signed-off-by: Yurii <iuriish@yahoo.com> * - pass biases in any way in dnnl lstm op, they are zeros when user doesn't provide them to us Signed-off-by: Yurii <iuriish@yahoo.com> * - start working on mkldnn concat op Signed-off-by: Yurii <iuriish@yahoo.com> * - further work on mkldnn concat Signed-off-by: Yurii <iuriish@yahoo.com> * missing declaration fix Signed-off-by: raver119@gmail.com <raver119@gmail.com> * - polishing mkl ops Signed-off-by: Yurii <iuriish@yahoo.com> * - testing and fixing bugs in mkl concat op Signed-off-by: Yurii <iuriish@yahoo.com> * - fix linkage error for windows cuda build Signed-off-by: Yurii <iuriish@yahoo.com> * - further conflicts resolving with master Signed-off-by: Yurii <iuriish@yahoo.com> * - fix format tags in mkldnn matmul op Signed-off-by: Yurii <iuriish@yahoo.com> * - provide additional type cast in clip.cu Signed-off-by: Yurii <iuriish@yahoo.com> * - finally bug in mkldnn tanh_bp was caught Co-authored-by: raver119@gmail.com <raver119@gmail.com> |
||
|---|---|---|
| .. | ||
| bitwise | ||
| blas | ||
| boolean | ||
| broadcastable | ||
| compat | ||
| compression | ||
| datatypes | ||
| flow | ||
| grad | ||
| helpers | ||
| images | ||
| kernels | ||
| linalg | ||
| list | ||
| loss | ||
| nlp | ||
| nn | ||
| parity_ops | ||
| random | ||
| reduce | ||
| shape | ||
| strings | ||
| tensor | ||
| tests | ||
| thrid_party | ||
| transforms | ||
| tsne | ||
| updaters | ||
| 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.