cavis/libnd4j/include/ops/declarable
Oleh 4d81af9fe9
Softmax operation implementation for mkldnn (#286)
* libnd4j first step of softmax mkldnn implementation

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j raw implementation of mkldnn softmax

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j merge master and added softmax to MklDnnTests

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j some corrections for softmax mkldnn

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j merge branch, fixed problem with negative axis, fixed dnnl::memory::format_tag selection, test cases added

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j minor corrections to avoid risk connected with negative axis usage

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j fixed windows builds, added switcher to use mkldnn sofmax version only for 3D, 4D, 5D, 6D arrays

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j fixed dataType selection per request

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j fix for mac and windows builds

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j builds fix

Signed-off-by: Oleg <oleg.semeniv@gmail.com>
2020-03-04 19:36:42 +03:00
..
generic profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
headers libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
helpers simplified addBias2D for CUDA (#285) 2020-03-04 09:50:55 +03:00
impl profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
platform Softmax operation implementation for mkldnn (#286) 2020-03-04 19:36:42 +03:00
BooleanOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
BroadcastableOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
CustomOperations.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
DeclarableCustomOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
DeclarableListOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
DeclarableOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
DeclarableReductionOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
EmptyHandling.h [WIP] Create and small fix (#67) 2019-11-21 13:31:20 +03:00
LegacyBroadcastBoolOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyBroadcastOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyIndexReduceOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyPairwiseTransformBoolOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyPairwiseTransformOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyRandomOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyReduce3Op.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyReduceBoolOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyReduceFloatOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyReduceLongOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyReduceOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyReduceSameOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyScalarBoolOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyScalarOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyStatsOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyTransformAnyOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyTransformBoolOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyTransformFloatOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyTransformOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyTransformSameOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LegacyTransformStrictOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
LogicOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
OpDescriptor.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
OpRegistrator.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
OpTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
PlatformHelper.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00