Skip to content

GitLab

Commit fcc28e95 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

fix stride

parent 784994a5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
PyTorch/sparseconvnet/SCN/generic/CPU/SparseToDense.cpp
View file @ fcc28e95
......@@ -36,7 +36,7 @@ extern "C" void scn_DR_(SparseToDense_updateOutput)(THLongTensor *inputSize,
uInt nHot = r.size() / 2;
SparseToDense_ForwardPass<real>(iF, oF, nPlanes, spatialVolume, &r[0],
nHot);
oF += spatialVolume;
oF++;
}
}
extern "C" void scn_DR_(SparseToDense_updateGradInput)(
......@@ -58,7 +58,7 @@ extern "C" void scn_DR_(SparseToDense_updateGradInput)(
uInt nHot = r.size() / 2;
SparseToDense_BackwardPass<real>(diF, doF, nPlanes, spatialVolume, &r[0],
nHot);
doF += spatialVolume;
doF++;
}
}
#endif
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment