|
|
|
|
|
|
|
|
|
|
|
#include <limits> |
|
|
|
#include <gtest/gtest.h> |
|
|
|
#include "clamp-operator-tester.h" |
|
|
|
TEST(CLAMP_NC_F32, unit_batch) { |
|
for (size_t channels = 1; channels < 100; channels++) { |
|
ClampOperatorTester() |
|
.batch_size(1) |
|
.channels(channels) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, unit_batch_with_qmin) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
for (int32_t qmin = std::numeric_limits<int16_t>::min() + 1; |
|
qmin < std::numeric_limits<int16_t>::max(); |
|
qmin += 257) |
|
{ |
|
ClampOperatorTester() |
|
.batch_size(1) |
|
.channels(channels) |
|
.qmin(qmin) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, unit_batch_with_qmax) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
for (int32_t qmax = std::numeric_limits<int16_t>::min() + 1; |
|
qmax < std::numeric_limits<int16_t>::max(); |
|
qmax += 257) |
|
{ |
|
ClampOperatorTester() |
|
.batch_size(1) |
|
.channels(channels) |
|
.qmax(qmax) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, unit_batch_with_relu) { |
|
for (size_t channels = 1; channels < 100; channels++) { |
|
ClampOperatorTester() |
|
.batch_size(1) |
|
.channels(channels) |
|
.relu_activation(true) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, small_batch) { |
|
for (size_t channels = 1; channels < 100; channels++) { |
|
ClampOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, small_batch_with_input_stride) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
ClampOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.input_stride(129) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, small_batch_with_output_stride) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
ClampOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.output_stride(117) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(CLAMP_NC_F32, small_batch_with_input_and_output_stride) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
ClampOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.input_stride(129) |
|
.output_stride(117) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |