Lines Matching refs:output_channels
71 return output_channels() * block_size() * block_size(); in input_channels()
74 inline DepthToSpaceOperatorTester& output_channels(size_t output_channels) { in output_channels() argument
75 assert(output_channels != 0); in output_channels()
76 this->output_channels_ = output_channels; in output_channels()
80 inline size_t output_channels() const { in output_channels() function
117 return output_channels(); in output_channels_stride()
119 assert(this->output_channels_stride_ >= output_channels()); in output_channels_stride()
143 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNHWCxX8()
154 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX8()
176 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX8()
179 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX8()
189 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX8()
209 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNHWCxX16()
220 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX16()
242 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX16()
245 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX16()
255 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX16()
275 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNHWCxX32()
286 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX32()
308 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX32()
311 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX32()
321 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX32()
341 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNCHW2NHWCxX32()
352 output_channels(), input_channels_stride(), output_channels_stride(), in TestNCHW2NHWCxX32()
374 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNCHW2NHWCxX32()
377 … (((by * block_size() + bx) * output_channels() + oc) * input_height() + iy) * input_width() + ix; in TestNCHW2NHWCxX32()
387 << ", output channel: " << oc << " / " << output_channels() in TestNCHW2NHWCxX32()