blob: 1dd8dd8e8c9fee77fd60e568b49338a881814172 [file] [log] [blame] [edit]
#include <gtest/gtest.h>
#include <algorithm>
#include "transpose-operator-tester.h"
TEST(TRANSPOSE_ND_X32_2, 1D_redundant_dim) {
TransposeOperatorTester()
.num_dims(1)
.shape({1})
.perm({0})
.TestX32();
}
TEST(TRANSPOSE_ND_X8, 1D) {
TransposeOperatorTester()
.num_dims(1)
.shape({713})
.perm({0})
.TestX8();
}
TEST(TRANSPOSE_ND_X8, Run1D) {
TransposeOperatorTester()
.num_dims(1)
.shape({713})
.perm({0})
.TestRunX8();
}
TEST(TRANSPOSE_ND_X8, 2D) {
std::vector<size_t> perm{0,1};
do {
TransposeOperatorTester()
.num_dims(2)
.shape({37, 113})
.perm(perm)
.TestX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, Run2D) {
std::vector<size_t> perm{0,1};
do {
TransposeOperatorTester()
.num_dims(2)
.shape({37, 113})
.perm(perm)
.TestRunX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, 3D) {
std::vector<size_t> perm{0,1,2};
do {
TransposeOperatorTester()
.num_dims(3)
.shape({5, 7, 11})
.perm(perm)
.TestX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, Run3D) {
std::vector<size_t> perm{0,1,2};
do {
TransposeOperatorTester()
.num_dims(3)
.shape({5, 7, 11})
.perm(perm)
.TestRunX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 4D_copy) {
TransposeOperatorTester()
.num_dims(4)
.shape({2,2,1,1})
.perm({0,2,3,1})
.TestX32();
}
TEST(TRANSPOSE_ND_X8, 4D) {
std::vector<size_t> perm{0,1,2,3};
do {
TransposeOperatorTester()
.num_dims(4)
.shape({5,7,11,13})
.perm(perm)
.TestX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, Run4D) {
std::vector<size_t> perm{0,1,2,3};
do {
TransposeOperatorTester()
.num_dims(4)
.shape({5,7,11,13})
.perm(perm)
.TestRunX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, 5D) {
std::vector<size_t> perm{0,1,2,3,4};
do {
TransposeOperatorTester()
.num_dims(5)
.shape({3,5,7,11,13})
.perm(perm)
.TestX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, Run5D) {
std::vector<size_t> perm{0,1,2,3,4};
do {
TransposeOperatorTester()
.num_dims(5)
.shape({3,5,7,11,13})
.perm(perm)
.TestRunX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, 6D) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({2,3,5,7,11,13})
.perm(perm)
.TestX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X8, Run6D) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({2,3,5,7,11,13})
.perm(perm)
.TestRunX8();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, 1D) {
TransposeOperatorTester()
.num_dims(1)
.shape({713})
.perm({0})
.TestX16();
}
TEST(TRANSPOSE_ND_X16, Run1D) {
TransposeOperatorTester()
.num_dims(1)
.shape({713})
.perm({0})
.TestRunX16();
}
TEST(TRANSPOSE_ND_X16, 2D) {
std::vector<size_t> perm{0,1};
do {
TransposeOperatorTester()
.num_dims(2)
.shape({37, 113})
.perm(perm)
.TestX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, Run2D) {
std::vector<size_t> perm{0,1};
do {
TransposeOperatorTester()
.num_dims(2)
.shape({37, 113})
.perm(perm)
.TestRunX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, 3D) {
std::vector<size_t> perm{0,1,2};
do {
TransposeOperatorTester()
.num_dims(3)
.shape({5, 7, 11})
.perm(perm)
.TestX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, Run3D) {
std::vector<size_t> perm{0,1,2};
do {
TransposeOperatorTester()
.num_dims(3)
.shape({5, 7, 11})
.perm(perm)
.TestRunX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, 4D) {
std::vector<size_t> perm{0,1,2,3};
do {
TransposeOperatorTester()
.num_dims(4)
.shape({5,7,11,13})
.perm(perm)
.TestX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, Run4D) {
std::vector<size_t> perm{0,1,2,3};
do {
TransposeOperatorTester()
.num_dims(4)
.shape({5,7,11,13})
.perm(perm)
.TestRunX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, 5D) {
std::vector<size_t> perm{0,1,2,3,4};
do {
TransposeOperatorTester()
.num_dims(5)
.shape({3,5,7,11,13})
.perm(perm)
.TestX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, Run5D) {
std::vector<size_t> perm{0,1,2,3,4};
do {
TransposeOperatorTester()
.num_dims(5)
.shape({3,5,7,11,13})
.perm(perm)
.TestX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, 6D) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({2,3,5,7,11,13})
.perm(perm)
.TestX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X16, Run6D) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({2,3,5,7,11,13})
.perm(perm)
.TestRunX16();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 1D) {
TransposeOperatorTester()
.num_dims(1)
.shape({713})
.perm({0})
.TestX32();
}
TEST(TRANSPOSE_ND_X32, Run1D) {
TransposeOperatorTester()
.num_dims(1)
.shape({713})
.perm({0})
.TestRunX32();
}
TEST(TRANSPOSE_ND_X32, 2D_all_dimensions_redundant) {
TransposeOperatorTester()
.num_dims(2)
.shape({1, 1})
.perm({1, 0})
.TestX32();
}
TEST(TRANSPOSE_ND_X32, 2D) {
std::vector<size_t> perm{0,1};
do {
TransposeOperatorTester()
.num_dims(2)
.shape({37, 113})
.perm(perm)
.TestX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, Run2D) {
std::vector<size_t> perm{0,1};
do {
TransposeOperatorTester()
.num_dims(2)
.shape({37, 113})
.perm(perm)
.TestRunX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 3D_redundant_dimension) {
TransposeOperatorTester()
.num_dims(3)
.shape({2, 1, 3})
.perm({0, 2, 1})
.TestX32();
}
TEST(TRANSPOSE_ND_X32, 3D) {
std::vector<size_t> perm{0,1,2};
do {
TransposeOperatorTester()
.num_dims(3)
.shape({5, 7, 11})
.perm(perm)
.TestX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, Run3D) {
std::vector<size_t> perm{0,1,2};
do {
TransposeOperatorTester()
.num_dims(3)
.shape({5, 7, 11})
.perm(perm)
.TestRunX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 4D) {
std::vector<size_t> perm{0,1,2,3};
do {
TransposeOperatorTester()
.num_dims(4)
.shape({5,7,11,13})
.perm(perm)
.TestX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, Run4D) {
std::vector<size_t> perm{0,1,2,3};
do {
TransposeOperatorTester()
.num_dims(4)
.shape({5,7,11,13})
.perm(perm)
.TestRunX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 5D) {
std::vector<size_t> perm{0,1,2,3,4};
do {
TransposeOperatorTester()
.num_dims(5)
.shape({3,5,7,11,13})
.perm(perm)
.TestX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, Run5D) {
std::vector<size_t> perm{0,1,2,3,4};
do {
TransposeOperatorTester()
.num_dims(5)
.shape({3,5,7,11,13})
.perm(perm)
.TestRunX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 6D) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({2,3,5,7,11,13})
.perm(perm)
.TestX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, Run6D) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({2,3,5,7,11,13})
.perm(perm)
.TestRunX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}
TEST(TRANSPOSE_ND_X32, 6D_DIMS_1) {
std::vector<size_t> perm{0,1,2,3,4,5};
do {
TransposeOperatorTester()
.num_dims(6)
.shape({1,1,1,2,3,4})
.perm(perm)
.TestX32();
} while (std::next_permutation(perm.begin(), perm.end()));
}