shugeo
|
1bb3ae4b03
|
Shugeo unordered map (#256)
* Refactored usage of std::map to std::unordered_map instead.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Eliminated crash with wrong ShapeDescriptor hash.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Eliminated crash with TadDescriptor hash.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Refactored Stash hash.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Refactored hashes.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Refactored TadDescriptor hash and top_k mapping.
* Refactored hashes for ShapeDescriptor and TadDescriptor classes.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Refactored hash for ConstantDescriptor and ShapeDescriptor classes.
Signed-off-by: shugeo <sgazeos@gmail.com>
* Fixed map using with cuda platform.
Signed-off-by: shugeo <sgazeos@gmail.com>
* - few rearrangements for hash functions
- shared openblas allowed
Signed-off-by: raver119 <raver119@gmail.com>
* exports
Signed-off-by: raver119 <raver119@gmail.com>
* exports
Signed-off-by: raver119 <raver119@gmail.com>
* Stash reverted to std::map
Signed-off-by: raver119@gmail.com <raver119@gmail.com>
* Added additional test.
Signed-off-by: shugeo <sgazeos@gmail.com>
* different maps for different compilers
Signed-off-by: raver119 <raver119@gmail.com>
* missing include
Signed-off-by: raver119 <raver119@gmail.com>
* fix the leak
Signed-off-by: raver119 <raver119@gmail.com>
Co-authored-by: raver119 <raver119@gmail.com>
|
2020-02-24 07:51:01 +03:00 |
raver119
|
9e3c1b02b1
|
Perf improvements (#242)
* initial commit
Signed-off-by: raver119 <raver119@gmail.com>
* meh
Signed-off-by: raver119 <raver119@gmail.com>
* better ExpandDims impl
Signed-off-by: raver119 <raver119@gmail.com>
* better Squeeze impl
Signed-off-by: raver119 <raver119@gmail.com>
* better Softmax impl
Signed-off-by: raver119 <raver119@gmail.com>
* one test disabled
Signed-off-by: raver119 <raver119@gmail.com>
* more accurate impl
Signed-off-by: raver119 <raver119@gmail.com>
* - GraphProfiler now prints full shapeInfo instead of shape
- softmax typo fix
Signed-off-by: raver119 <raver119@gmail.com>
|
2020-02-14 16:20:31 +03:00 |
|
raver119
|
6de00bf75f
|
[WIP] Weekly update of repo (#8390)
* [WIP] Fix compilation after nd4j changes (#37)
* Fix compilation.
* Some tests fixed
* Disable tests temporarily.
* Restored test
* Tests restored.
* Test restored.
* [WIP] perf tests (#40)
* special maxpool test
Signed-off-by: raver119 <raver119@gmail.com>
* special maxpool test
Signed-off-by: raver119 <raver119@gmail.com>
* Shyrma bnorm bp (#41)
Batchnorm backprop mkldnn
* Add SameDiff memory reuse memory manager (array cache) (#39)
* Attention op comments
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* ArrayCacheMemoryMgr - first pass
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* Tweak array cache for use with SameDiff identity arrays
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* ArrayCacheMemoryMgr javadoc and properly get max memory
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* LRU cache policy + add tests
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* Fixes
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* Resize arrays internally if required for ArrayCacheMemoryMgr
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* Test improvement
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* Small polish
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* SameDiff op runtime benchmarking listener (#42)
Signed-off-by: AlexDBlack <blacka101@gmail.com>
* INLINE_LOOPS for windows
Signed-off-by: raver119 <raver119@gmail.com>
* [WIP] ThreadPool (#8)
This PR removes OpenMP use in 95% of cases
|
2019-11-13 17:15:18 +03:00 |
skymindops
|
b5f0ec072f
|
Eclipse Migration Initial Commit
|
2019-06-06 15:21:15 +03:00 |