1 // Copyright 2014 BVLC and contributors.
4 - kloudkl@github, 2014.
12 #include "caffe/blob.hpp"
13 #include "caffe/common.hpp"
14 #include "caffe/layer.hpp"
15 #include "caffe/util/io.hpp"
16 #include "caffe/vision_layers.hpp"
21 template <typename Dtype>
22 Dtype HDF5OutputLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
23 vector<Blob<Dtype>*>* top) {
24 CHECK_GE(bottom.size(), 2);
25 CHECK_EQ(bottom[0]->num(), bottom[1]->num());
26 data_blob_.Reshape(bottom[0]->num(), bottom[0]->channels(),
27 bottom[0]->height(), bottom[0]->width());
28 label_blob_.Reshape(bottom[1]->num(), bottom[1]->channels(),
29 bottom[1]->height(), bottom[1]->width());
30 const int data_datum_dim = bottom[0]->count() / bottom[0]->num();
31 const int label_datum_dim = bottom[1]->count() / bottom[1]->num();
33 for (int i = 0; i < bottom[0]->num(); ++i) {
34 CUDA_CHECK(cudaMemcpy(&data_blob_.mutable_cpu_data()[i * data_datum_dim],
35 &bottom[0]->gpu_data()[i * data_datum_dim],
36 sizeof(Dtype) * data_datum_dim, cudaMemcpyDeviceToHost));
37 CUDA_CHECK(cudaMemcpy(&label_blob_.mutable_cpu_data()[i * label_datum_dim],
38 &bottom[1]->gpu_data()[i * label_datum_dim],
39 sizeof(Dtype) * label_datum_dim, cudaMemcpyDeviceToHost));
45 template <typename Dtype>
46 void HDF5OutputLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
47 const bool propagate_down, vector<Blob<Dtype>*>* bottom) {
51 INSTANTIATE_CLASS(HDF5OutputLayer);