| /external/XNNPACK/test/ |
| D | fill-microkernel-tester.h | 44 inline FillMicrokernelTester& output_stride(size_t output_stride) { in output_stride() argument
45 assert(output_stride != 0); in output_stride()
46 this->output_stride_ = output_stride; in output_stride()
50 inline size_t output_stride() const { in output_stride() function
68 ASSERT_GE(output_stride(), channels()); in Test()
74 std::vector<uint32_t> output((rows() - 1) * output_stride() + channels()); in Test()
86 output_stride() * sizeof(uint32_t), in Test()
92 ASSERT_EQ(output[i * output_stride() + c], fill_value) in Test()
96 …tput value 0x" << std::hex << std::setw(8) << std::setfill('0') << output[i * output_stride() + c]; in Test()
100 for (size_t c = channels(); c < output_stride(); c++) { in Test()
[all …]
|
| D | clamp-operator-tester.h | 52 inline ClampOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
53 assert(output_stride != 0); in output_stride()
54 this->output_stride_ = output_stride; in output_stride()
58 inline size_t output_stride() const { in output_stride() function
120 std::vector<uint8_t> output((batch_size() - 1) * output_stride() + channels()); in TestU8()
141 channels(), input_stride(), output_stride(), in TestU8()
162 ASSERT_LE(uint32_t(output[i * output_stride() + c]), uint32_t(qmax())) in TestU8()
164 ASSERT_GE(uint32_t(output[i * output_stride() + c]), uint32_t(qmin())) in TestU8()
166 … ASSERT_EQ(uint32_t(output_ref[i * channels() + c]), uint32_t(output[i * output_stride() + c])) in TestU8()
181 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
[all …]
|
| D | channel-shuffle-nc.cc | 162 .output_stride(513) in TEST()
174 .output_stride(513) in TEST()
186 .output_stride(513) in TEST()
199 .output_stride(1111) in TEST()
213 .output_stride(513) in TEST()
226 .output_stride(513) in TEST()
239 .output_stride(513) in TEST()
253 .output_stride(1111) in TEST()
408 .output_stride(513) in TEST()
420 .output_stride(513) in TEST()
[all …]
|
| D | channel-shuffle-operator-tester.h | 66 inline ChannelShuffleOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
67 assert(output_stride != 0); in output_stride()
68 this->output_stride_ = output_stride; in output_stride()
72 inline size_t output_stride() const { in output_stride() function
106 std::vector<uint8_t> output((batch_size() - 1) * output_stride() + channels()); in TestX8()
118 input_stride(), output_stride(), in TestX8()
140 uint32_t(output[i * output_stride() + c * groups() + g])) in TestX8()
154 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestX32()
166 input_stride(), output_stride(), in TestX32()
188 output[i * output_stride() + c * groups() + g]) in TestX32()
|
| D | hardswish-operator-tester.h | 50 inline HardSwishOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
51 assert(output_stride != 0); in output_stride()
52 this->output_stride_ = output_stride; in output_stride()
56 inline size_t output_stride() const { in output_stride() function
92 std::vector<uint16_t> output((batch_size() - 1) * output_stride() + channels()); in TestF16()
111 channels(), input_stride(), output_stride(), in TestF16()
134 …ASSERT_NEAR(fp16_ieee_to_fp32_value(output[i * output_stride() + c]), output_ref[i * channels() + … in TestF16()
148 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
169 channels(), input_stride(), output_stride(), in TestF32()
189 …put_ref[i * channels() + c], output[i * output_stride() + c], std::max(1.0e-7f, std::abs(output[i … in TestF32()
|
| D | global-average-pooling-operator-tester.h | 64 inline GlobalAveragePoolingOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
65 assert(output_stride != 0); in output_stride()
66 this->output_stride_ = output_stride; in output_stride()
70 inline size_t output_stride() const { in output_stride() function
162 std::vector<uint8_t> output(batch_size() * output_stride()); in TestNWCxQU8()
187 channels(), input_stride(), output_stride(), in TestNWCxQU8()
214 ASSERT_LE(uint32_t(output[i * output_stride() + c]), uint32_t(qmax())); in TestNWCxQU8()
215 ASSERT_GE(uint32_t(output[i * output_stride() + c]), uint32_t(qmin())); in TestNWCxQU8()
216 …ASSERT_NEAR(float(int32_t(output[i * output_stride() + c])), output_ref[i * channels() + c], 0.80f) in TestNWCxQU8()
231 std::vector<int8_t> output(batch_size() * output_stride()); in TestNWCxQS8()
[all …]
|
| D | copy-operator-tester.h | 49 inline CopyOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
90 std::vector<uint32_t> output((batch_size() - 1) * output_stride() + channels()); in TestX32()
109 channels(), input_stride(), output_stride(), in TestX32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestX32()
|
| D | negate-operator-tester.h | 48 inline NegateOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
49 assert(output_stride != 0); in output_stride()
50 this->output_stride_ = output_stride; in output_stride()
54 inline size_t output_stride() const { in output_stride() function
89 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
108 channels(), input_stride(), output_stride(), in TestF32()
128 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | ceiling-operator-tester.h | 49 inline CeilingOperatorTester& output_stride(size_t output_stride) { in output_stride() function
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
90 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
109 channels(), input_stride(), output_stride(), in TestF32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | square-operator-tester.h | 48 inline SquareOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
49 assert(output_stride != 0); in output_stride()
50 this->output_stride_ = output_stride; in output_stride()
54 inline size_t output_stride() const { in output_stride() function
89 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
109 channels(), input_stride(), output_stride(), in TestF32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | square-root-operator-tester.h | 48 inline SquareRootOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
49 assert(output_stride != 0); in output_stride()
50 this->output_stride_ = output_stride; in output_stride()
54 inline size_t output_stride() const { in output_stride() function
89 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
108 channels(), input_stride(), output_stride(), in TestF32()
128 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | truncation-operator-tester.h | 49 inline TruncationOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
90 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
109 channels(), input_stride(), output_stride(), in TestF32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | abs-operator-tester.h | 49 inline AbsOperatorTester& output_stride(size_t output_stride) { in output_stride() function
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
90 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
109 channels(), input_stride(), output_stride(), in TestF32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | bankers-rounding-operator-tester.h | 49 inline BankersRoundingOperatorTester& output_stride(size_t output_stride) { in output_stride() argument
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
90 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
109 channels(), input_stride(), output_stride(), in TestF32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | floor-operator-tester.h | 49 inline FloorOperatorTester& output_stride(size_t output_stride) { in output_stride() function
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
90 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
109 channels(), input_stride(), output_stride(), in TestF32()
129 ASSERT_EQ(output_ref[i * channels() + c], output[i * output_stride() + c]) in TestF32()
|
| D | prelu-microkernel-tester.h | 63 inline PReLUMicrokernelTester& output_stride(size_t output_stride) { in output_stride() function
64 assert(output_stride != 0); in output_stride()
65 this->output_stride_ = output_stride; in output_stride()
69 inline size_t output_stride() const { in output_stride() function
106 …std::vector<uint16_t> y(channels() + (rows() - 1) * output_stride() + XNN_EXTRA_BYTES / sizeof(uin… in Test()
131 y.data(), output_stride() * sizeof(uint16_t)); in Test()
136 ASSERT_EQ(fp16_ieee_to_fp32_value(y[n * output_stride() + c]), y_ref[n * channels() + c]) in Test()
152 …std::vector<float> y(channels() + (rows() - 1) * output_stride() + XNN_EXTRA_BYTES / sizeof(float)… in Test()
176 y.data(), output_stride() * sizeof(float)); in Test()
181 ASSERT_EQ(y[n * output_stride() + c], y_ref[n * channels() + c]) in Test()
|
| D | f32-ibilinear.cc | 59 TEST(F32_IBILINEAR__SCALAR_C1, output_stride) { in TEST() argument
65 .output_stride(7) in TEST()
128 TEST(F32_IBILINEAR__SCALAR_C2, output_stride) { in TEST() argument
134 .output_stride(13) in TEST()
197 TEST(F32_IBILINEAR__SCALAR_C4, output_stride) { in TEST() argument
203 .output_stride(23) in TEST()
273 TEST(F32_IBILINEAR__NEON_C4, output_stride) { in TEST() argument
280 .output_stride(23) in TEST()
352 TEST(F32_IBILINEAR__NEON_C8, output_stride) { in TEST() argument
359 .output_stride(43) in TEST()
[all …]
|
| D | maxpool-microkernel-tester.h | 119 inline MaxPoolMicrokernelTester& output_stride(size_t output_stride) { in output_stride() function
120 assert(output_stride != 0); in output_stride()
121 this->output_stride_ = output_stride; in output_stride()
125 inline size_t output_stride() const { in output_stride() function
170 (output_pixels() - 1) * output_stride() + channels());
212 (output_stride() - channels()) * sizeof(uint8_t),
218 ASSERT_GE(uint32_t(output[x * output_stride() + c]), uint32_t(qmin()))
222 ASSERT_LE(uint32_t(output[x * output_stride() + c]), uint32_t(qmax()))
226 … ASSERT_EQ(uint32_t(output_ref[x * channels() + c]), uint32_t(output[x * output_stride() + c]))
244 (output_pixels() - 1) * output_stride() + channels());
[all …]
|
| D | elu-operator-tester.h | 49 inline ELUOperatorTester& output_stride(size_t output_stride) { in output_stride() function
50 assert(output_stride != 0); in output_stride()
51 this->output_stride_ = output_stride; in output_stride()
55 inline size_t output_stride() const { in output_stride() function
100 std::vector<float> output((batch_size() - 1) * output_stride() + channels()); in TestF32()
120 channels(), input_stride(), output_stride(), in TestF32()
141 ASSERT_NEAR(output[i * output_stride() + c], in TestF32()
|
| D | pad-microkernel-tester.h | 81 inline PadMicrokernelTester& output_stride(size_t output_stride) { in output_stride() argument
82 assert(output_stride != 0); in output_stride()
83 this->output_stride_ = output_stride; in output_stride()
87 inline size_t output_stride() const { in output_stride() function
111 …32_t> output((pre_padding() + input_channels() + post_padding()) + (rows() - 1) * output_stride()); in Test()
135 output.data(), output_stride() * sizeof(uint32_t)); in Test()
140 ASSERT_EQ(output_ref[i * output_channels() + c], output[i * output_stride() + c]) in Test()
|
| /external/XNNPACK/src/operators/ |
| D | unary-elementwise-nc.c | 23 size_t output_stride, in create_unary_elementwise_nc() argument
54 if (output_stride < channels) { in create_unary_elementwise_nc()
58 xnn_operator_type_to_string(xnn_operator_type_clamp_nc_f32), output_stride, channels); in create_unary_elementwise_nc()
72 unary_elementwise_op->output_pixel_stride = output_stride; in create_unary_elementwise_nc()
108 const size_t output_stride = unary_elementwise_op->output_pixel_stride; in setup_unary_elementwise_nc() local
112 if ((((input_stride ^ channels) | (output_stride ^ channels)) == 0) || batch_size == 1) { in setup_unary_elementwise_nc()
118 .y_stride = output_stride << log2_element_size, in setup_unary_elementwise_nc()
134 .y_stride = output_stride << log2_element_size, in setup_unary_elementwise_nc()
153 size_t output_stride, in xnn_create_clamp_nc_u8() argument
168 channels, input_stride, output_stride, flags, in xnn_create_clamp_nc_u8()
[all …]
|
| /external/XNNPACK/src/f32-spmm/gen/ |
| D | 8x4-minmax-neonfma.c | 25 size_t output_stride, in xnn_f32_spmm_minmax_ukernel_8x4__neonfma() argument
34 size_t output_decrement = output_stride * nc - 8 * sizeof(float); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
89 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
92 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
95 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
98 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
129 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
174 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
176 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
178 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_8x4__neonfma()
[all …]
|
| D | 4x4-minmax-neonfma.c | 25 size_t output_stride, in xnn_f32_spmm_minmax_ukernel_4x4__neonfma() argument
34 size_t output_decrement = output_stride * nc - 4 * sizeof(float); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
71 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
73 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
75 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
77 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
102 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
147 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
149 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
151 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_4x4__neonfma()
[all …]
|
| D | 12x4-minmax-neonfma.c | 25 size_t output_stride, in xnn_f32_spmm_minmax_ukernel_12x4__neonfma() argument
34 size_t output_decrement = output_stride * nc - 12 * sizeof(float); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
107 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
111 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
115 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
119 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
156 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
219 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
222 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
225 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_12x4__neonfma()
[all …]
|
| D | 16x4-minmax-neonfma.c | 25 size_t output_stride, in xnn_f32_spmm_minmax_ukernel_16x4__neonfma() argument
34 size_t output_decrement = output_stride * nc - 16 * sizeof(float); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
125 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
130 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
135 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
140 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
183 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
246 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
249 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
252 output = (float*restrict) ((uintptr_t) output + output_stride); in xnn_f32_spmm_minmax_ukernel_16x4__neonfma()
[all …]
|