Searched refs:conv_2d (Results 1 – 1 of 1) sorted by relevance
33 auto conv_2d = std::make_shared<Conv2D>(); in TEST_F() local34 conv_2d->Init(64, {7, 7}); in TEST_F()35 std::vector<int64_t> kernel_size = conv_2d->get_kernel_size(); in TEST_F()39 std::vector<int64_t> stride = conv_2d->get_stride(); in TEST_F()43 std::vector<int64_t> dilation = conv_2d->get_dilation(); in TEST_F()47 EXPECT_EQ(conv_2d->get_pad_mode(), VALID); in TEST_F()48 std::vector<int64_t> pad = conv_2d->get_pad(); in TEST_F()52 EXPECT_EQ(conv_2d->get_mode(), 1); in TEST_F()53 EXPECT_EQ(conv_2d->get_group(), 1); in TEST_F()54 EXPECT_EQ(conv_2d->get_out_channel(), 64); in TEST_F()[all …]