Home
last modified time | relevance | path

Searched refs:input_channels_stride (Results 1 – 5 of 5) sorted by relevance

/external/XNNPACK/test/
Dspace-to-depth-operator-tester.h96 inline SpaceToDepthOperatorTester& input_channels_stride(size_t input_channels_stride) { in input_channels_stride() function
97 assert(input_channels_stride >= 1); in input_channels_stride()
98 this->input_channels_stride_ = input_channels_stride; in input_channels_stride()
102 inline size_t input_channels_stride() const { in input_channels_stride() function
137 … (batch_size() * input_height() * input_width() - 1) * input_channels_stride() + input_channels()); in TestNHWCxX8()
150 input_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX8()
174 + bx * input_channels_stride() in TestNHWCxX8()
175 + ix * block_size() * input_channels_stride() in TestNHWCxX8()
176 + by * output_width() * block_size() * input_channels_stride() in TestNHWCxX8()
177 + iy * block_size() * output_width() * block_size() * input_channels_stride() in TestNHWCxX8()
[all …]
Ddepth-to-space-operator-tester.h94 inline DepthToSpaceOperatorTester& input_channels_stride(size_t input_channels_stride) { in input_channels_stride() function
95 assert(input_channels_stride >= 1); in input_channels_stride()
96 this->input_channels_stride_ = input_channels_stride; in input_channels_stride()
100 inline size_t input_channels_stride() const { in input_channels_stride() function
141 … (batch_size() * input_height() * input_width() - 1) * input_channels_stride() + input_channels()); in TestNHWCxX8()
154 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX8()
178 ((i * input_height() + iy) * input_width() + ix) * input_channels_stride() + in TestNHWCxX8()
190 << ", input stride: " << input_channels_stride() in TestNHWCxX8()
207 … (batch_size() * input_height() * input_width() - 1) * input_channels_stride() + input_channels()); in TestNHWCxX16()
220 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX16()
[all …]
Ddepth-to-space-nhwc.cc82 TEST(DEPTH_TO_SPACE_NHWC_X8, input_channels_stride) { in TEST() argument
87 .input_channels_stride(157) in TEST()
173 TEST(DEPTH_TO_SPACE_NHWC_X16, input_channels_stride) { in TEST() argument
178 .input_channels_stride(157) in TEST()
265 TEST(DEPTH_TO_SPACE_NHWC_X32, input_channels_stride) { in TEST() argument
270 .input_channels_stride(157) in TEST()
Dspace-to-depth-nhwc.cc88 TEST(SPACE_TO_DEPTH_NHWC_X8, input_channels_stride) { in TEST() argument
94 .input_channels_stride(10) in TEST()
187 TEST(SPACE_TO_DEPTH_NHWC_X16, input_channels_stride) { in TEST() argument
193 .input_channels_stride(10) in TEST()
286 TEST(SPACE_TO_DEPTH_NHWC_X32, input_channels_stride) { in TEST() argument
292 .input_channels_stride(10) in TEST()
Ddepth-to-space-nchw2nhwc.cc82 TEST(DEPTH_TO_SPACE_NCHW2NHWC_X32, input_channels_stride) { in TEST() argument
87 .input_channels_stride(157) in TEST()