fix compile error

cross_channel_norm
tensor-tang 7 years ago
parent 2e101df7c6
commit 0f8aad2934

@ -27,7 +27,7 @@ protected:
size_t weightWidth_;
public:
MKLPackedGemm(MatrixPtr weight) {
explicit MKLPackedGemm(MatrixPtr weight) {
weightHeight_ = weight->getHeight();
weightWidth_ = weight->getWidth();
weightPacked_ =
@ -91,4 +91,5 @@ public:
cblas_sgemm_free(weightTPacked_);
}
};
} // namespace paddle

@ -12,6 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License. */
#pragma once
#include <gflags/gflags.h>
#include "Layer.h"
#include "MKLPackedGemm.h"
@ -128,4 +130,5 @@ protected:
std::unique_ptr<MKLPackedGemm> sgemm_packed_;
};
}
} // namespace paddle

Loading…
Cancel
Save