1 // Copyright 2020 Google LLC
2 //
3 // This source code is licensed under the BSD-style license found in the
4 // LICENSE file in the root directory of this source tree.
5
6 #include <gtest/gtest.h>
7
8 #include "ceiling-operator-tester.h"
9
10
TEST(CEILING_NC_F16,unit_batch)11 TEST(CEILING_NC_F16, unit_batch) {
12 for (size_t channels = 1; channels < 100; channels++) {
13 CeilingOperatorTester()
14 .batch_size(1)
15 .channels(channels)
16 .iterations(3)
17 .TestF16();
18 }
19 }
20
TEST(CEILING_NC_F16,small_batch)21 TEST(CEILING_NC_F16, small_batch) {
22 for (size_t channels = 1; channels < 100; channels++) {
23 CeilingOperatorTester()
24 .batch_size(3)
25 .channels(channels)
26 .iterations(3)
27 .TestF16();
28 }
29 }
30
TEST(CEILING_NC_F16,small_batch_with_input_stride)31 TEST(CEILING_NC_F16, small_batch_with_input_stride) {
32 for (size_t channels = 1; channels < 100; channels += 15) {
33 CeilingOperatorTester()
34 .batch_size(3)
35 .channels(channels)
36 .input_stride(129)
37 .iterations(3)
38 .TestF16();
39 }
40 }
41
TEST(CEILING_NC_F16,small_batch_with_output_stride)42 TEST(CEILING_NC_F16, small_batch_with_output_stride) {
43 for (size_t channels = 1; channels < 100; channels += 15) {
44 CeilingOperatorTester()
45 .batch_size(3)
46 .channels(channels)
47 .output_stride(117)
48 .iterations(3)
49 .TestF16();
50 }
51 }
52
TEST(CEILING_NC_F16,small_batch_with_input_and_output_stride)53 TEST(CEILING_NC_F16, small_batch_with_input_and_output_stride) {
54 for (size_t channels = 1; channels < 100; channels += 15) {
55 CeilingOperatorTester()
56 .batch_size(3)
57 .channels(channels)
58 .input_stride(129)
59 .output_stride(117)
60 .iterations(3)
61 .TestF16();
62 }
63 }
64
65
TEST(CEILING_NC_F32,unit_batch)66 TEST(CEILING_NC_F32, unit_batch) {
67 for (size_t channels = 1; channels < 100; channels++) {
68 CeilingOperatorTester()
69 .batch_size(1)
70 .channels(channels)
71 .iterations(3)
72 .TestF32();
73 }
74 }
75
TEST(CEILING_NC_F32,small_batch)76 TEST(CEILING_NC_F32, small_batch) {
77 for (size_t channels = 1; channels < 100; channels++) {
78 CeilingOperatorTester()
79 .batch_size(3)
80 .channels(channels)
81 .iterations(3)
82 .TestF32();
83 }
84 }
85
TEST(CEILING_NC_F32,small_batch_with_input_stride)86 TEST(CEILING_NC_F32, small_batch_with_input_stride) {
87 for (size_t channels = 1; channels < 100; channels += 15) {
88 CeilingOperatorTester()
89 .batch_size(3)
90 .channels(channels)
91 .input_stride(129)
92 .iterations(3)
93 .TestF32();
94 }
95 }
96
TEST(CEILING_NC_F32,small_batch_with_output_stride)97 TEST(CEILING_NC_F32, small_batch_with_output_stride) {
98 for (size_t channels = 1; channels < 100; channels += 15) {
99 CeilingOperatorTester()
100 .batch_size(3)
101 .channels(channels)
102 .output_stride(117)
103 .iterations(3)
104 .TestF32();
105 }
106 }
107
TEST(CEILING_NC_F32,small_batch_with_input_and_output_stride)108 TEST(CEILING_NC_F32, small_batch_with_input_and_output_stride) {
109 for (size_t channels = 1; channels < 100; channels += 15) {
110 CeilingOperatorTester()
111 .batch_size(3)
112 .channels(channels)
113 .input_stride(129)
114 .output_stride(117)
115 .iterations(3)
116 .TestF32();
117 }
118 }
119