CaffeOnACL/include/caffe/layers/acl_batch_norm_layer.hpp at master · dlyshare/CaffeOnACL · GitHub

Repository navigation

History

54 lines (45 loc) · 1.62 KB

Raw

#ifndef CAFFE_ACL_BATCH_NORMAL_HPP_

#define CAFFE_ACL_BATCH_NORMAL_HPP_

#include <vector>

#include "caffe/blob.hpp"

#include "caffe/layer.hpp"

#include "caffe/proto/caffe.pb.h"

#include "caffe/layers/batch_norm_layer.hpp"

#ifdef USE_ACL

#include "caffe/acl_layer.hpp"

#endif

namespace caffe {

#ifdef USE_ACL

* @brief ACL implementation of BatchNormLayer.

* Fallback to BatchNormLayer for some corner cases.

template <typename Dtype>

class ACLBatchNormLayer : public ACLBaseLayer<CLBatchNormalizationLayer,NEBatchNormalizationLayer>,public BatchNormLayer<Dtype> {

public:

explicit ACLBatchNormLayer(const LayerParameter& param)

: BatchNormLayer<Dtype>(param) {}

virtual void LayerSetUp(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual void Reshape(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual ~ACLBatchNormLayer();

protected:

virtual void Forward_gpu(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

virtual void Backward_gpu(const vector<Blob<Dtype>*>& top,

const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom){

NOT_IMPLEMENTED;

}

virtual void Backward_cpu(const vector<Blob<Dtype>*>& top,

const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom){

NOT_IMPLEMENTED;

}

virtual void SetupACLLayer(const vector<Blob<Dtype>*>& bottom,

const vector<Blob<Dtype>*>& top);

};

#endif

} // namespace caffe

#endif // CAFFE_ACL_BATCH_NORMAL_HPP_

CaffeOnACL/include/caffe/layers/acl_batch_norm_layer.hpp at master · dlyshare/CaffeOnACL · GitHub

Repository navigation

Expand file tree

Breadcrumbs

acl_batch_norm_layer.hpp

Latest commit

History

Breadcrumbs

acl_batch_norm_layer.hpp

File metadata and controls