test_transpose_ut_cases.inc 651 Bytes
Newer Older
arai713's avatar
arai713 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#pragma once

TYPED_TEST(TestTranspose, Test1)
{
    // for 16, 8, 16, 32, 8
    std::vector<int> Ms{1, 2, 3, 4, 5, 6};
    std::vector<index_t> lengths{16, 8, 16, 32, 8};
    /**constexpr int N = 16;
    constexpr int C = 8;
    constexpr int D = 16;
    constexpr int H = 32;
    constexpr int W = 8;**/

    this->Run();
}


TYPED_TEST(TestTranpose, Test2)
{
    std::vector<int> Ms{127, 255, 312, 799, 1573};
    std::vector<index_t> lengths{16, 8, 16, 32, 16};
    /**constexpr int N = 16;
    constexpr int C = 8;
    constexpr int D = 16;
    constexpr int H = 32;
    constexpr int W = 8;**/

    this->Run();
}