Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

upsampling2d fix CUDA

Browse Source 
Signed-off-by: raver119 <raver119@gmail.com>
This commit is contained in:
raver119 2019-07-24 14:45:09 +03:00 committed by AlexDBlack
parent 4f2dae23a1
commit 2cd95dc517
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libnd4j/include/ops/declarable/helpers/cuda/convolutions.cu
View File

@ -1362,7 +1362,7 @@ static void depthwiseConv2dBP_(const NDArray* input, const NDArray* weights, con
gradBR = new NDArray(gradB->reshape(gradB->ordering(), {(int)gradB->lengthOf()})); gradBR = new NDArray(gradB->reshape(gradB->ordering(), {(int)gradB->lengthOf()}));
gradO->reduceAlongDimension(reduce::Sum, gradBR, {0,indOoH,indOoH+1}); // sum over bS, oH, oW gradO->reduceAlongDimension(reduce::Sum, gradBR, {0,indOoH,indOoH+1}); // sum over bS, oH, oW
if(gradBR != gradB) if(gradBR != gradB)
delete gradB; delete gradBR;
} }
//----- calculation of gradI -----// //----- calculation of gradI -----//