From 13c9dc662081d4fe25e32ab1612efd084e488a8d Mon Sep 17 00:00:00 2001 From: Francesco Gatti Date: Fri, 24 Apr 2020 00:08:06 +0200 Subject: [PATCH] RegionRT softmax fix --- include/tkDNN/pluginsRT/RegionRT.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/tkDNN/pluginsRT/RegionRT.h b/include/tkDNN/pluginsRT/RegionRT.h index 176d676..f0d127e 100644 --- a/include/tkDNN/pluginsRT/RegionRT.h +++ b/include/tkDNN/pluginsRT/RegionRT.h @@ -61,7 +61,7 @@ public: //softmax start int index = entry_index(0, 0, coords + 1); softmaxForward( srcData + index, classes, batchSize*num, - (batchSize*c*h*w)/num, + (c*h*w)/num, w*h, 1, w*h, 1, dstData + index, stream); return 0;