|
|
|
|
|
|
|
|
|
|
|
#include <gtest/gtest.h> |
|
|
|
#include "constant-pad-operator-tester.h" |
|
|
|
constexpr size_t kDim1 = 2; |
|
constexpr size_t kDim2 = 3; |
|
constexpr size_t kDim3 = 2; |
|
constexpr size_t kDim4 = 3; |
|
constexpr size_t kDim5 = 2; |
|
constexpr size_t kDim6 = 3; |
|
constexpr size_t kDim1PrePad = kDim1 / 2; |
|
constexpr size_t kDim1PostPad = kDim1 / 2 + 1; |
|
constexpr size_t kDim2PrePad = kDim2 / 2; |
|
constexpr size_t kDim2PostPad = kDim2 / 2 + 1; |
|
constexpr size_t kDim3PrePad = kDim3 / 2; |
|
constexpr size_t kDim3PostPad = kDim3 / 2 + 1; |
|
constexpr size_t kDim4PrePad = kDim4 / 2; |
|
constexpr size_t kDim4PostPad = kDim4 / 2 + 1; |
|
constexpr size_t kDim5PrePad = kDim5 / 2; |
|
constexpr size_t kDim5PostPad = kDim5 / 2 + 1; |
|
constexpr size_t kDim6PrePad = kDim6 / 2; |
|
constexpr size_t kDim6PostPad = kDim6 / 2 + 1; |
|
|
|
|
|
TEST(CONSTANT_PAD_ND_X8, 0d) { |
|
ConstantPadOperatorTester() |
|
.TestX8(); |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X8, 1d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1}) |
|
.pre_paddings({dim1_pre_pad}) |
|
.post_paddings({dim1_post_pad}) |
|
.TestX8(); |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X8, 2d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad}) |
|
.TestX8(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X8, 3d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad}) |
|
.TestX8(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X8, 4d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad}) |
|
.TestX8(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X8, 5d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
for (size_t dim5_pre_pad = 0; dim5_pre_pad <= kDim5PrePad; dim5_pre_pad += kDim5PrePad) { |
|
for (size_t dim5_post_pad = 0; dim5_post_pad <= kDim5PostPad; dim5_post_pad += kDim5PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4, kDim5}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad, dim5_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad, dim5_post_pad}) |
|
.TestX8(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X8, 6d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
for (size_t dim5_pre_pad = 0; dim5_pre_pad <= kDim5PrePad; dim5_pre_pad += kDim5PrePad) { |
|
for (size_t dim5_post_pad = 0; dim5_post_pad <= kDim5PostPad; dim5_post_pad += kDim5PostPad) { |
|
for (size_t dim6_pre_pad = 0; dim6_pre_pad <= kDim6PrePad; dim6_pre_pad += kDim6PrePad) { |
|
for (size_t dim6_post_pad = 0; dim6_post_pad <= kDim6PostPad; dim6_post_pad += kDim6PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4, kDim5, kDim6}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad, dim5_pre_pad, dim6_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad, dim5_post_pad, dim6_post_pad}) |
|
.TestX8(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
|
|
TEST(CONSTANT_PAD_ND_X16, 0d) { |
|
ConstantPadOperatorTester() |
|
.TestX16(); |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X16, 1d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1}) |
|
.pre_paddings({dim1_pre_pad}) |
|
.post_paddings({dim1_post_pad}) |
|
.TestX16(); |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X16, 2d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad}) |
|
.TestX16(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X16, 3d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad}) |
|
.TestX16(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X16, 4d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad}) |
|
.TestX16(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X16, 5d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
for (size_t dim5_pre_pad = 0; dim5_pre_pad <= kDim5PrePad; dim5_pre_pad += kDim5PrePad) { |
|
for (size_t dim5_post_pad = 0; dim5_post_pad <= kDim5PostPad; dim5_post_pad += kDim5PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4, kDim5}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad, dim5_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad, dim5_post_pad}) |
|
.TestX16(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X16, 6d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
for (size_t dim5_pre_pad = 0; dim5_pre_pad <= kDim5PrePad; dim5_pre_pad += kDim5PrePad) { |
|
for (size_t dim5_post_pad = 0; dim5_post_pad <= kDim5PostPad; dim5_post_pad += kDim5PostPad) { |
|
for (size_t dim6_pre_pad = 0; dim6_pre_pad <= kDim6PrePad; dim6_pre_pad += kDim6PrePad) { |
|
for (size_t dim6_post_pad = 0; dim6_post_pad <= kDim6PostPad; dim6_post_pad += kDim6PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4, kDim5, kDim6}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad, dim5_pre_pad, dim6_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad, dim5_post_pad, dim6_post_pad}) |
|
.TestX16(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
|
|
TEST(CONSTANT_PAD_ND_X32, 0d) { |
|
ConstantPadOperatorTester() |
|
.TestX32(); |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X32, 1d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1}) |
|
.pre_paddings({dim1_pre_pad}) |
|
.post_paddings({dim1_post_pad}) |
|
.TestX32(); |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X32, 2d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad}) |
|
.TestX32(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X32, 3d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad}) |
|
.TestX32(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X32, 4d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad}) |
|
.TestX32(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X32, 5d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
for (size_t dim5_pre_pad = 0; dim5_pre_pad <= kDim5PrePad; dim5_pre_pad += kDim5PrePad) { |
|
for (size_t dim5_post_pad = 0; dim5_post_pad <= kDim5PostPad; dim5_post_pad += kDim5PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4, kDim5}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad, dim5_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad, dim5_post_pad}) |
|
.TestX32(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
TEST(CONSTANT_PAD_ND_X32, 6d) { |
|
for (size_t dim1_pre_pad = 0; dim1_pre_pad <= kDim1PrePad; dim1_pre_pad += kDim1PrePad) { |
|
for (size_t dim1_post_pad = 0; dim1_post_pad <= kDim1PostPad; dim1_post_pad += kDim1PostPad) { |
|
for (size_t dim2_pre_pad = 0; dim2_pre_pad <= kDim2PrePad; dim2_pre_pad += kDim2PrePad) { |
|
for (size_t dim2_post_pad = 0; dim2_post_pad <= kDim2PostPad; dim2_post_pad += kDim2PostPad) { |
|
for (size_t dim3_pre_pad = 0; dim3_pre_pad <= kDim3PrePad; dim3_pre_pad += kDim3PrePad) { |
|
for (size_t dim3_post_pad = 0; dim3_post_pad <= kDim3PostPad; dim3_post_pad += kDim3PostPad) { |
|
for (size_t dim4_pre_pad = 0; dim4_pre_pad <= kDim4PrePad; dim4_pre_pad += kDim4PrePad) { |
|
for (size_t dim4_post_pad = 0; dim4_post_pad <= kDim4PostPad; dim4_post_pad += kDim4PostPad) { |
|
for (size_t dim5_pre_pad = 0; dim5_pre_pad <= kDim5PrePad; dim5_pre_pad += kDim5PrePad) { |
|
for (size_t dim5_post_pad = 0; dim5_post_pad <= kDim5PostPad; dim5_post_pad += kDim5PostPad) { |
|
for (size_t dim6_pre_pad = 0; dim6_pre_pad <= kDim6PrePad; dim6_pre_pad += kDim6PrePad) { |
|
for (size_t dim6_post_pad = 0; dim6_post_pad <= kDim6PostPad; dim6_post_pad += kDim6PostPad) { |
|
ConstantPadOperatorTester() |
|
.input_shape({kDim1, kDim2, kDim3, kDim4, kDim5, kDim6}) |
|
.pre_paddings({dim1_pre_pad, dim2_pre_pad, dim3_pre_pad, dim4_pre_pad, dim5_pre_pad, dim6_pre_pad}) |
|
.post_paddings({dim1_post_pad, dim2_post_pad, dim3_post_pad, dim4_post_pad, dim5_post_pad, dim6_post_pad}) |
|
.TestX32(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|