009007120b
* Implementation for non_max_suppression_v3 was added. Initial version * Added check for overcome threshold. * Added definition for V3 method. * java remapping for NonMaxSuppressionV3 Signed-off-by: raver119 <raver119@gmail.com> * Fixed proporly processing of an empty output and test. * Refactored op to less threshold data to float. * Implemented cuda-based helper for non_max_suppression_v3 op. * Fixed fake_quant_with_min_max_vars op. * Fixed tests with float numbers. * - assert now stops execution - sortByKey/sortByValue now have input validation Signed-off-by: raver119 <raver119@gmail.com> * missing var Signed-off-by: raver119 <raver119@gmail.com> * Fixed proper processing for zero max_size inputs. * Refactored kernel callers. * Fixed return statement for logdet op helper. * Refactored unsorted segment SqrtN op. * get back 8 tail bytes on CUDA Signed-off-by: raver119 <raver119@gmail.com> * Refactored segment prod ops and helpers for cuda and tests. * Additional test. * CudaWorkspace tests updated for 8 tail bytes Signed-off-by: raver119 <raver119@gmail.com> * special atomic test Signed-off-by: raver119 <raver119@gmail.com> * atomicMul/atomicDiv fix for 16bit values Signed-off-by: raver119 <raver119@gmail.com> * Eliminated waste prints. |
||
|---|---|---|
| .. | ||
| activations | ||
| bitwise | ||
| blas | ||
| boolean | ||
| broadcastable | ||
| datatypes | ||
| flow | ||
| grad | ||
| helpers | ||
| kernels | ||
| 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.