| 88 | |
| 89 | template <typename Dtype> |
| 90 | void Blob<Dtype>::set_cpu_data(Dtype* data) { |
| 91 | CHECK(data); |
| 92 | // Make sure CPU and GPU sizes remain equal |
| 93 | size_t size = count_ * sizeof(Dtype); |
| 94 | if (data_->size() != size) { |
| 95 | data_.reset(new SyncedMemory(size)); |
| 96 | diff_.reset(new SyncedMemory(size)); |
| 97 | } |
| 98 | data_->set_cpu_data(data); |
| 99 | } |
| 100 | |
| 101 | template <typename Dtype> |
| 102 | const Dtype* Blob<Dtype>::gpu_data() const { |
no test coverage detected