| 19 | |
| 20 | template <typename TypeParam> |
| 21 | class BatchNormLayerTest : public MultiDeviceTest<TypeParam> { |
| 22 | typedef typename TypeParam::Dtype Dtype; |
| 23 | protected: |
| 24 | BatchNormLayerTest() |
| 25 | : blob_bottom_(new Blob<Dtype>(5, 2, 3, 4)), |
| 26 | blob_top_(new Blob<Dtype>()) { |
| 27 | // fill the values |
| 28 | FillerParameter filler_param; |
| 29 | GaussianFiller<Dtype> filler(filler_param); |
| 30 | filler.Fill(this->blob_bottom_); |
| 31 | blob_bottom_vec_.push_back(blob_bottom_); |
| 32 | blob_top_vec_.push_back(blob_top_); |
| 33 | } |
| 34 | virtual ~BatchNormLayerTest() { delete blob_bottom_; delete blob_top_; } |
| 35 | Blob<Dtype>* const blob_bottom_; |
| 36 | Blob<Dtype>* const blob_top_; |
| 37 | vector<Blob<Dtype>*> blob_bottom_vec_; |
| 38 | vector<Blob<Dtype>*> blob_top_vec_; |
| 39 | }; |
| 40 | |
| 41 | TYPED_TEST_CASE(BatchNormLayerTest, TestDtypesAndDevices); |
| 42 |
nothing calls this directly
no outgoing calls
no test coverage detected