summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 0fbb28d)
raw | patch | inline | side by side (parent: 0fbb28d)
author | Yangqing Jia <jiayq84@gmail.com> | |
Tue, 24 Sep 2013 22:55:25 +0000 (15:55 -0700) | ||
committer | Yangqing Jia <jiayq84@gmail.com> | |
Tue, 24 Sep 2013 22:55:25 +0000 (15:55 -0700) |
src/caffe/common.hpp | patch | blob | history | |
src/caffe/syncedmem.cpp | patch | blob | history | |
src/caffe/vision_layers.hpp | patch | blob | history |
diff --git a/src/caffe/common.hpp b/src/caffe/common.hpp
index c9b9f65e64ae8e355121766e1039c93afc77371a..67177a664b5ce88d766015148df30285a41858ca 100644 (file)
--- a/src/caffe/common.hpp
+++ b/src/caffe/common.hpp
#include <cublas_v2.h>
#include <cuda.h>
#include <curand.h>
-//cuda driver types
+// cuda driver types
#include <driver_types.h>
#include <glog/logging.h>
#include <mkl_vsl.h>
index 8e6996d64fbf9f46d29f136c317855acec3dba0d..c991d8d84cb88fcbf24a6173725fff41d7cea262 100644 (file)
--- a/src/caffe/syncedmem.cpp
+++ b/src/caffe/syncedmem.cpp
// Copyright 2013 Yangqing Jia
-#include <cstring>
#include <cuda_runtime.h>
+#include <cstring>
+
#include "caffe/common.hpp"
#include "caffe/syncedmem.hpp"
index 4744a95296fa5a1b29f0a938575ce77f4782c26d..336ed0fd516cb91f8dcb4dfdaa4e5fa3b71c2fbf 100644 (file)
#include <leveldb/db.h>
+#include <vector>
+
#include "caffe/layer.hpp"
namespace caffe {
class NeuronLayer : public Layer<Dtype> {
public:
explicit NeuronLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
};
class ReLULayer : public NeuronLayer<Dtype> {
public:
explicit ReLULayer(const LayerParameter& param)
- : NeuronLayer<Dtype>(param) {};
+ : NeuronLayer<Dtype>(param) {}
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class DropoutLayer : public NeuronLayer<Dtype> {
public:
explicit DropoutLayer(const LayerParameter& param)
- : NeuronLayer<Dtype>(param) {};
+ : NeuronLayer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class InnerProductLayer : public Layer<Dtype> {
public:
explicit InnerProductLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class PaddingLayer : public Layer<Dtype> {
public:
explicit PaddingLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class LRNLayer : public Layer<Dtype> {
public:
explicit LRNLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class Im2colLayer : public Layer<Dtype> {
public:
explicit Im2colLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class PoolingLayer : public Layer<Dtype> {
public:
explicit PoolingLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class ConvolutionLayer : public Layer<Dtype> {
public:
explicit ConvolutionLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
class DataLayer : public Layer<Dtype> {
public:
explicit DataLayer(const LayerParameter& param)
- : Layer<Dtype>(param) {};
+ : Layer<Dtype>(param) {}
virtual void SetUp(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);
+
protected:
virtual void Forward_cpu(const vector<Blob<Dtype>*>& bottom,
vector<Blob<Dtype>*>* top);