Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops/declarable/platform/mkldnn
History
Yurii Shyrma 0bc9785508
mkldnn concat call cases correction (#471) 
* - disable mkldnn concat when number of input arrays > 3072

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

* - get rid of loop in calculating of input arrays number

Signed-off-by: Yurii <iuriish@yahoo.com>
2020-05-19 21:56:41 +03:00
..
avgpooling2d.cpp
Shyrma weights format (#329)
2020-03-20 12:11:27 +03:00
avgpooling3d.cpp
Shyrma weights format (#329)
2020-03-20 12:11:27 +03:00
batchnorm.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
concat.cpp
mkldnn concat call cases correction (#471)
2020-05-19 21:56:41 +03:00
conv2d.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
conv3d.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
deconv2d_tf.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
deconv2d.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
deconv3d.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
depthwiseConv2d.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
lrn.cpp
libnd4j polishing (#273)
2020-03-02 12:49:41 +03:00
lstmLayer.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
matmul.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
maxpooling2d.cpp
Shyrma weights format (#329)
2020-03-20 12:11:27 +03:00
maxpooling3d.cpp
Shyrma weights format (#329)
2020-03-20 12:11:27 +03:00
mkldnnUtils.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
mkldnnUtils.h
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
softmax.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
tanh.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00
xw_plus_b.cpp
Shyrma merge max ind (#443)
2020-05-12 07:47:09 +03:00