|
|
|
|
|
|
|
|
|
|
|
#include <gtest/gtest.h> |
|
|
|
#include "bankers-rounding-operator-tester.h" |
|
|
|
|
|
TEST(BANKERS_ROUNDING_NC_F32, unit_batch) { |
|
for (size_t channels = 1; channels < 100; channels++) { |
|
BankersRoundingOperatorTester() |
|
.batch_size(1) |
|
.channels(channels) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(BANKERS_ROUNDING_NC_F32, small_batch) { |
|
for (size_t channels = 1; channels < 100; channels++) { |
|
BankersRoundingOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(BANKERS_ROUNDING_NC_F32, small_batch_with_input_stride) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
BankersRoundingOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.input_stride(129) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(BANKERS_ROUNDING_NC_F32, small_batch_with_output_stride) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
BankersRoundingOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.output_stride(117) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |
|
|
|
TEST(BANKERS_ROUNDING_NC_F32, small_batch_with_input_and_output_stride) { |
|
for (size_t channels = 1; channels < 100; channels += 15) { |
|
BankersRoundingOperatorTester() |
|
.batch_size(3) |
|
.channels(channels) |
|
.input_stride(129) |
|
.output_stride(117) |
|
.iterations(3) |
|
.TestRunF32(); |
|
} |
|
} |