tensor-tang
8 years ago
parent
752f29a3ab
commit
d3a86c673b
@ -0,0 +1,120 @@
|
||||
/* Copyright (c) 2017 PaddlePaddle Authors. All Rights Reserve.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
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. */
|
||||
|
||||
#include "MKLDNNPoolLayer.h"
|
||||
#include "paddle/utils/Logging.h"
|
||||
|
||||
using namespace mkldnn; // NOLINT
|
||||
typedef memory::format format;
|
||||
|
||||
namespace paddle {
|
||||
|
||||
REGISTER_LAYER(mkldnn_pool, MKLDNNPoolLayer);
|
||||
|
||||
bool MKLDNNPoolLayer::init(const LayerMap& layerMap,
|
||||
const ParameterMap& parameterMap) {
|
||||
if (!MKLDNNLayer::init(layerMap, parameterMap)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void MKLDNNPoolLayer::reshape(
|
||||
int& bs, int& ic, int& ih, int& iw, int oc, int& oh, int& ow) {
|
||||
reshapeInput(bs, ih, iw);
|
||||
|
||||
// cal output sizes
|
||||
// oc can not be changed
|
||||
|
||||
reshapeOutput(oh, ow);
|
||||
resizeOutput(bs, oc * oh * ow);
|
||||
|
||||
printSizeInfo();
|
||||
}
|
||||
|
||||
void MKLDNNPoolLayer::resetFwd(std::vector<primitive>& pipeline,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& wgt,
|
||||
MKLDNNMatrixPtr& bias,
|
||||
MKLDNNMatrixPtr& out) {
|
||||
resetFwdBuffers(in, out);
|
||||
|
||||
resetFwdPD(fwdPD_, in, out);
|
||||
|
||||
resetFwdPipeline(pipeline, fwdPD_, in, out);
|
||||
|
||||
printValueFormatFlow();
|
||||
}
|
||||
|
||||
void MKLDNNPoolLayer::resetBwd(std::vector<primitive>& pipeline,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& wgt,
|
||||
MKLDNNMatrixPtr& bias,
|
||||
MKLDNNMatrixPtr& out) {
|
||||
std::shared_ptr<pool_bwd::primitive_desc> pd;
|
||||
|
||||
resetBwdBuffers(in, out);
|
||||
|
||||
resetBwdPD(pd, in, out);
|
||||
|
||||
resetBwdPipeline(pipeline, pd, in, out);
|
||||
|
||||
printGradFormatFlow();
|
||||
}
|
||||
|
||||
void MKLDNNPoolLayer::updateInputData() {
|
||||
inVal_->setData(getInputValue(0, CPU_DEVICE)->getData());
|
||||
}
|
||||
|
||||
void MKLDNNPoolLayer::resetFwdBuffers(MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out) {
|
||||
resetInValue(in);
|
||||
resetOutValue(out);
|
||||
}
|
||||
|
||||
void MKLDNNPoolLayer::resetInValue(MKLDNNMatrixPtr& in) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetOutValue(MKLDNNMatrixPtr& out) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetFwdPD(std::shared_ptr<pool_fwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr in,
|
||||
MKLDNNMatrixPtr out) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetFwdPipeline(
|
||||
std::vector<mkldnn::primitive>& pipeline,
|
||||
std::shared_ptr<pool_fwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetBwdBuffers(MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out) {
|
||||
resetOutGrad(out);
|
||||
resetInGrad(in);
|
||||
}
|
||||
void MKLDNNPoolLayer::resetOutGrad(MKLDNNMatrixPtr& out) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetInGrad(MKLDNNMatrixPtr& in) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetBwdPD(std::shared_ptr<pool_bwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out) {}
|
||||
|
||||
void MKLDNNPoolLayer::resetBwdPipeline(
|
||||
std::vector<mkldnn::primitive>& pipeline,
|
||||
std::shared_ptr<pool_bwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out) {}
|
||||
|
||||
} // namespace paddle
|
||||
@ -0,0 +1,93 @@
|
||||
/* Copyright (c) 2017 PaddlePaddle Authors. All Rights Reserve.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
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 "MKLDNNLayer.h"
|
||||
#include "mkldnn.hpp"
|
||||
|
||||
namespace paddle {
|
||||
typedef mkldnn::pooling_forward pool_fwd;
|
||||
typedef mkldnn::pooling_backward pool_bwd;
|
||||
|
||||
/**
|
||||
* @brief A subclass of MKLDNNLayer pool layer.
|
||||
*
|
||||
* The config file api is mkldnn_pool
|
||||
*/
|
||||
class MKLDNNPoolLayer : public MKLDNNLayer {
|
||||
protected:
|
||||
// save forward primitive_desc, which can be used backward
|
||||
std::shared_ptr<pool_fwd::primitive_desc> fwdPD_;
|
||||
|
||||
public:
|
||||
explicit MKLDNNPoolLayer(const LayerConfig& config) : MKLDNNLayer(config) {}
|
||||
|
||||
~MKLDNNPoolLayer() {}
|
||||
|
||||
bool init(const LayerMap& layerMap,
|
||||
const ParameterMap& parameterMap) override;
|
||||
|
||||
void reshape(
|
||||
int& bs, int& ic, int& ih, int& iw, int oc, int& oh, int& ow) override;
|
||||
|
||||
void resetFwd(std::vector<mkldnn::primitive>& pipeline,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& wgt,
|
||||
MKLDNNMatrixPtr& bias,
|
||||
MKLDNNMatrixPtr& out) override;
|
||||
|
||||
void resetBwd(std::vector<mkldnn::primitive>& pipeline,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& wgt,
|
||||
MKLDNNMatrixPtr& bias,
|
||||
MKLDNNMatrixPtr& out) override;
|
||||
|
||||
void updateInputData() override;
|
||||
|
||||
protected:
|
||||
/**
|
||||
* Forward functions: reset buffers(input, output),
|
||||
* reset primitive descriptor,
|
||||
* reset pipeline.
|
||||
*/
|
||||
void resetFwdBuffers(MKLDNNMatrixPtr& in, MKLDNNMatrixPtr& out);
|
||||
void resetInValue(MKLDNNMatrixPtr& in);
|
||||
void resetOutValue(MKLDNNMatrixPtr& out);
|
||||
void resetFwdPD(std::shared_ptr<pool_fwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr in,
|
||||
MKLDNNMatrixPtr out);
|
||||
void resetFwdPipeline(std::vector<mkldnn::primitive>& pipeline,
|
||||
std::shared_ptr<pool_fwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out);
|
||||
|
||||
/**
|
||||
* Backward functions: reset buffers(input, output),
|
||||
* reset primitive descriptor,
|
||||
* reset pipeline.
|
||||
*/
|
||||
void resetBwdBuffers(MKLDNNMatrixPtr& in, MKLDNNMatrixPtr& out);
|
||||
void resetOutGrad(MKLDNNMatrixPtr& out);
|
||||
void resetInGrad(MKLDNNMatrixPtr& in);
|
||||
void resetBwdPD(std::shared_ptr<pool_bwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out);
|
||||
void resetBwdPipeline(std::vector<mkldnn::primitive>& pipeline,
|
||||
std::shared_ptr<pool_bwd::primitive_desc>& pd,
|
||||
MKLDNNMatrixPtr& in,
|
||||
MKLDNNMatrixPtr& out);
|
||||
};
|
||||
|
||||
} // namespace paddle
|
||||
Loading…
Reference in new issue