cavis/libnd4j/include/ops
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
..
declarable xw_plus_b mkldnn implementation (#247) 2020-03-31 13:03:10 +03:00
impl [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
BroadcastBoolOpsTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
BroadcastIntOpsTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
BroadcastOpsTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
InputType.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gemm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
meta_ops.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
ops.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random_ops.h Shugeo random expo fix2 (#295) 2020-03-20 11:33:20 +03:00
special_random_ops.h Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +03:00
specials.h profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
specials_cuda.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
specials_sparse.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00