diff options
author | Daya Khudia <dskhudia@fb.com> | 2019-07-16 03:34:48 +0300 |
---|---|---|
committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-07-16 03:47:38 +0300 |
commit | feca34d3d0b52527a1caa53186bc2ebf72d227c2 (patch) | |
tree | 09c856f5f8bbe03ff8d594c8c3083f6cb244ae49 /src | |
parent | e69972dad13a116049a8f9e6657c74ac5d04207a (diff) |
Add functions needed for unpacking in PackWeightsForConv (#106)
Summary:
Pull Request resolved: https://github.com/pytorch/FBGEMM/pull/106
The values returned by these functions is needed while unpacking weights.
Reviewed By: jianyuh
Differential Revision: D16193425
fbshipit-source-id: 8ee3a0dc46768d7cb572bf383be1ce2b450c44c9
Diffstat (limited to 'src')
-rw-r--r-- | src/PackWeightsForConv.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/PackWeightsForConv.cc b/src/PackWeightsForConv.cc index e16843c..085adc0 100644 --- a/src/PackWeightsForConv.cc +++ b/src/PackWeightsForConv.cc @@ -13,7 +13,8 @@ template <int SPATIAL_DIM, typename T, typename accT> PackWeightsForConv<SPATIAL_DIM, T, accT>::PackWeightsForConv( const conv_param_t<SPATIAL_DIM>& conv_p, const T* sdata, - const BlockingFactors* blocking_params) { + const BlockingFactors* blocking_params) + : conv_param_(conv_p) { static_assert( SPATIAL_DIM == 2 || SPATIAL_DIM == 3, "Only 2D and 3D convolutions are supported"); |