Home
last modified time | relevance | path

Searched refs:conv_2d (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_conv2d.cc33 auto conv_2d = std::make_shared<Conv2D>(); in TEST_F() local
34 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 …]