|
|
|
@ -189,7 +189,8 @@ TEST_F(TestConvert, TestConvertBatchNorm) {
|
|
|
|
|
|
|
|
|
|
TEST_F(TestConvert, TestConvertConvBackpropInput) {
|
|
|
|
|
auto prim = prim::kPrimConv2DBackpropInput;
|
|
|
|
|
prim->AddAttr("stride", MakeValue(1));
|
|
|
|
|
const std::vector<int> list{1,1};
|
|
|
|
|
prim->AddAttr("stride", MakeValue(list));
|
|
|
|
|
prim->AddAttr("pad", MakeValue(0));
|
|
|
|
|
prim->AddAttr("pad_mode", MakeValue(std::string("pad")));
|
|
|
|
|
prim->AddAttr("dilation", MakeValue(1));
|
|
|
|
@ -218,7 +219,8 @@ TEST_F(TestConvert, TestConvertConvBackpropInput) {
|
|
|
|
|
|
|
|
|
|
TEST_F(TestConvert, TestConvertConvBackpropFilter) {
|
|
|
|
|
auto prim = prim::kPrimConv2DBackpropFilter;
|
|
|
|
|
prim->AddAttr("stride", MakeValue(1));
|
|
|
|
|
const std::vector<int> list{1,1};
|
|
|
|
|
prim->AddAttr("stride", MakeValue(list));
|
|
|
|
|
prim->AddAttr("pad", MakeValue(0));
|
|
|
|
|
prim->AddAttr("pad_mode", MakeValue(std::string("pad")));
|
|
|
|
|
prim->AddAttr("dilation", MakeValue(1));
|
|
|
|
|