| // Copyright (c) Facebook, Inc. and its affiliates. |
| // All rights reserved. |
| // |
| // Copyright 2019 Google LLC |
| // |
| // This source code is licensed under the BSD-style license found in the |
| // LICENSE file in the root directory of this source tree. |
| |
| #include <gtest/gtest.h> |
| |
| #include "sigmoid-operator-tester.h" |
| |
| #include <xnnpack/params.h> |
| |
| |
| TEST(SIGMOID_OP, unit_batch) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(1) |
| .channels(channels) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, unit_batch_with_qmin) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(1) |
| .channels(channels) |
| .qmin(128) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, unit_batch_with_qmax) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(1) |
| .channels(channels) |
| .qmax(128) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, unit_batch_with_input_scale) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { |
| SigmoidOperatorTester() |
| .batch_size(1) |
| .channels(channels) |
| .input_scale(input_scale) |
| .iterations(1) |
| .TestQ8(); |
| } |
| } |
| } |
| |
| TEST(SIGMOID_OP, unit_batch_with_input_zero_point) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) { |
| SigmoidOperatorTester() |
| .batch_size(1) |
| .channels(channels) |
| .input_zero_point(uint8_t(input_zero_point)) |
| .iterations(1) |
| .TestQ8(); |
| } |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch_with_input_stride) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_stride(129) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch_with_output_stride) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .output_stride(117) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch_with_qmin) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .qmin(128) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch_with_qmax) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .qmax(128) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch_with_input_scale) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_scale(input_scale) |
| .iterations(1) |
| .TestQ8(); |
| } |
| } |
| } |
| |
| TEST(SIGMOID_OP, small_batch_with_input_zero_point) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_zero_point(uint8_t(input_zero_point)) |
| .iterations(1) |
| .TestQ8(); |
| } |
| } |
| } |
| |
| TEST(SIGMOID_OP, strided_batch) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_stride(129) |
| .output_stride(117) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, strided_batch_with_qmin) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_stride(129) |
| .output_stride(117) |
| .qmin(128) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, strided_batch_with_qmax) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_stride(129) |
| .output_stride(117) |
| .qmax(128) |
| .iterations(3) |
| .TestQ8(); |
| } |
| } |
| |
| TEST(SIGMOID_OP, strided_batch_with_input_scale) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_stride(129) |
| .output_stride(117) |
| .input_scale(input_scale) |
| .iterations(1) |
| .TestQ8(); |
| } |
| } |
| } |
| |
| TEST(SIGMOID_OP, strided_batch_with_input_zero_point) { |
| for (size_t channels = 1; channels < 100; channels += 15) { |
| for (int32_t input_zero_point = 0; input_zero_point <= 255; input_zero_point += 51) { |
| SigmoidOperatorTester() |
| .batch_size(3) |
| .channels(channels) |
| .input_stride(129) |
| .output_stride(117) |
| .input_zero_point(uint8_t(input_zero_point)) |
| .iterations(1) |
| .TestQ8(); |
| } |
| } |
| } |