FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
CaffeOnACL/include/caffe/layers/acl_sigmoid_layer.hpp at master · 2php/CaffeOnACL · GitHub
2php
CaffeOnACL
Repository navigation
Code
Pull requests
Actions
Projects
Wiki
Security and quality
Insights
Expand file tree
Breadcrumbs
CaffeOnACL
/
include
/
caffe
/
layers
/
acl_sigmoid_layer.hpp
Copy path
More file actions
More file actions
Latest commit
History
History
History
55 lines (46 loc) · 1.69 KB
Breadcrumbs
CaffeOnACL
/
include
/
caffe
/
layers
/
acl_sigmoid_layer.hpp
Copy path
File metadata and controls
55 lines (46 loc) · 1.69 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#
ifndef
CAFFE_ACL_SIGMOID_LAYER_HPP_
#
define
CAFFE_ACL_SIGMOID_LAYER_HPP_
#
include
<
vector
>
#
include
"
caffe/blob.hpp
"
#
include
"
caffe/layer.hpp
"
#
include
"
caffe/proto/caffe.pb.h
"
#
include
"
caffe/layers/neuron_layer.hpp
"
#
include
"
caffe/layers/sigmoid_layer.hpp
"
#
ifdef
USE_ACL
#
include
"
caffe/acl_layer.hpp
"
#
include
"
caffe/layers/acl_base_activation_layer.hpp
"
#
endif
namespace
caffe
{
#
ifdef
USE_ACL
/*
*
* @brief ACL acceleration of SigmoidLayer.
*/
template
<
typename
Dtype>
class
ACLSigmoidLayer
:
public
ACLBaseActivationLayer
<Dtype>,public SigmoidLayer<Dtype> {
public:
explicit
ACLSigmoidLayer
(
const
LayerParameter& param)
: ACLBaseActivationLayer<Dtype>(param),SigmoidLayer<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
~ACLSigmoidLayer
();
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, ActivationLayerInfo::ActivationFunction type);
};
#
endif
}
//
namespace caffe
#
endif
//
CAFFE_ACL_SIGMOID_LAYER_HPP_
Back
|
FazBrowse Home
|
New Git URL