diff options
| author | Soumith Chintala <soumith@gmail.com> | 2016-09-14 02:13:21 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-09-14 02:13:21 +0300 |
| commit | 8f2b88a5a4f2b541b9efc2fb72a47572811efd17 (patch) | |
| tree | 428d6b71ebd9cfa6f691c222695f67a8e81aecc6 | |
| parent | fde63523f149b61fc7a95be70d58b1646380966e (diff) | |
adding storageOffset in VolumetricConvolutionMM
| -rw-r--r-- | lib/THNN/generic/VolumetricConvolutionMM.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THNN/generic/VolumetricConvolutionMM.c b/lib/THNN/generic/VolumetricConvolutionMM.c index ef49a30..4d3b873 100644 --- a/lib/THNN/generic/VolumetricConvolutionMM.c +++ b/lib/THNN/generic/VolumetricConvolutionMM.c @@ -10,7 +10,7 @@ static int THNN_(view_weight)(THTensor **_weight) if (weight->nDimension == 5) { long s1 = weight->size[0]; long s2 = weight->size[1] * weight->size[2] * weight->size[3] * weight->size[4]; - *_weight = THTensor_(newWithStorage2d)(weight->storage, 0, s1, -1, s2, -1); + *_weight = THTensor_(newWithStorage2d)(weight->storage, weight->storageOffset, s1, -1, s2, -1); return 1; } return 0; |