cavis/libnd4j/include/ops/declarable/generic/nn
Yurii Shyrma 948646b32d
Shyrma mkl test (#211)
* - provide nhwc format in mkl conv ops

Signed-off-by: Yurii <iuriish@yahoo.com>

* - corrections in mkl conv3d

Signed-off-by: Yurii <iuriish@yahoo.com>

* - corrections in mkl batchnorm

Signed-off-by: Yurii <iuriish@yahoo.com>

* - corrections in mkl maxpooling2d

Signed-off-by: Yurii <iuriish@yahoo.com>

* - add format format_tag::any to outputs in mkl conv ops

Signed-off-by: Yurii <iuriish@yahoo.com>

* - complete corrections in mkl conv ops

Signed-off-by: Yurii <iuriish@yahoo.com>

* - add test for comparison of execution speeds of mkl conv2d op with different weights format

Signed-off-by: Yurii <iuriish@yahoo.com>

* - take into account order f in mkl conv ops

Signed-off-by: Yurii <iuriish@yahoo.com>
2020-02-06 21:12:54 +03:00
..
convo Null arrays fix (#208) 2020-02-02 23:14:00 +03:00
pooling Shyrma mkl test (#211) 2020-02-06 21:12:54 +03:00
apply_sgd.cpp [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
batchnorm.cpp Shyrma cudnn (#192) 2020-01-28 18:23:07 +03:00
dot_product_attention.cpp DataTypes & FlatBuffers (#197) 2020-01-30 10:07:24 +03:00
fusedBatchNorm.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
logSoftmax.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00
lrn.cpp Platform helpers (#8216) 2019-09-11 21:50:28 +03:00
multi_head_dot_product_attention.cpp DataTypes & FlatBuffers (#197) 2020-01-30 10:07:24 +03:00
relu_layer.cpp DataTypes & FlatBuffers (#197) 2020-01-30 10:07:24 +03:00
softmax.cpp Shyrma temp (#131) 2019-12-20 22:35:39 +03:00