}
-template <typename Dtype>
class SplitLayerInsertionTest : public ::testing::Test {
protected:
void RunInsertionTest(
}
};
-typedef ::testing::Types<float> InsertionDtypes;
-TYPED_TEST_CASE(SplitLayerInsertionTest, InsertionDtypes);
-
-TYPED_TEST(SplitLayerInsertionTest, TestNoInsertion1) {
+TEST_F(SplitLayerInsertionTest, TestNoInsertion1) {
const string& input_proto =
"name: 'TestNetwork' "
"layers: { "
this->RunInsertionTest(input_proto, input_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestNoInsertion2) {
+TEST_F(SplitLayerInsertionTest, TestNoInsertion2) {
const string& input_proto =
"name: 'TestNetwork' "
"layers: { "
this->RunInsertionTest(input_proto, input_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestNoInsertionImageNet) {
+TEST_F(SplitLayerInsertionTest, TestNoInsertionImageNet) {
const string& input_proto =
"name: 'CaffeNet' "
"layers { "
this->RunInsertionTest(input_proto, input_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestNoInsertionWithInPlace) {
+TEST_F(SplitLayerInsertionTest, TestNoInsertionWithInPlace) {
const string& input_proto =
"name: 'TestNetwork' "
"layers: { "
this->RunInsertionTest(input_proto, input_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestInsertion) {
+TEST_F(SplitLayerInsertionTest, TestInsertion) {
const string& input_proto =
"name: 'TestNetwork' "
"layers: { "
this->RunInsertionTest(input_proto, expected_output_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestInsertionTwoTop) {
+TEST_F(SplitLayerInsertionTest, TestInsertionTwoTop) {
const string& input_proto =
"name: 'TestNetwork' "
"layers: { "
this->RunInsertionTest(input_proto, expected_output_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestInputInsertion) {
+TEST_F(SplitLayerInsertionTest, TestInputInsertion) {
const string& input_proto =
"name: 'TestNetwork' "
"input: 'data' "
this->RunInsertionTest(input_proto, expected_output_proto);
}
-TYPED_TEST(SplitLayerInsertionTest, TestWithInPlace) {
+TEST_F(SplitLayerInsertionTest, TestWithInPlace) {
const string& input_proto =
"name: 'TestNetwork' "
"layers: { "
namespace caffe {
-template <typename Dtype>
class PaddingLayerUpgradeTest : public ::testing::Test {
protected:
void RunPaddingUpgradeTest(
}
};
-typedef ::testing::Types<float> PaddingUpgradeDtypes;
-TYPED_TEST_CASE(PaddingLayerUpgradeTest, PaddingUpgradeDtypes);
-
-TYPED_TEST(PaddingLayerUpgradeTest, TestSimple) {
+TEST_F(PaddingLayerUpgradeTest, TestSimple) {
const string& input_proto =
"name: 'CaffeNet' "
"layers { "
this->RunPaddingUpgradeTest(input_proto, expected_output_proto);
}
-TYPED_TEST(PaddingLayerUpgradeTest, TestTwoTops) {
+TEST_F(PaddingLayerUpgradeTest, TestTwoTops) {
const string& input_proto =
"name: 'CaffeNet' "
"layers { "
this->RunPaddingUpgradeTest(input_proto, expected_output_proto);
}
-TYPED_TEST(PaddingLayerUpgradeTest, TestImageNet) {
+TEST_F(PaddingLayerUpgradeTest, TestImageNet) {
const string& input_proto =
"name: 'CaffeNet' "
"layers { "
this->RunPaddingUpgradeTest(input_proto, expected_output_proto);
}
-template <typename Dtype>
class V0UpgradeTest : public ::testing::Test {
protected:
void RunV0UpgradeTest(
}
};
-typedef ::testing::Types<float> V0UpgradeDtypes;
-TYPED_TEST_CASE(V0UpgradeTest, V0UpgradeDtypes);
-
-TYPED_TEST(V0UpgradeTest, TestSimple) {
+TEST_F(V0UpgradeTest, TestSimple) {
const string& input_proto =
"name: 'CaffeNet' "
"layers { "
}
// Test any layer or parameter upgrades not covered by other tests.
-TYPED_TEST(V0UpgradeTest, TestAllParams) {
+TEST_F(V0UpgradeTest, TestAllParams) {
const string& input_proto =
"name: 'CaffeNet' "
"input: 'input_data' "
this->RunV0UpgradeTest(input_proto, expected_output_proto);
}
-TYPED_TEST(V0UpgradeTest, TestImageNet) {
+TEST_F(V0UpgradeTest, TestImageNet) {
const string& input_proto =
"name: 'CaffeNet' "
"layers { "