cavis/libnd4j/include/ops
Oleh ead5162c97
Tanh mkldnn implementation (#296)
* 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>

* libnd4j first spet of elementwize tanh implementation on mkldnn

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

* libnd4j fixed typo in error message for softmax MKLDNN, test case added, implementation of tanh on MKLDNN, need supported DataType testing

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

* libnd4j several fixes for tanh and temporary performance test added

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

* libnd4j fixed mkldnn platform loader for tanh

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

* libnd4j MklDnn tanh removed unsupported data types, removed performance test case, added more appropriate equivalence test case, code clean up

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

* libnd4j fixed problem with empty input case for MklDnn tanh and softmax

Signed-off-by: Oleg <oleg.semeniv@gmail.com>
2020-03-06 17:11:22 +03:00
..
declarable Tanh mkldnn implementation (#296) 2020-03-06 17:11:22 +03:00
impl profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
special_random_ops.h libnd4j polishing (#273) 2020-03-02 12:49:41 +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