cavis/libnd4j/include/ops/declarable
Oleh 1d004b542a
xw_plus_b mkldnn implementation (#247)
* libnd4j first step of mkldnn for xw_plus_b and test of aurora crash in imageHelper

* libnd4j sync folders with master

* libnd4j merge master, raw implementation of xw_plus_b on mkldnn, clean up, need testing and adding checks for corresponded input shapes

* libnd4j corrections and checks added to xw_plus_b mkl

* libnd4j corrected dataType description based on mkl operation description, need more investigation

* libnd4j fixe xw_blus_b mkl implementation, need testing

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

* libnd4j two unit tests added

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

* libnd4j fixed check input dimensions bug

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

* libndj4 one more test added to cover different order handling

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

* libnd4j added optional int arg support to define weights format, if arg == 1, mkldnn (do not need transpose in mkldnn implementation), else mmul weights format, corrected check points, added unit test

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

* libnd4j merge master

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

* libnd4j some improvements to avoid NDArray transpose in xw_plus_b operation

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

* libnd4j fixed issues connected with weights rank, also added support of one case based on tf (for mkldnn, cpu, cuda), test case added

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

* libnd4j added proper handling of empty inputs (all implementations)

* libnd4j fixed compilation error

* libnd4j several more corrections after conflict solve and fixed typos

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

* libnd4j removed unsupported data types

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

* libnd4j merge master and fixed issues

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

* libnd4j added propagation implementation for xw_plus_b, fixed issue connected with mkl weights data format, avoided data copy in transpose mode, test cases added, manually tested with gradCheck

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

* libnd4j one minor fix of double operation declaration

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

* libnd4j code clean up

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

* libnd4j minor tests fixes

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

* libnd4j fixed build problem, integrate helpers changes

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

Co-authored-by: raver119 <raver119@gmail.com>
2020-03-31 13:03:10 +03:00
..
generic xw_plus_b mkldnn implementation (#247) 2020-03-31 13:03:10 +03:00
headers xw_plus_b mkldnn implementation (#247) 2020-03-31 13:03:10 +03:00
helpers libnd4j fixes for context sync in operation execution (#350) 2020-03-30 16:33:51 +03:00
impl Nullify (#304) 2020-03-20 08:49:28 +03:00
platform xw_plus_b mkldnn implementation (#247) 2020-03-31 13:03:10 +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 Learning updaters for gradient (#335) 2020-03-23 07:28:31 +03:00
DeclarableCustomOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
DeclarableListOp.h Improve ResultSet usage in libnd4j (#281) 2020-03-10 07:42:50 +03:00
DeclarableOp.h Nullify (#304) 2020-03-20 08:49:28 +03:00
DeclarableReductionOp.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
EmptyHandling.h Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +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 benchmarks fixes (#321) 2020-03-16 10:31:06 +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 Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +03:00
OpTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
PlatformHelper.h Nullify (#304) 2020-03-20 08:49:28 +03:00