| 53 | |
| 54 | template <typename Dtype> |
| 55 | void PReLULayer<Dtype>::Reshape(const vector<Blob<Dtype>*>& bottom, |
| 56 | const vector<Blob<Dtype>*>& top) { |
| 57 | CHECK_GE(bottom[0]->num_axes(), 2) |
| 58 | << "Number of axes of bottom blob must be >=2."; |
| 59 | top[0]->ReshapeLike(*bottom[0]); |
| 60 | if (bottom[0] == top[0]) { |
| 61 | // For in-place computation |
| 62 | bottom_memory_.ReshapeLike(*bottom[0]); |
| 63 | } |
| 64 | } |
| 65 | |
| 66 | template <typename Dtype> |
| 67 | void PReLULayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, |
no test coverage detected